major refactor

grpc_server.go

@@ -9,32 +9,42 @@ import (
 
 	messages "git.tornberg.me/go-cart-actor/proto"
 	"google.golang.org/grpc"
-	"google.golang.org/grpc/metadata"
 	"google.golang.org/grpc/reflection"
 )
 
-// cartActorGRPCServer implements the CartActor and ControlPlane gRPC services.
+// 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       GrainPool   // For cart state mutations and queries
+	//pool       GrainPool   // For cart state mutations and queries
 	syncedPool *SyncedPool // For cluster membership and control
 }
 
 // NewCartActorGRPCServer creates and initializes the server.
-func NewCartActorGRPCServer(pool GrainPool, syncedPool *SyncedPool) *cartActorGRPCServer {
+func NewCartActorGRPCServer(syncedPool *SyncedPool) *cartActorGRPCServer {
 	return &cartActorGRPCServer{
-		pool:       pool,
+		//pool:       pool,
 		syncedPool: syncedPool,
 	}
 }
 
+func (s *cartActorGRPCServer) AnnounceOwnership(ctx context.Context, req *messages.OwnershipAnnounce) (*messages.OwnerChangeAck, error) {
+	for _, cartId := range req.CartIds {
+		s.syncedPool.removeLocalGrain(CartId(cartId))
+	}
+	log.Printf("Ack count: %d", len(req.CartIds))
+	return &messages.OwnerChangeAck{
+		Accepted: true,
+		Message:  "ownership announced",
+	}, 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())))
+	//_ = grpc.SendHeader(ctx, metadata.Pairs("set-cookie", fmt.Sprintf("cartowner=%s; Path=/; HttpOnly", s.syncedPool.Hostname())))
 	return &messages.PingReply{
 		Host:     s.syncedPool.Hostname(),
 		UnixTime: time.Now().Unix(),
@@ -93,14 +103,14 @@ func (s *cartActorGRPCServer) Closing(ctx context.Context, req *messages.Closing
 
 // 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 GrainPool, syncedPool *SyncedPool) (*grpc.Server, error) {
+func StartGRPCServer(addr string, syncedPool *SyncedPool) (*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, syncedPool)
+	server := NewCartActorGRPCServer(syncedPool)
 
 	messages.RegisterControlPlaneServer(grpcServer, server)
 	reflection.Register(grpcServer)