122 lines
3.6 KiB
Go
122 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"time"
|
|
|
|
messages "git.tornberg.me/go-cart-actor/proto"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
)
|
|
|
|
// cartActorGRPCServer implements the ControlPlane gRPC services.
|
|
// It delegates cart operations to a grain pool and cluster operations to a synced pool.
|
|
type cartActorGRPCServer struct {
|
|
messages.UnimplementedControlPlaneServer
|
|
|
|
pool *CartPool
|
|
}
|
|
|
|
// NewCartActorGRPCServer creates and initializes the server.
|
|
func NewCartActorGRPCServer(pool *CartPool) *cartActorGRPCServer {
|
|
return &cartActorGRPCServer{
|
|
pool: pool,
|
|
}
|
|
}
|
|
|
|
func (s *cartActorGRPCServer) AnnounceOwnership(ctx context.Context, req *messages.OwnershipAnnounce) (*messages.OwnerChangeAck, error) {
|
|
for _, cartId := range req.CartIds {
|
|
s.pool.removeLocalGrain(CartId(cartId))
|
|
}
|
|
log.Printf("Ack count: %d", len(req.CartIds))
|
|
return &messages.OwnerChangeAck{
|
|
Accepted: true,
|
|
Message: "ownership announced",
|
|
}, nil
|
|
}
|
|
|
|
func (s *cartActorGRPCServer) AnnounceExpiry(ctx context.Context, req *messages.ExpiryAnnounce) (*messages.OwnerChangeAck, error) {
|
|
s.pool.HandleRemoteExpiry(req.GetHost(), req.GetCartIds())
|
|
return &messages.OwnerChangeAck{
|
|
Accepted: true,
|
|
Message: "expiry acknowledged",
|
|
}, nil
|
|
}
|
|
|
|
// ControlPlane: Ping
|
|
func (s *cartActorGRPCServer) Ping(ctx context.Context, _ *messages.Empty) (*messages.PingReply, error) {
|
|
// Expose cart owner cookie (first-touch owner = this host) for HTTP gateways translating gRPC metadata.
|
|
// Gateways that propagate Set-Cookie can help establish sticky sessions at the edge.
|
|
//_ = grpc.SendHeader(ctx, metadata.Pairs("set-cookie", fmt.Sprintf("cartowner=%s; Path=/; HttpOnly", s.syncedPool.Hostname())))
|
|
return &messages.PingReply{
|
|
Host: s.pool.Hostname(),
|
|
UnixTime: time.Now().Unix(),
|
|
}, nil
|
|
}
|
|
|
|
// ControlPlane: Negotiate (merge host views)
|
|
func (s *cartActorGRPCServer) Negotiate(ctx context.Context, req *messages.NegotiateRequest) (*messages.NegotiateReply, error) {
|
|
hostSet := make(map[string]struct{})
|
|
// Caller view
|
|
for _, h := range req.GetKnownHosts() {
|
|
if h != "" {
|
|
hostSet[h] = struct{}{}
|
|
}
|
|
}
|
|
// This host
|
|
hostSet[s.pool.Hostname()] = struct{}{}
|
|
// Known remotes
|
|
for _, h := range s.pool.RemoteHostNames() {
|
|
hostSet[h] = struct{}{}
|
|
}
|
|
|
|
out := make([]string, 0, len(hostSet))
|
|
for h := range hostSet {
|
|
out = append(out, h)
|
|
}
|
|
return &messages.NegotiateReply{Hosts: out}, nil
|
|
}
|
|
|
|
// ControlPlane: GetCartIds (locally owned carts only)
|
|
func (s *cartActorGRPCServer) GetCartIds(ctx context.Context, _ *messages.Empty) (*messages.CartIdsReply, error) {
|
|
return &messages.CartIdsReply{CartIds: s.pool.LocalCartIDs()}, nil
|
|
}
|
|
|
|
// ControlPlane: Closing (peer shutdown notification)
|
|
func (s *cartActorGRPCServer) Closing(ctx context.Context, req *messages.ClosingNotice) (*messages.OwnerChangeAck, error) {
|
|
if req.GetHost() != "" {
|
|
s.pool.RemoveHost(req.GetHost())
|
|
}
|
|
return &messages.OwnerChangeAck{
|
|
Accepted: true,
|
|
Message: "removed host",
|
|
}, nil
|
|
}
|
|
|
|
// StartGRPCServer configures and starts the unified gRPC server on the given address.
|
|
// It registers both the CartActor and ControlPlane services.
|
|
func StartGRPCServer(addr string, pool *CartPool) (*grpc.Server, error) {
|
|
lis, err := net.Listen("tcp", addr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to listen: %w", err)
|
|
}
|
|
|
|
grpcServer := grpc.NewServer()
|
|
server := NewCartActorGRPCServer(pool)
|
|
|
|
messages.RegisterControlPlaneServer(grpcServer, server)
|
|
reflection.Register(grpcServer)
|
|
|
|
log.Printf("gRPC server listening on %s", addr)
|
|
go func() {
|
|
if err := grpcServer.Serve(lis); err != nil {
|
|
log.Fatalf("failed to serve gRPC: %v", err)
|
|
}
|
|
}()
|
|
|
|
return grpcServer, nil
|
|
}
|