feat: facades.Orm supports cursor

contracts/database/orm/orm.go

@@ -30,6 +30,7 @@ type Query interface {
 	Driver() Driver
 	Count(count *int64) error
 	Create(value any) error
+	Cursor() (chan Cursor, error)
 	Delete(value any, conds ...any) (*Result, error)
 	Distinct(args ...any) Query
 	Exec(sql string, values ...any) (*Result, error)
@@ -83,6 +84,11 @@ type Association interface {
 	Count() int64
 }
 
+//go:generate mockery --name=Cursor
+type Cursor interface {
+	Scan(value any) error
+}
+
 type Result struct {
 	RowsAffected int64
 }

database/gorm/cursor.go

@@ -0,0 +1,21 @@
+package gorm
+
+import "github.com/mitchellh/mapstructure"
+
+type CursorImpl struct {
+	row map[string]any
+}
+
+func (c *CursorImpl) Scan(value any) error {
+	msConfig := &mapstructure.DecoderConfig{
+		Squash: true,
+		Result: value,
+	}
+
+	decoder, err := mapstructure.NewDecoder(msConfig)
+	if err != nil {
+		return err
+	}
+
+	return decoder.Decode(c.row)
+}

database/gorm/query.go

@@ -81,6 +81,29 @@ func (r *QueryImpl) Create(value any) error {
 	return r.create(value)
 }
 
+func (r *QueryImpl) Cursor() (chan ormcontract.Cursor, error) {
+	var err error
+	cursorChan := make(chan ormcontract.Cursor)
+	go func() {
+		rows, err := r.instance.Rows()
+		if err != nil {
+			return
+		}
+		defer rows.Close()
+
+		for rows.Next() {
+			val := make(map[string]any)
+			err := r.instance.ScanRows(rows, val)
+			if err != nil {
+				return
+			}
+			cursorChan <- &CursorImpl{row: val}
+		}
+		close(cursorChan)
+	}()
+	return cursorChan, err
+}
+
 func (r *QueryImpl) Delete(dest any, conds ...any) (*ormcontract.Result, error) {
 	if err := r.deleting(dest); err != nil {
 		return nil, err

database/gorm/query_test.go

@@ -2458,6 +2458,32 @@ func (s *QueryTestSuite) TestSum() {
 	}
 }
 
+func (s *QueryTestSuite) TestCursor() {
+	for driver, query := range s.queries {
+		s.Run(driver.String(), func() {
+			user := User{Name: "cursor_user", Avatar: "cursor_avatar"}
+			s.Nil(query.Create(&user))
+			s.True(user.ID > 0)
+
+			user1 := User{Name: "cursor_user", Avatar: "cursor_avatar1"}
+			s.Nil(query.Create(&user1))
+			s.True(user1.ID > 0)
+
+			users, err := query.Model(&User{}).Where("name = ?", "cursor_user").Cursor()
+			s.Nil(err)
+			var size int
+			for row := range users {
+				var user User
+				s.Nil(row.Scan(&user))
+				s.True(user.ID > 0)
+				s.True(len(user.Name) > 0)
+				size++
+			}
+			s.Equal(2, size)
+		})
+	}
+}
+
 func (s *QueryTestSuite) TestTransactionSuccess() {
 	for driver, query := range s.queries {
 		s.Run(driver.String(), func() {