improve telemetry

Author: pierre-b

api

@@ -46,6 +46,7 @@ type AppInterface interface {
 	GetMessageHistoryRepository() domain.MessageHistoryRepository
 	GetContactListRepository() domain.ContactListRepository
 	GetTransactionalNotificationRepository() domain.TransactionalNotificationRepository
+	GetTelemetryRepository() domain.TelemetryRepository
 
 	// Server status methods
 	IsServerCreated() bool
@@ -81,6 +82,7 @@ type App struct {
 	transactionalNotificationRepo domain.TransactionalNotificationRepository
 	messageHistoryRepo            domain.MessageHistoryRepository
 	webhookEventRepo              domain.WebhookEventRepository
+	telemetryRepo                 domain.TelemetryRepository
 
 	// Services
 	authService                      *service.AuthService
@@ -288,6 +290,7 @@ func (a *App) InitRepositories() error {
 	a.transactionalNotificationRepo = repository.NewTransactionalNotificationRepository(a.workspaceRepo)
 	a.messageHistoryRepo = repository.NewMessageHistoryRepository(a.workspaceRepo)
 	a.webhookEventRepo = repository.NewWebhookEventRepository(a.workspaceRepo)
+	a.telemetryRepo = repository.NewTelemetryRepository(a.workspaceRepo)
 
 	return nil
 }
@@ -531,6 +534,7 @@ func (a *App) InitServices() error {
 		Enabled:       a.config.Telemetry,
 		APIEndpoint:   a.config.APIEndpoint,
 		WorkspaceRepo: a.workspaceRepo,
+		TelemetryRepo: a.telemetryRepo,
 		Logger:        a.logger,
 		HTTPClient:    httpClient, // Reuse the HTTP client created above
 	}
@@ -830,6 +834,10 @@ func (a *App) GetTransactionalNotificationRepository() domain.TransactionalNotif
 	return a.transactionalNotificationRepo
 }
 
+func (a *App) GetTelemetryRepository() domain.TelemetryRepository {
+	return a.telemetryRepo
+}
+
 // SetHandler allows setting a custom HTTP handler
 func (a *App) SetHandler(handler http.Handler) {
 	a.mux = handler.(*http.ServeMux)

internal/app/app.go

@@ -0,0 +1,46 @@
+package domain
+
+import (
+	"context"
+	"database/sql"
+)
+
+//go:generate mockgen -destination mocks/mock_telemetry_repository.go -package mocks github.com/Notifuse/notifuse/internal/domain TelemetryRepository
+
+// TelemetryMetrics represents aggregated metrics for a workspace
+type TelemetryMetrics struct {
+	ContactsCount      int    `json:"contacts_count"`
+	BroadcastsCount    int    `json:"broadcasts_count"`
+	TransactionalCount int    `json:"transactional_count"`
+	MessagesCount      int    `json:"messages_count"`
+	ListsCount         int    `json:"lists_count"`
+	UsersCount         int    `json:"users_count"`
+	LastMessageAt      string `json:"last_message_at"`
+}
+
+// TelemetryRepository defines the interface for telemetry data operations
+type TelemetryRepository interface {
+	// GetWorkspaceMetrics retrieves aggregated metrics for a specific workspace
+	GetWorkspaceMetrics(ctx context.Context, workspaceID string) (*TelemetryMetrics, error)
+
+	// CountContacts counts the total number of contacts in a workspace
+	CountContacts(ctx context.Context, db *sql.DB) (int, error)
+
+	// CountBroadcasts counts the total number of broadcasts in a workspace
+	CountBroadcasts(ctx context.Context, db *sql.DB) (int, error)
+
+	// CountTransactional counts the total number of transactional notifications in a workspace
+	CountTransactional(ctx context.Context, db *sql.DB) (int, error)
+
+	// CountMessages counts the total number of messages in a workspace
+	CountMessages(ctx context.Context, db *sql.DB) (int, error)
+
+	// CountLists counts the total number of lists in a workspace
+	CountLists(ctx context.Context, db *sql.DB) (int, error)
+
+	// CountUsers counts the total number of users in a workspace from the system database
+	CountUsers(ctx context.Context, systemDB *sql.DB, workspaceID string) (int, error)
+
+	// GetLastMessageAt gets the timestamp of the last message sent from the workspace
+	GetLastMessageAt(ctx context.Context, db *sql.DB) (string, error)
+}

internal/domain/mocks/mock_telemetry_repository.go

@@ -642,6 +642,7 @@ type WorkspaceRepository interface {
 
 	// Database management
 	GetConnection(ctx context.Context, workspaceID string) (*sql.DB, error)
+	GetSystemConnection(ctx context.Context) (*sql.DB, error)
 	CreateDatabase(ctx context.Context, workspaceID string) error
 	DeleteDatabase(ctx context.Context, workspaceID string) error