fix: set foreign key reference automatically for belongs to associations

sunecko · sunecko · commit 71781b1a303f · 2024-10-28T12:00:59.000-04:00
diff --git a/repository.go b/repository.go
@@ -613,9 +613,14 @@ func (r repository) saveBelongsTo(cw contextWrapper, doc *Document, mutation *Mu
 
 			var (
 				rField = assoc.ReferenceField()
+				rValue = assoc.ReferenceValue()
 				fValue = assoc.ForeignValue()
 			)
 
+			if rValue == 0 {
+				rValue = fValue
+			}
+
 			mutation.Add(Set(rField, fValue))
 			doc.SetValue(rField, fValue)
 		}

Original file line number	Diff line number	Diff line change
`@@ -613,9 +613,14 @@ func (r repository) saveBelongsTo(cw contextWrapper, doc Document, mutation Mu`
`613`	`613`
`614`	`614`	`var (`
`615`	`615`	`rField = assoc.ReferenceField()`
	`616`	`+ rValue = assoc.ReferenceValue()`
`616`	`617`	`fValue = assoc.ForeignValue()`
`617`	`618`	`)`
`618`	`619`
	`620`	`+ if rValue == 0 {`
	`621`	`+ rValue = fValue`
	`622`	`+ }`
	`623`	`+`
`619`	`624`	`mutation.Add(Set(rField, fValue))`
`620`	`625`	`doc.SetValue(rField, fValue)`
`621`	`626`	`}`