feat: integrate creator hub douyin workflows

internal/creator/source_lease.go

@@ -0,0 +1,44 @@
+package creator
+
+import (
+	"context"
+	"database/sql"
+)
+
+func (s *Store) ClaimSourceSync(ctx context.Context, sourceType, sourceID string) (string, error) {
+	if sourceType != SourceOwned && sourceType != SourceCompetitor || sourceID == "" {
+		return "", ErrInvalid
+	}
+	token := newID("source-lease")
+	var claimed string
+	err := s.db.QueryRowContext(ctx, `
+		INSERT INTO creator_source_sync_lease (source_type,source_id,lease_token,lease_until,updated_at)
+		VALUES ($1,$2,$3,now() + interval '10 minutes',now())
+		ON CONFLICT (source_type,source_id) DO UPDATE SET lease_token=EXCLUDED.lease_token,
+			lease_until=EXCLUDED.lease_until,updated_at=now()
+		WHERE creator_source_sync_lease.lease_until <= now()
+		RETURNING lease_token`, sourceType, sourceID, token).Scan(&claimed)
+	if err == sql.ErrNoRows {
+		return "", ErrConflict
+	}
+	if err != nil {
+		return "", databaseError(err)
+	}
+	return claimed, nil
+}
+
+func (s *Store) ReleaseSourceSync(ctx context.Context, sourceType, sourceID, token string) error {
+	if sourceType != SourceOwned && sourceType != SourceCompetitor || sourceID == "" || token == "" {
+		return ErrInvalid
+	}
+	result, err := s.db.ExecContext(ctx, `DELETE FROM creator_source_sync_lease WHERE source_type=$1 AND source_id=$2 AND lease_token=$3`, sourceType, sourceID, token)
+	if err != nil {
+		return databaseError(err)
+	}
+	if rows, err := result.RowsAffected(); err != nil {
+		return err
+	} else if rows != 1 {
+		return ErrConflict
+	}
+	return nil
+}