feat: add Exists method (#421)

* feat: add Exists method * update Exists method and test --------- Co-authored-by: Wenbo Han <[email protected]>
goravel · Feb 15, 2024 · 1a9f352 · 1a9f352
1 parent 300c1c3
commit 1a9f352
Show file tree

Hide file tree

Showing 5 changed files with 56 additions and 0 deletions.
diff --git a/contracts/database/orm/orm.go b/contracts/database/orm/orm.go
@@ -49,6 +49,8 @@ type Query interface {
 	Distinct(args ...any) Query
 	// Exec executes raw sql
 	Exec(sql string, values ...any) (*Result, error)
+	// Exists returns true if matching records exist; otherwise, it returns false.
+	Exists(exists *bool) error
 	// Find finds records that match given conditions.
 	Find(dest any, conds ...any) error
 	// FindOrFail finds records that match given conditions or throws an error.

diff --git a/database/gorm/query.go b/database/gorm/query.go
@@ -157,6 +157,10 @@ func (r *QueryImpl) Exec(sql string, values ...any) (*ormcontract.Result, error)
 	}, result.Error
 }
 
+func (r *QueryImpl) Exists(exists *bool) error {
+	return r.instance.Select("1").Limit(1).Find(exists).Error
+}
+
 func (r *QueryImpl) Find(dest any, conds ...any) error {
 	if err := r.refreshConnection(dest); err != nil {
 		return err

diff --git a/database/gorm/query_test.go b/database/gorm/query_test.go
@@ -1552,6 +1552,28 @@ func (s *QueryTestSuite) TestExec() {
 	}
 }
 
+func (s *QueryTestSuite) TestExists() {
+	for driver, query := range s.queries {
+		s.Run(driver.String(), func() {
+			user := User{Name: "exists_user", Avatar: "exists_avatar"}
+			s.Nil(query.Create(&user))
+			s.True(user.ID > 0)
+
+			user1 := User{Name: "exists_user", Avatar: "exists_avatar_1"}
+			s.Nil(query.Create(&user1))
+			s.True(user1.ID > 0)
+
+			var t bool
+			s.Nil(query.Model(&User{}).Where("name = ?", "exists_user").Exists(&t))
+			s.True(t)
+
+			var f bool
+			s.Nil(query.Model(&User{}).Where("name = ?", "no_exists_user").Exists(&f))
+			s.False(f)
+		})
+	}
+}
+
 func (s *QueryTestSuite) TestFind() {
 	for _, query := range s.queries {
 		tests := []struct {

diff --git a/mocks/database/orm/Query.go b/mocks/database/orm/Query.go
diff --git a/mocks/database/orm/Transaction.go b/mocks/database/orm/Transaction.go