feat: interface template add method Rows

go-gorm · Dec 19, 2024 · ba97558 · ba97558
1 parent a31c34b
commit ba97558
Show file tree

Hide file tree

Showing 13 changed files with 41 additions and 4 deletions.
diff --git a/internal/template/struct.go b/internal/template/struct.go
@@ -201,6 +201,8 @@ type I{{.ModelStructName}}Do interface {
 	FirstOrCreate() (*{{.StructInfo.Package}}.{{.StructInfo.Type}}, error)
 	FindByPage(offset int, limit int) (result []*{{.StructInfo.Package}}.{{.StructInfo.Type}}, count int64, err error)
 	ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
+	Rows() (*sql.Rows, error)
+	Row() *sql.Row
 	Scan(result interface{}) (err error)
 	Returning(value interface{}, columns ...string) I{{.ModelStructName}}Do
 	UnderlyingDB() *gorm.DB

diff --git a/tests/.expect/dal_3/query/banks.gen.go b/tests/.expect/dal_3/query/banks.gen.go
diff --git a/tests/.expect/dal_3/query/credit_cards.gen.go b/tests/.expect/dal_3/query/credit_cards.gen.go
diff --git a/tests/.expect/dal_3/query/customers.gen.go b/tests/.expect/dal_3/query/customers.gen.go
diff --git a/tests/.expect/dal_3/query/people.gen.go b/tests/.expect/dal_3/query/people.gen.go
diff --git a/tests/.expect/dal_3/query/users.gen.go b/tests/.expect/dal_3/query/users.gen.go
diff --git a/tests/.expect/dal_4/query/banks.gen.go b/tests/.expect/dal_4/query/banks.gen.go
diff --git a/tests/.expect/dal_4/query/credit_cards.gen.go b/tests/.expect/dal_4/query/credit_cards.gen.go
diff --git a/tests/.expect/dal_4/query/customers.gen.go b/tests/.expect/dal_4/query/customers.gen.go
diff --git a/tests/.expect/dal_4/query/people.gen.go b/tests/.expect/dal_4/query/people.gen.go
diff --git a/tests/.expect/dal_4/query/users.gen.go b/tests/.expect/dal_4/query/users.gen.go
diff --git a/tests/.expect/dal_5/query/users.gen.go b/tests/.expect/dal_5/query/users.gen.go
diff --git a/tests/.expect/dal_6/query/users.gen.go b/tests/.expect/dal_6/query/users.gen.go