Method UpsertSubscription

backend/store/subscription.go:26–49

UpsertSubscription creates or updates a subscription by workspace.

(ctx context.Context, workspace string, payload *storepb.SubscriptionPayload)

// UpsertSubscription creates or updates a subscription by workspace.
func (s *Store) UpsertSubscription(ctx context.Context, workspace string, payload *storepb.SubscriptionPayload) (*SubscriptionMessage, error) {
	p, err := protojson.Marshal(payload)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to marshal payload")
	}

	sub := &SubscriptionMessage{
		Payload: &storepb.SubscriptionPayload{},
	}
	var payloadBytes []byte
	if err := s.GetDB().QueryRowContext(ctx,
		`INSERT INTO subscription (workspace, payload) VALUES ($1, $2)
		 ON CONFLICT (workspace) DO UPDATE SET payload = $2, updated_at = now()
		 RETURNING workspace, payload, created_at, updated_at`,
		workspace, p,
	).Scan(&sub.Workspace, &payloadBytes, &sub.CreatedAt, &sub.UpdatedAt); err != nil {
		return nil, errors.Wrapf(err, "failed to upsert subscription")
	}
	if err := common.ProtojsonUnmarshaler.Unmarshal(payloadBytes, sub.Payload); err != nil {
		return nil, errors.Wrapf(err, "failed to unmarshal payload")
	}
	sub.Etag = fmt.Sprintf("%d", sub.UpdatedAt.UnixMilli())
	return sub, nil
}

// GetSubscriptionByWorkspace gets a subscription by workspace.
// Returns (nil, nil) if not found.