fix RelatedObjectID suffix assignement on doc.Set

sourcenetwork · Jan 3, 2024 · 39b7b30 · 39b7b30
1 parent ea2e8e1
commit 39b7b30
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/client/document.go b/client/document.go
@@ -465,16 +465,12 @@ func (doc *Document) Set(field string, value any) error {
 		return NewErrFieldNotExist(field)
 	}
 	if fd.IsPrimaryRelation() {
-		if strings.HasSuffix(field, request.RelatedObjectID) {
-			fd, exists = doc.schemaDescription.GetField(field)
-			if !exists {
-				return NewErrFieldNotExist(field)
-			}
-		} else {
-			fd, exists = doc.schemaDescription.GetField(field + request.RelatedObjectID)
-			if !exists {
-				return NewErrFieldNotExist(field)
-			}
+		if !strings.HasSuffix(field, request.RelatedObjectID) {
+			field = field + request.RelatedObjectID
+		}
+		fd, exists = doc.schemaDescription.GetField(field)
+		if !exists {
+			return NewErrFieldNotExist(field)
 		}
 	}
 	val, err := validateFieldSchema(value, fd)