CosmicClash/server/allocator/service.go

// Package allocator coordinates provider allocation with durable control-plane
// state. It does not expose an endpoint until both boundaries succeed.
package allocator

import (
	"context"
	"time"

	"github.com/cosmic-clash/cosmic-clash/server/agones"
	"github.com/cosmic-clash/cosmic-clash/server/domain"
)

type Provider interface {
	Allocate(context.Context, domain.AllocationRequest, map[string]string, time.Time) (agones.AllocatedServer, error)
}

type Durable interface {
	RecordProviderAllocation(context.Context, domain.Allocation, time.Time) (domain.Allocation, error)
}

type RosterPublisher interface {
	PublishRoster(context.Context, domain.Assignment, []domain.SignedJoinAuthorisation, func([]byte, []byte) bool) error
}

type Service struct {
	Provider Provider
	Durable  Durable
	Roster   RosterPublisher
	Now      func() time.Time
}

func (s Service) PublishRoster(ctx context.Context, assignment domain.Assignment, roster []domain.SignedJoinAuthorisation, verify func([]byte, []byte) bool) error {
	if s.Roster == nil {
		return errNotConfigured
	}
	if assignment.Allocation.State != domain.ServerAllocated || assignment.Endpoint == "" {
		return domain.ErrManifestRejected
	}
	return s.Roster.PublishRoster(ctx, assignment, roster, verify)
}

func (s Service) Allocate(ctx context.Context, request domain.AllocationRequest, labels map[string]string) (agones.AllocatedServer, error) {
	if s.Provider == nil || s.Durable == nil || s.Now == nil {
		return agones.AllocatedServer{}, errNotConfigured
	}
	now := s.Now()
	result, err := s.Provider.Allocate(ctx, request, labels, now)
	if err != nil {
		return agones.AllocatedServer{}, err
	}
	if _, err := s.Durable.RecordProviderAllocation(ctx, result.Allocation, now); err != nil {
		return agones.AllocatedServer{}, err
	}
	return result, nil
}

var errNotConfigured = &configurationError{}

type configurationError struct{}

func (*configurationError) Error() string { return "allocator service is not configured" }