feat: allow counting with filter

Author: Lutherwaves

storage/dynamodb.go

@@ -257,7 +257,7 @@ func (s *DynamoDBAdapter) Search(dest any, sortKey string, query string, limit i
 	})
 }
 
-func (s *DynamoDBAdapter) Count(dest any) (int64, error) {
+func (s *DynamoDBAdapter) Count(dest any, filter map[string]any) (int64, error) {
 	// TODO Implement
 	var total int64
 	return total, nil

storage/memory.go

@@ -100,10 +100,10 @@ func (m *MemoryAdapter) Search(dest any, sortKey string, query string, limit int
 	return m.DB.Search(dest, sortKey, query, limit, cursor)
 }
 
-func (s *MemoryAdapter) Count(dest any) (int64, error) {
-	//TODO: Implement
+func (m *MemoryAdapter) Count(dest any, filter map[string]any) (int64, error) {
 	var total int64
-	return total, nil
+	total, err := m.DB.Count(dest, filter)
+	return total, err
 }
 
 func (m *MemoryAdapter) Query(dest any, statement string, limit int, cursor string) (string, error) {

storage/sql.go

@@ -288,9 +288,13 @@ func (s *SQLAdapter) Search(dest any, sortKey string, query string, limit int, c
 	})
 }
 
-func (s *SQLAdapter) Count(dest any) (int64, error) {
+func (s *SQLAdapter) Count(dest any, filter map[string]any) (int64, error) {
 	q := s.DB.Model(dest)
 
+	if len(filter) > 0 {
+		q = q.Where(s.buildQuery(filter), filter)
+	}
+
 	var total int64
 	if err := q.Count(&total).Error; err != nil {
 		slog.Error("Error finding count")

storage/storage.go

@@ -24,7 +24,7 @@ type StorageAdapter interface {
 	Delete(item any, filter map[string]any) error
 	List(dest any, sortKey string, filter map[string]any, limit int, cursor string) (string, error)
 	Search(dest any, sortKey string, query string, limit int, cursor string) (string, error)
-	Count(dest any) (int64, error)
+	Count(dest any, filter map[string]any) (int64, error)
 	Query(dest any, statement string, limit int, cursor string) (string, error)
 }