fix migration

reearth · Mar 6, 2024 · 5767ed6 · 5767ed6
1 parent 7734417
commit 5767ed6
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/server/cmd/db-migrations/ref_field_schema.go b/server/cmd/db-migrations/ref_field_schema.go
@@ -26,8 +26,9 @@ type TypeProperty struct {
 }
 
 type Reference struct {
-	Model  string `bson:"model"`
-	Schema string `bson:"schema"`
+	Model               string  `bson:"model"`
+	Schema              string  `bson:"schema"`
+	CorrespondingSchema *string `bson:"correspondingschema"`
 }
 
 func (r *Reference) SetSchema(s string) {
@@ -86,10 +87,15 @@ func RefFieldSchema(ctx context.Context, dbURL, dbName string, wetRun bool) erro
 				return
 			}
 			m, ok := models[f.TypeProperty.Reference.Model]
-			if !ok {
-				fmt.Printf("no model found for schema '%s' model id '%s'\n", s.ID, f.TypeProperty.Reference.Model)
+			if ok {
+				f.TypeProperty.Reference.SetSchema(m.Schema)
+				return
+			}
+			if f.TypeProperty.Reference.CorrespondingSchema != nil {
+				f.TypeProperty.Reference.SetSchema(*f.TypeProperty.Reference.CorrespondingSchema)
+				return
 			}
-			f.TypeProperty.Reference.SetSchema(m.Schema)
+			fmt.Printf("no model found for schema '%s' model id '%s'\n", s.ID, f.TypeProperty.Reference.Model)
 		})
 	})
 
@@ -99,6 +105,7 @@ func RefFieldSchema(ctx context.Context, dbURL, dbName string, wetRun bool) erro
 			if f.TypeProperty.Type != "reference" {
 				return nil, false
 			}
+			fmt.Printf("updating schema '%s' field '%s' referenced schema '%s'\n", s.ID, f.ID, f.TypeProperty.Reference.Schema)
 			return mongo.NewUpdateOneModel().
 				SetFilter(bson.M{
 					"id":        s.ID,