Update metadata when new staging (#231)

Author: trueleo

server/src/storage.rs

@@ -142,15 +142,20 @@ pub async fn resolve_parseable_metadata() -> Result<(), ObjectStorageError> {
             Some("Could not start the server because metadata not found in storage")
         }
         EnvChange::NewStaging => {
-            put_staging_metadata(remote_metadata.as_ref().expect("remote metadata exists"))?;
+            let mut remote_meta = remote_metadata.expect("remote metadata exists");
+            create_dir_all(CONFIG.staging_dir())?;
+            remote_meta.staging = CONFIG.staging_dir().canonicalize()?;
+            create_remote_metadata(&remote_meta).await?;
+            put_staging_metadata(&remote_meta)?;
 
-            // allow new staging directories
-            return Ok(());
+            None
         }
         EnvChange::CreateBoth => {
             let metadata = StorageMetadata::new();
             create_remote_metadata(&metadata).await?;
-            create_staging_metadata(&metadata)?;
+            create_dir_all(CONFIG.staging_dir())?;
+            put_staging_metadata(&metadata)?;
+
             None
         }
     };
@@ -173,11 +178,6 @@ async fn create_remote_metadata(metadata: &StorageMetadata) -> Result<(), Object
     client.put_metadata(metadata).await
 }
 
-fn create_staging_metadata(metadata: &StorageMetadata) -> std::io::Result<()> {
-    create_dir_all(CONFIG.staging_dir())?;
-    put_staging_metadata(metadata)
-}
-
 lazy_static! {
     pub static ref CACHED_FILES: Mutex<FileTable<FileLink>> = Mutex::new(FileTable::new());
     pub static ref STORAGE_RUNTIME: Arc<RuntimeEnv> = CONFIG.storage().get_datafusion_runtime();