| 22 | } |
| 23 | |
| 24 | func (p PubsubWorkspaceClaimPublisher) PublishWorkspaceClaim(claim agentsdk.ReinitializationEvent) error { |
| 25 | channel := agentsdk.PrebuildClaimedChannel(claim.WorkspaceID) |
| 26 | payload, err := json.Marshal(claim) |
| 27 | if err != nil { |
| 28 | return xerrors.Errorf("marshal claim event: %w", err) |
| 29 | } |
| 30 | if err := p.ps.Publish(channel, payload); err != nil { |
| 31 | return xerrors.Errorf("failed to trigger prebuilt workspace agent reinitialization: %w", err) |
| 32 | } |
| 33 | return nil |
| 34 | } |
| 35 | |
| 36 | func NewPubsubWorkspaceClaimListener(ps pubsub.Pubsub, logger slog.Logger) *PubsubWorkspaceClaimListener { |
| 37 | return &PubsubWorkspaceClaimListener{ps: ps, logger: logger} |