restore: metrics_write at shutdown

Author: jherrera-jump

src/discof/restore/fd_snapdc_tile.c

@@ -147,6 +147,7 @@ handle_control_frag( fd_snapdc_tile_t *  ctx,
     case FD_SNAPSHOT_MSG_CTRL_SHUTDOWN:
       FD_TEST( ctx->state==FD_SNAPDC_STATE_DONE );
       ctx->state = FD_SNAPDC_STATE_SHUTDOWN;
+      metrics_write( ctx ); /* ensures that shutdown state is written to metrics workspace before the tile actually shuts down */
       break;
     default:
       FD_LOG_ERR(( "unexpected control sig %lu", sig ));

src/discof/restore/fd_snapin_tile.c

@@ -250,6 +250,7 @@ handle_control_frag( fd_snapin_tile_t *  ctx,
       break;
     case FD_SNAPSHOT_MSG_CTRL_SHUTDOWN:
       ctx->state = FD_SNAPIN_STATE_SHUTDOWN;
+      metrics_write( ctx ); /* ensures that shutdown state is written to metrics workspace before the tile actually shuts down */
       break;
     default:
       FD_LOG_ERR(( "unexpected control sig %lu", sig ));

src/discof/restore/fd_snaprd_tile.c

@@ -584,6 +584,7 @@ after_credit( fd_snaprd_tile_t *  ctx,
 
       rename_snapshots( ctx );
       ctx->state = FD_SNAPRD_STATE_SHUTDOWN;
+      metrics_write( ctx ); /* ensures that shutdown state is written to metrics workspace before the tile actually shuts down */
       fd_stem_publish( stem, 0UL, FD_SNAPSHOT_MSG_CTRL_SHUTDOWN, 0UL, 0UL, 0UL, 0UL, 0UL );
       break;
     case FD_SNAPRD_STATE_FLUSHING_FULL_FILE:
@@ -592,6 +593,7 @@ after_credit( fd_snaprd_tile_t *  ctx,
 
       if( FD_LIKELY( !ctx->config.incremental_snapshot_fetch ) ) {
         ctx->state = FD_SNAPRD_STATE_SHUTDOWN;
+        metrics_write( ctx ); /* ensures that shutdown state is written to metrics workspace before the tile actually shuts down */
         fd_stem_publish( stem, 0UL, FD_SNAPSHOT_MSG_CTRL_SHUTDOWN, 0UL, 0UL, 0UL, 0UL, 0UL );
         break;
       }
@@ -616,6 +618,7 @@ after_credit( fd_snaprd_tile_t *  ctx,
       if( FD_LIKELY( !ctx->config.incremental_snapshot_fetch ) ) {
         rename_snapshots( ctx );
         ctx->state = FD_SNAPRD_STATE_SHUTDOWN;
+        metrics_write( ctx ); /* ensures that shutdown state is written to metrics workspace before the tile actually shuts down */
         fd_stem_publish( stem, 0UL, FD_SNAPSHOT_MSG_CTRL_SHUTDOWN, 0UL, 0UL, 0UL, 0UL, 0UL );
         break;
       }