refactor: Refactor audit logs count to support filtering (#4113)

coderd/database/databasefake/databasefake.go

@@ -2402,11 +2402,25 @@ func (q *fakeQuerier) GetAuditLogsOffset(ctx context.Context, arg database.GetAu
 	return logs, nil
 }
 
-func (q *fakeQuerier) GetAuditLogCount(_ context.Context) (int64, error) {
+func (q *fakeQuerier) GetAuditLogCount(_ context.Context, arg database.GetAuditLogCountParams) (int64, error) {
 	q.mutex.RLock()
 	defer q.mutex.RUnlock()
 
-	return int64(len(q.auditLogs)), nil
+	logs := make([]database.AuditLog, 0)
+
+	for _, alog := range q.auditLogs {
+		if arg.Action != "" && !strings.Contains(string(alog.Action), arg.Action) {
+			continue
+		}
+
+		if arg.ResourceType != "" && !strings.Contains(string(alog.ResourceType), arg.ResourceType) {
+			continue
+		}
+
+		logs = append(logs, alog)
+	}
+
+	return int64(len(logs)), nil
 }
 
 func (q *fakeQuerier) InsertAuditLog(_ context.Context, arg database.InsertAuditLogParams) (database.AuditLog, error) {

coderd/database/querier.go

@@ -27,7 +27,7 @@ type querier interface {
 	GetAPIKeyByID(ctx context.Context, id string) (APIKey, error)
 	GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Time) ([]APIKey, error)
 	GetActiveUserCount(ctx context.Context) (int64, error)
-	GetAuditLogCount(ctx context.Context) (int64, error)
+	GetAuditLogCount(ctx context.Context, arg GetAuditLogCountParams) (int64, error)
 	// GetAuditLogsBefore retrieves `row_limit` number of audit logs before the provided
 	// ID.
 	GetAuditLogsOffset(ctx context.Context, arg GetAuditLogsOffsetParams) ([]GetAuditLogsOffsetRow, error)

coderd/database/queries.sql.go

@@ -292,10 +292,28 @@ SELECT
     COUNT(*) as count
 FROM
     audit_logs
+WHERE
+    -- Filter resource_type
+	CASE
+		WHEN $1 :: text != '' THEN
+			resource_type = $1 :: resource_type
+		ELSE true
+	END
+	-- Filter action
+	AND CASE
+		WHEN $2 :: text != '' THEN
+			action = $2 :: audit_action
+		ELSE true
+	END
 `
 
-func (q *sqlQuerier) GetAuditLogCount(ctx context.Context) (int64, error) {
-	row := q.db.QueryRowContext(ctx, getAuditLogCount)
+type GetAuditLogCountParams struct {
+	ResourceType string `db:"resource_type" json:"resource_type"`
+	Action       string `db:"action" json:"action"`
+}
+
+func (q *sqlQuerier) GetAuditLogCount(ctx context.Context, arg GetAuditLogCountParams) (int64, error) {
+	row := q.db.QueryRowContext(ctx, getAuditLogCount, arg.ResourceType, arg.Action)
 	var count int64
 	err := row.Scan(&count)
 	return count, err

coderd/database/queries/auditlogs.sql

@@ -37,7 +37,20 @@ OFFSET
 SELECT
     COUNT(*) as count
 FROM
-    audit_logs;
+    audit_logs
+WHERE
+    -- Filter resource_type
+	CASE
+		WHEN @resource_type :: text != '' THEN
+			resource_type = @resource_type :: resource_type
+		ELSE true
+	END
+	-- Filter action
+	AND CASE
+		WHEN @action :: text != '' THEN
+			action = @action :: audit_action
+		ELSE true
+	END;
 
 -- name: InsertAuditLog :one
 INSERT INTO