feat: fix read/write db

internal/template/query.go

@@ -49,11 +49,11 @@ func (q *Query) clone(db *gorm.DB) *Query {
 }
 
 func (q *Query) ReadDB() *Query {
-	return q.ReplaceDB(q.db.Clauses(dbresolver.Read))
+	return q.Clauses(dbresolver.Read)
 }
 
 func (q *Query) WriteDB() *Query {
-	return q.ReplaceDB(q.db.Clauses(dbresolver.Write))
+	return q.Clauses(dbresolver.Write)
 }
 
 func (q *Query) ReplaceDB(db *gorm.DB) *Query {
@@ -65,6 +65,16 @@ func (q *Query) ReplaceDB(db *gorm.DB) *Query {
 	}
 }
 
+
+func (q *Query) Clauses(conds ...clause.Expression) *Query {
+	return &Query{
+		db: q.db.Clauses(conds...),
+		{{range $name,$d :=.Data -}}
+		{{$d.ModelStructName}}: q.{{$d.ModelStructName}}.clauses(conds...),
+		{{end}}
+	}
+}
+
 type queryCtx struct{ 
 	{{range $name,$d :=.Data -}}
 	{{$d.ModelStructName}} {{$d.ReturnObject}}

internal/template/struct.go

@@ -8,7 +8,7 @@ const (
 		{{.QueryStructName}}Do
 		` + fields + `
 	}
-	` + tableMethod + asMethond + updateFieldMethod + getFieldMethod + fillFieldMapMethod + cloneMethod + replaceMethod + relationship + defineMethodStruct
+	` + tableMethod + asMethond + updateFieldMethod + getFieldMethod + fillFieldMapMethod + cloneMethod + replaceMethod + clausesMethod + relationship + defineMethodStruct
 
 	// TableQueryStructWithContext table query struct with context
 	TableQueryStructWithContext = createMethod + `
@@ -27,7 +27,7 @@ const (
 
 	func ({{.S}} {{.QueryStructName}}) Columns(cols ...field.Expr) gen.Columns { return {{.S}}.{{.QueryStructName}}Do.Columns(cols...) }
 
-	` + getFieldMethod + fillFieldMapMethod + cloneMethod + replaceMethod + relationship + defineMethodStruct
+	` + getFieldMethod + fillFieldMapMethod + cloneMethod + replaceMethod + clausesMethod + relationship + defineMethodStruct
 
 	// TableQueryIface table query interface
 	TableQueryIface = defineDoInterface
@@ -116,6 +116,12 @@ func ({{.S}} {{.QueryStructName}}) replaceDB(db *gorm.DB) {{.QueryStructName}} {
 	{{.S}}.{{.QueryStructName}}Do.ReplaceDB(db)
 	return {{.S}}
 }
+`
+	clausesMethod = `
+func ({{.S}} {{.QueryStructName}}) clauses(conds ...clause.Expression) {{.QueryStructName}} {
+	{{.S}}.{{.QueryStructName}}Do.ReplaceDB({{.S}}.{{.QueryStructName}}Do.UnderlyingDB().Clauses(conds...))
+	return {{.S}}
+}
 `
 	getFieldMethod = `
 func ({{.S}} *{{.QueryStructName}}) GetFieldByName(fieldName string) (field.OrderExpr, bool) {