add on_conflict query arg in insert_one

Author: imperfect-fourth

cmd/eywagen/eywatest/eywa_generated.go

@@ -38,6 +38,7 @@ func testTable_AgeVar(val *int) eywa.Field[testTable] {
 		Value: eywa.QueryVar("testTable_Age", eywa.NullableIntVar[*int](val)),
 	}
 }
+const testTable_PkeyConstraint eywa.Constraint[testTable] = "testTable_pkey"
 const testTable_ID eywa.FieldName[testTable] = "id"
 
 func testTable_IDField(val int) eywa.Field[testTable] {
@@ -53,19 +54,19 @@ func testTable_IDVar(val int) eywa.Field[testTable] {
 		Value: eywa.QueryVar("testTable_ID", eywa.IntVar[int](val)),
 	}
 }
-const testTable_iD eywa.FieldName[testTable] = "idd"
+const testTable_IDd eywa.FieldName[testTable] = "idd"
 
-func testTable_iDField(val int32) eywa.Field[testTable] {
+func testTable_IDdField(val int32) eywa.Field[testTable] {
 	return eywa.Field[testTable]{
 		Name: "idd",
 		Value: val,
 	}
 }
 
-func testTable_iDVar(val int32) eywa.Field[testTable] {
+func testTable_IDdVar(val int32) eywa.Field[testTable] {
 	return eywa.Field[testTable]{
 		Name: "idd",
-		Value: eywa.QueryVar("testTable_iD", eywa.IntVar[int32](val)),
+		Value: eywa.QueryVar("testTable_IDd", eywa.IntVar[int32](val)),
 	}
 }
 const testTable_custom eywa.FieldName[testTable] = "custom"
@@ -155,6 +156,7 @@ func testTable_FVar(val X[string, int]) eywa.Field[testTable] {
 	}
 }
 
+const testTable2_PkeyConstraint eywa.Constraint[testTable2] = "testTable2_pkey"
 const testTable2_ID eywa.FieldName[testTable2] = "id"
 
 func testTable2_IDField(val uuid.UUID) eywa.Field[testTable2] {
@@ -163,3 +165,18 @@ func testTable2_IDField(val uuid.UUID) eywa.Field[testTable2] {
 		Value: val,
 	}
 }
+const testTable2_Age eywa.FieldName[testTable2] = "age"
+
+func testTable2_AgeField(val int) eywa.Field[testTable2] {
+	return eywa.Field[testTable2]{
+		Name: "age",
+		Value: val,
+	}
+}
+
+func testTable2_AgeVar(val int) eywa.Field[testTable2] {
+	return eywa.Field[testTable2]{
+		Name: "age",
+		Value: eywa.QueryVar("testTable2_Age", eywa.IntVar[int](val)),
+	}
+}

cmd/eywagen/eywatest/eywa_test.go

@@ -154,3 +154,43 @@ id
 		assert.Equal(t, &testTable2{ID: id}, resp)
 	}
 }
+
+func TestInsertOneQueryOnConflict(t *testing.T) {
+	id := uuid.New()
+	accessKey := os.Getenv("TEST_HGE_ACCESS_KEY")
+	c := eywa.NewClient("https://aware-cowbird-80.hasura.app/v1/graphql", &eywa.ClientOpts{
+		Headers: map[string]string{
+			"x-hasura-access-key": accessKey,
+		},
+	})
+	q := eywa.InsertOne(
+		testTable2_IDField(id),
+		testTable2_AgeField(10),
+	).Select(
+		testTable2_ID,
+	)
+	q.Exec(c)
+
+	q2 := eywa.InsertOne(
+		testTable2_IDField(id),
+		testTable2_AgeField(20),
+	).OnConflict(
+		testTable2_PkeyConstraint,
+		testTable2_Age,
+	).Select(
+		testTable2_ID,
+		testTable2_Age,
+	)
+	expected := fmt.Sprintf(`mutation insert_test_table2_one {
+insert_test_table2_one(object: {age: 20, id: "%s"}, on_conflict: {constraint: testTable2_pkey, update_columns: [age]}) {
+age
+id
+}
+}`, id.String())
+
+	if assert.Equal(t, expected, q2.Query()) {
+		resp, err := q2.Exec(c)
+		assert.NoError(t, err)
+		assert.Equal(t, &testTable2{ID: id, Age: 20}, resp)
+	}
+}

cmd/eywagen/eywatest/eywatest.go

@@ -9,7 +9,7 @@ import (
 type testTable struct {
 	Name       string            `json:"name"`
 	Age        *int              `json:"age"`
-	ID         int               `json:"id,omitempty"`
+	ID         int               `json:"id,omitempty",eywa:"pkey"`
 	IDd        int32             `json:"idd,omitempty"`
 	custom     *customType       `json:"custom"`
 	testTable2 *testTable2       `json:"test_table2"`
@@ -36,7 +36,8 @@ func (t testTable) ModelName() string {
 type customType struct{}
 
 type testTable2 struct {
-	ID uuid.UUID `json:"id"`
+	ID  uuid.UUID `json:"id",eywa:"pkey"`
+	Age int       `json:"age"`
 }
 
 func (t testTable2) ModelName() string {

cmd/eywagen/main.go

@@ -23,6 +23,7 @@ func usage() {
 }
 
 var tagPattern = re.MustCompile(`json:"([^"]+)"`)
+var eywaTagPattern = re.MustCompile(`eywa:"([^"]+)"`)
 
 const (
 	genHeader           = "// generated by eywa. DO NOT EDIT. Any changes will be overwritten.\npackage "
@@ -66,6 +67,10 @@ func %s(subField eywa.FieldName[%s], subFields ...eywa.FieldName[%s]) eywa.Field
 `
 )
 
+func pkeyConstraint(typeName string) string {
+	return fmt.Sprintf("const %s_PkeyConstraint eywa.Constraint[%s] = \"%s_pkey\"\n", typeName, typeName, typeName)
+}
+
 func main() {
 	flag.Usage = usage
 	flag.Parse()
@@ -91,7 +96,9 @@ func main() {
 		content:    bytes.NewBufferString(""),
 	}
 	for _, t := range types {
-		parseType(t, pkg, contents)
+		if err := parseType(t, pkg, contents); err != nil {
+			panic(err)
+		}
 	}
 	if len(contents.importsMap) > 0 {
 		contents.imports.WriteString("\nimport (\n")
@@ -117,39 +124,55 @@ type fileContent struct {
 
 var parsed = make(map[string]bool)
 
-func parseType(typeName string, pkg *types.Package, contents *fileContent) {
+func parseType(typeName string, pkg *types.Package, contents *fileContent) error {
 	if parsed[typeName] {
-		return
+		return nil
 	}
 	parsed[typeName] = true
 
 	typeObj := pkg.Scope().Lookup(typeName)
 	if typeObj == nil {
 		fmt.Printf("type %s not found in package, skipping...", typeName)
-		return
+		return nil
 	}
 	typeStruct, ok := typeObj.Type().Underlying().(*types.Struct)
 	if !ok {
 		fmt.Printf("type %s is not a struct, skipping...", typeName)
-		return
+		return nil
 	}
 	if types.NewMethodSet(types.NewPointer(typeObj.Type())).Lookup(pkg, "ModelName") == nil {
 		fmt.Printf("struct type %s does not implement eywa.Model interface, skipping...", typeName)
-		return
+		return nil
 	}
 
 	contents.content.WriteString("\n")
 	recurseParse := make([]string, 0, typeStruct.NumFields())
+	foundPkey := false
+	pkey := ""
 	for i := 0; i < typeStruct.NumFields(); i++ {
 		tag := tagPattern.FindStringSubmatch(typeStruct.Tag(i))
 		if tag == nil {
 			continue
 		}
-		tagValue := strings.Split(tag[1], ",")
-		if len(tagValue) == 0 {
+		tagValues := strings.Split(tag[1], ",")
+		if len(tagValues) == 0 {
 			continue
 		}
-		fieldName := tagValue[0]
+		fieldName := tagValues[0]
+		if eywaTag := eywaTagPattern.FindStringSubmatch(typeStruct.Tag(i)); eywaTag != nil {
+			if eywaTagValues := strings.Split(eywaTag[1], ","); len(eywaTagValues) != 0 {
+				for _, v := range eywaTagValues {
+					if v == "pkey" && foundPkey {
+						return fmt.Errorf("model %s has two primary keys: %s, %s", typeName, pkey, fieldName)
+					}
+					if v == "pkey" {
+						foundPkey = true
+						contents.content.WriteString(pkeyConstraint(typeName))
+						pkey = fieldName
+					}
+				}
+			}
+		}
 		field := typeStruct.Field(i)
 		fieldType := field.Type()
 		importPackages, fieldTypeNameFull := parseFieldTypeName(field.Type().String(), pkg.Path())
@@ -278,8 +301,11 @@ func parseType(typeName string, pkg *types.Package, contents *fileContent) {
 		}
 	}
 	for _, t := range recurseParse {
-		parseType(t, pkg, contents)
+		if err := parseType(t, pkg, contents); err != nil {
+			return err
+		}
 	}
+	return nil
 }
 
 func writeToFile(filename string, contents *fileContent) error {

eywa.go

@@ -104,3 +104,5 @@ type QuerySkeleton[M Model] struct {
 func (qs QuerySkeleton[M]) MarshalGQL() string {
 	return fmt.Sprintf("%s%s", qs.ModelName, qs.queryArgs.MarshalGQL())
 }
+
+type Constraint[M Model] string

insert_one.go

@@ -23,12 +23,13 @@ type InsertOneQueryBuilder[M Model] struct {
 	QuerySkeleton[M]
 }
 
-// func (iq InsertOneQueryBuilder[M]) OnConstraint(constraint eywa.Constraint[M], field FieldName[M], fields ...FieldName[M]) InsertOneQuery[M] {
-// 	return InsertOneQuery[M]{
-// 		iq:     &iq,
-// 		fields: append(fields, field),
-// 	}
-// }
+func (iq InsertOneQueryBuilder[M]) OnConflict(constraint Constraint[M], fields ...FieldName[M]) InsertOneQueryBuilder[M] {
+	iq.QuerySkeleton.queryArgs.onConflict = &onConflict[M]{
+		constraint:    constraint,
+		updateColumns: fields,
+	}
+	return iq
+}
 
 func (iq *InsertOneQueryBuilder[M]) MarshalGQL() string {
 	return fmt.Sprintf(

query_arg.go

@@ -14,6 +14,7 @@ type queryArgs[M Model] struct {
 	orderBy    *orderBy
 	set        *set[M]
 	object     *object[M]
+	onConflict *onConflict[M]
 }
 
 func (qa queryArgs[M]) MarshalGQL() string {
@@ -25,6 +26,7 @@ func (qa queryArgs[M]) MarshalGQL() string {
 	args = appendArg(args, qa.orderBy)
 	args = appendArg(args, qa.set)
 	args = appendArg(args, qa.object)
+	args = appendArg(args, qa.onConflict)
 
 	return fmt.Sprintf("(%s)", strings.Join(args, ", "))
 }
@@ -268,3 +270,19 @@ func (o object[M]) queryArgName() string {
 func (o object[M]) MarshalGQL() string {
 	return fmt.Sprintf("%s: {%s}", o.queryArgName(), o.fields.MarshalGQL())
 }
+
+type onConflict[M Model] struct {
+	constraint    Constraint[M]
+	updateColumns FieldNameArray[M]
+}
+
+func (oc onConflict[M]) queryArgName() string {
+	return "on_conflict"
+}
+
+func (oc onConflict[M]) MarshalGQL() string {
+	if oc.updateColumns == nil {
+		return fmt.Sprintf("%s: {constraint: %s}", oc.queryArgName(), string(oc.constraint))
+	}
+	return fmt.Sprintf("%s: {constraint: %s, update_columns: [%s]}", oc.queryArgName(), string(oc.constraint), oc.updateColumns.MarshalGQL())
+}