Implement backfill retry logic with CORS

Add retry mechanism to backfill-ride-data, backfill-company-data, and backfill-park-locations edge functions using withEdgeRetry, including isRetryableError and isDeadlockError. Apply CORS headers to ride-data and company-data responses, and ensure park-locations already contains CORS remains with retry wrapping. Update unauthorized and success responses to include CORS headers.
This commit is contained in:
gpt-engineer-app[bot]
2025-11-11 18:12:48 +00:00
parent 10df39c7d4
commit 0e9ea18be8
3 changed files with 65 additions and 34 deletions

View File

@@ -1,10 +1,13 @@
import { createEdgeFunction } from '../_shared/edgeFunctionWrapper.ts'; import { createEdgeFunction } from '../_shared/edgeFunctionWrapper.ts';
import { edgeLogger } from '../_shared/logger.ts'; import { edgeLogger } from '../_shared/logger.ts';
import { withEdgeRetry, isRetryableError, isDeadlockError } from '../_shared/retryHelper.ts';
import { corsHeadersWithMethods } from '../_shared/cors.ts';
export default createEdgeFunction( export default createEdgeFunction(
{ {
name: 'backfill-company-data', name: 'backfill-company-data',
requireAuth: true, requireAuth: true,
corsHeaders: corsHeadersWithMethods,
}, },
async (req, context, supabase) => { async (req, context, supabase) => {
edgeLogger.info('Starting company data backfill', { requestId: context.requestId }); edgeLogger.info('Starting company data backfill', { requestId: context.requestId });
@@ -23,20 +26,28 @@ export default createEdgeFunction(
}); });
return new Response( return new Response(
JSON.stringify({ error: 'Unauthorized: Superuser access required' }), JSON.stringify({ error: 'Unauthorized: Superuser access required' }),
{ status: 403, headers: { 'Content-Type': 'application/json' } } { status: 403, headers: { ...corsHeadersWithMethods, 'Content-Type': 'application/json' } }
); );
} }
// Execute the backfill function // Execute the backfill function with retry logic
const { data, error } = await supabase.rpc('backfill_company_data'); const { data, error } = await withEdgeRetry(
async () => {
if (error) { const result = await supabase.rpc('backfill_company_data');
edgeLogger.error('Error running company data backfill', { if (result.error) throw result.error;
error, return result;
requestId: context.requestId },
}); {
throw error; maxAttempts: 5,
} baseDelay: 2000,
maxDelay: 30000,
backoffMultiplier: 2,
jitter: true,
shouldRetry: (error) => isRetryableError(error) || isDeadlockError(error)
},
context.requestId,
'backfill_company_data'
);
edgeLogger.info('Company data backfill completed', { edgeLogger.info('Company data backfill completed', {
results: data, results: data,
@@ -48,7 +59,7 @@ export default createEdgeFunction(
success: true, success: true,
...data, ...data,
}), }),
{ headers: { 'Content-Type': 'application/json' } } { headers: { ...corsHeadersWithMethods, 'Content-Type': 'application/json' } }
); );
} }
); );

View File

@@ -1,6 +1,7 @@
import { createEdgeFunction } from '../_shared/edgeFunctionWrapper.ts'; import { createEdgeFunction } from '../_shared/edgeFunctionWrapper.ts';
import { edgeLogger } from '../_shared/logger.ts'; import { edgeLogger } from '../_shared/logger.ts';
import { corsHeadersWithMethods } from '../_shared/cors.ts'; import { corsHeadersWithMethods } from '../_shared/cors.ts';
import { withEdgeRetry, isRetryableError, isDeadlockError } from '../_shared/retryHelper.ts';
export default createEdgeFunction( export default createEdgeFunction(
{ {
@@ -29,16 +30,24 @@ export default createEdgeFunction(
); );
} }
// Execute the backfill function // Execute the backfill function with retry logic
const { data, error } = await supabase.rpc('backfill_park_locations'); const { data, error } = await withEdgeRetry(
async () => {
if (error) { const result = await supabase.rpc('backfill_park_locations');
edgeLogger.error('Error running park location backfill', { if (result.error) throw result.error;
error, return result;
requestId: context.requestId },
}); {
throw error; maxAttempts: 5,
} baseDelay: 2000,
maxDelay: 30000,
backoffMultiplier: 2,
jitter: true,
shouldRetry: (error) => isRetryableError(error) || isDeadlockError(error)
},
context.requestId,
'backfill_park_locations'
);
edgeLogger.info('Park location backfill completed', { edgeLogger.info('Park location backfill completed', {
results: data, results: data,

View File

@@ -1,10 +1,13 @@
import { createEdgeFunction } from '../_shared/edgeFunctionWrapper.ts'; import { createEdgeFunction } from '../_shared/edgeFunctionWrapper.ts';
import { edgeLogger } from '../_shared/logger.ts'; import { edgeLogger } from '../_shared/logger.ts';
import { withEdgeRetry, isRetryableError, isDeadlockError } from '../_shared/retryHelper.ts';
import { corsHeadersWithMethods } from '../_shared/cors.ts';
export default createEdgeFunction( export default createEdgeFunction(
{ {
name: 'backfill-ride-data', name: 'backfill-ride-data',
requireAuth: true, requireAuth: true,
corsHeaders: corsHeadersWithMethods,
}, },
async (req, context, supabase) => { async (req, context, supabase) => {
edgeLogger.info('Starting ride data backfill', { requestId: context.requestId }); edgeLogger.info('Starting ride data backfill', { requestId: context.requestId });
@@ -23,20 +26,28 @@ export default createEdgeFunction(
}); });
return new Response( return new Response(
JSON.stringify({ error: 'Unauthorized: Superuser access required' }), JSON.stringify({ error: 'Unauthorized: Superuser access required' }),
{ status: 403, headers: { 'Content-Type': 'application/json' } } { status: 403, headers: { ...corsHeadersWithMethods, 'Content-Type': 'application/json' } }
); );
} }
// Execute the backfill function // Execute the backfill function with retry logic
const { data, error } = await supabase.rpc('backfill_ride_data'); const { data, error } = await withEdgeRetry(
async () => {
if (error) { const result = await supabase.rpc('backfill_ride_data');
edgeLogger.error('Error running ride data backfill', { if (result.error) throw result.error;
error, return result;
requestId: context.requestId },
}); {
throw error; maxAttempts: 5,
} baseDelay: 2000,
maxDelay: 30000,
backoffMultiplier: 2,
jitter: true,
shouldRetry: (error) => isRetryableError(error) || isDeadlockError(error)
},
context.requestId,
'backfill_ride_data'
);
edgeLogger.info('Ride data backfill completed', { edgeLogger.info('Ride data backfill completed', {
results: data, results: data,
@@ -48,7 +59,7 @@ export default createEdgeFunction(
success: true, success: true,
...data, ...data,
}), }),
{ headers: { 'Content-Type': 'application/json' } } { headers: { ...corsHeadersWithMethods, 'Content-Type': 'application/json' } }
); );
} }
); );