ZDatabase: Add count method

Zondax · Oct 31, 2023 · 78c4030 · 78c4030
1 parent 9a60456
commit 78c4030
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/pkg/zdb/methods.go b/pkg/zdb/methods.go
@@ -82,6 +82,10 @@ func (z *zDatabase) Distinct(args ...interface{}) ZDatabase {
 	return wrap(z.db.Distinct(args...))
 }
 
+func (z *zDatabase) Count(count *int64) ZDatabase {
+	return wrap(z.db.Count(count))
+}
+
 func (z *zDatabase) Group(name string) ZDatabase {
 	return wrap(z.db.Group(name))
 }

diff --git a/pkg/zdb/mock_zdb.go b/pkg/zdb/mock_zdb.go
@@ -112,6 +112,11 @@ func (m *MockZDatabase) Distinct(params ...interface{}) ZDatabase {
 	return args.Get(0).(ZDatabase)
 }
 
+func (m *MockZDatabase) Count(count *int64) ZDatabase {
+	args := m.Called(count)
+	return args.Get(0).(ZDatabase)
+}
+
 func (m *MockZDatabase) Scopes(funcs ...func(ZDatabase) ZDatabase) ZDatabase {
 	args := m.Called(funcs)
 	return args.Get(0).(ZDatabase)

diff --git a/pkg/zdb/zdb.go b/pkg/zdb/zdb.go
@@ -28,6 +28,7 @@ type ZDatabase interface {
 	Offset(offset int) ZDatabase
 	Order(value interface{}) ZDatabase
 	Distinct(args ...interface{}) ZDatabase
+	Count(count *int64) ZDatabase
 	Group(name string) ZDatabase
 	Create(value interface{}) ZDatabase
 	Delete(value interface{}, where ...interface{}) ZDatabase