From 420a53efdaf03d8ccb0393eb524a06b3dddbf9ea Mon Sep 17 00:00:00 2001
From: Rohith BCS <rohith.bcs@gmail.com>
Date: Wed, 11 Dec 2024 17:26:13 +0530
Subject: [PATCH] chore: refactor 4

---
 warehouse/router/upload.go | 6 +++++-
 warehouse/schema/schema.go | 3 +++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/warehouse/router/upload.go b/warehouse/router/upload.go
index 3e1ef3fc45..2b255759b8 100644
--- 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
index 8cb06e1a0e..c6f885b22c 100644
--- 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
 }