feat: add query interface

generator.go

@@ -340,6 +340,17 @@ func (g *Generator) generateQueryFile() (err error) {
 		return err
 	}
 
+	if g.judgeMode(WithQueryInterface) {
+		queryInterfaceTmpl := tmpl.QueryInterfaceWithContext
+		if g.judgeMode(WithoutContext) {
+			queryInterfaceTmpl = tmpl.QueryInterfaceWithoutContext
+		}
+		err = render(queryInterfaceTmpl, &buf, g)
+		if err != nil {
+			return err
+		}
+	}
+
 	err = g.output(g.OutFile, buf.Bytes())
 	if err != nil {
 		return err

internal/template/query.go

@@ -111,6 +111,42 @@ func (q *QueryTx) RollbackTo(name string) error {
 
 `
 
+// QueryInterfaceWithContext interface method template with context
+const QueryInterfaceWithContext = `
+var _ IQuery = (*Query)(nil)
+
+type IQuery interface{
+	{{range $name,$d :=.Data -}}
+	{{$d.ModelStructName}}Do(ctx context.Context) I{{$d.ModelStructName}}Do
+	{{end -}}
+}
+
+{{range $name,$d :=.Data -}}
+func (q *Query) {{$d.ModelStructName}}Do(ctx context.Context) I{{$d.ModelStructName}}Do {
+	return q.{{$d.ModelStructName}}.WithContext(ctx)
+}
+
+{{end -}}
+`
+
+// QueryInterfaceWithoutContext interface method template without context
+const QueryInterfaceWithoutContext = `
+var _ IQuery = (*Query)(nil)
+
+type IQuery interface{
+	{{range $name,$d :=.Data -}}
+	{{$d.ModelStructName}}Do() I{{$d.ModelStructName}}Do
+	{{end -}}
+}
+
+{{range $name,$d :=.Data -}}
+func (q *Query) {{$d.ModelStructName}}Do() I{{$d.ModelStructName}}Do {
+	return &q.{{$d.ModelStructName}}.{{$d.QueryStructName}}Do
+}
+
+{{end -}}
+`
+
 // QueryMethodTest query method test template
 const QueryMethodTest = `
 

tests/go.mod

@@ -4,11 +4,10 @@ go 1.16
 
 require (
 	github.com/mattn/go-sqlite3 v1.14.16 // indirect
-	golang.org/x/tools v0.5.0 // indirect
-	gorm.io/driver/mysql v1.4.5
+	gorm.io/driver/mysql v1.5.1-0.20230509030346-3715c134c25b
 	gorm.io/driver/sqlite v1.4.4
 	gorm.io/gen v0.3.19
-	gorm.io/gorm v1.24.3
+	gorm.io/gorm v1.25.1-0.20230505075827-e61b98d69677
 	gorm.io/hints v1.1.1 // indirect
 	gorm.io/plugin/dbresolver v1.4.0
 )