@@ -229,17 +229,21 @@ export class UpdateMetadataService extends BaseService {
229
229
}
230
230
231
231
public async call (
232
- environment : AuthenticatedEnvironment ,
233
232
runId : string ,
234
- body : UpdateMetadataRequestBody
233
+ body : UpdateMetadataRequestBody ,
234
+ environment ?: AuthenticatedEnvironment
235
235
) {
236
236
const runIdType = runId . startsWith ( "run_" ) ? "friendly" : "internal" ;
237
237
238
238
const taskRun = await this . _prisma . taskRun . findFirst ( {
239
- where : {
240
- runtimeEnvironmentId : environment . id ,
241
- ...( runIdType === "internal" ? { id : runId } : { friendlyId : runId } ) ,
242
- } ,
239
+ where : environment
240
+ ? {
241
+ runtimeEnvironmentId : environment . id ,
242
+ ...( runIdType === "internal" ? { id : runId } : { friendlyId : runId } ) ,
243
+ }
244
+ : {
245
+ ...( runIdType === "internal" ? { id : runId } : { friendlyId : runId } ) ,
246
+ } ,
243
247
select : {
244
248
id : true ,
245
249
status : true ,
@@ -351,6 +355,15 @@ export class UpdateMetadataService extends BaseService {
351
355
} ) ;
352
356
353
357
if ( result . count === 0 ) {
358
+ if ( this . flushLoggingEnabled ) {
359
+ logger . debug (
360
+ `[UpdateMetadataService][updateRunMetadataWithOperations] Optimistic lock failed for run ${ runId } ` ,
361
+ {
362
+ metadataVersion : run . metadataVersion ,
363
+ }
364
+ ) ;
365
+ }
366
+
354
367
// If this was our last attempt, buffer the operations and return optimistically
355
368
if ( attempts === MAX_RETRIES ) {
356
369
this . #ingestRunOperations( runId , operations ) ;
@@ -363,6 +376,15 @@ export class UpdateMetadataService extends BaseService {
363
376
continue ;
364
377
}
365
378
379
+ if ( this . flushLoggingEnabled ) {
380
+ logger . debug (
381
+ `[UpdateMetadataService][updateRunMetadataWithOperations] Updated metadata for run ${ runId } ` ,
382
+ {
383
+ metadata : applyResults . newMetadata ,
384
+ }
385
+ ) ;
386
+ }
387
+
366
388
// Success! Return the new metadata
367
389
return applyResults . newMetadata ;
368
390
}
@@ -383,10 +405,15 @@ export class UpdateMetadataService extends BaseService {
383
405
metadataPacket . data !== "{}" ||
384
406
( existingMetadata . data && metadataPacket . data !== existingMetadata . data )
385
407
) {
386
- logger . debug ( `Updating metadata directly for run` , {
387
- metadata : metadataPacket . data ,
388
- runId,
389
- } ) ;
408
+ if ( this . flushLoggingEnabled ) {
409
+ logger . debug (
410
+ `[UpdateMetadataService][updateRunMetadataDirectly] Updating metadata directly for run` ,
411
+ {
412
+ metadata : metadataPacket . data ,
413
+ runId,
414
+ }
415
+ ) ;
416
+ }
390
417
391
418
// Update the metadata without version check
392
419
await this . _prisma . taskRun . update ( {
@@ -416,6 +443,13 @@ export class UpdateMetadataService extends BaseService {
416
443
} ;
417
444
} ) ;
418
445
446
+ if ( this . flushLoggingEnabled ) {
447
+ logger . debug ( `[UpdateMetadataService] Ingesting operations for run` , {
448
+ runId,
449
+ bufferedOperations,
450
+ } ) ;
451
+ }
452
+
419
453
const existingBufferedOperations = this . _bufferedOperations . get ( runId ) ?? [ ] ;
420
454
421
455
this . _bufferedOperations . set ( runId , [ ...existingBufferedOperations , ...bufferedOperations ] ) ;
0 commit comments