chore: refactor 4

rudderlabs · Dec 11, 2024 · 420a53e · 420a53e
1 parent 1161fbb
commit 420a53e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/warehouse/router/upload.go b/warehouse/router/upload.go
@@ -289,7 +289,11 @@ func (job *UploadJob) run() (err error) {
 		return err
 	}
 	defer whManager.Cleanup(job.ctx)
-
+	_, err = job.schemaHandle.GetLocalSchema(job.ctx)
+	if err != nil {
+		_, _ = job.setUploadError(err, InternalProcessingFailed)
+		return err
+	}
 	var (
 		newStatus       string
 		nextUploadState *state

diff --git a/warehouse/schema/schema.go b/warehouse/schema/schema.go
@@ -301,6 +301,9 @@ func (sh *Schema) GetLocalSchema(ctx context.Context) (model.Schema, error) {
 	if whSchema.Schema == nil {
 		return model.Schema{}, nil
 	}
+	sh.localSchemaMu.Lock()
+	sh.localSchema = whSchema.Schema
+	sh.localSchemaMu.Unlock()
 	return whSchema.Schema, nil
 }