fix: clickhouse logs and reg agents

backend/internal/repository/log_repository.go

@@ -2,44 +2,85 @@ package repository
 
 import (
 	"context"
+	"database/sql"
+	"fmt"
+	"sync"
 	"time"
 
 	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/backend/internal/storage"
-	"github.com/ClickHouse/clickhouse-go/v2/lib/driver"
 )
 
 type LogRepository struct {
-	Conn driver.Conn
+	mu sync.RWMutex
+	DB *sql.DB
 }
 
-func NewLogRepository(conn driver.Conn) *LogRepository {
-	return &LogRepository{Conn: conn}
+func NewLogRepository() *LogRepository {
+	return &LogRepository{}
+}
+
+func (r *LogRepository) SetDB(db *sql.DB) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	r.DB = db
+}
+
+func (r *LogRepository) IsConnected() bool {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	return r.DB != nil
+}
+
+func (r *LogRepository) getDB() *sql.DB {
+	r.mu.RLock()
+	defer r.mu.RUnlock()
+	return r.DB
 }
 
 func (r *LogRepository) Init(ctx context.Context) error {
-	return r.Conn.Exec(ctx, storage.CreateLogsTable)
+	db := r.getDB()
+	if db == nil {
+		return nil
+	}
+	_, err := db.ExecContext(ctx, storage.CreateLogsTable)
+	return err
 }
 
 func (r *LogRepository) Insert(ctx context.Context, log storage.LogEntry) error {
-	return r.Conn.Exec(ctx, `
+	db := r.getDB()
+	if db == nil {
+		return nil
+	}
+	_, err := db.ExecContext(ctx, `
 		INSERT INTO logs (timestamp, level, service, agent, message)
 		VALUES ($1, $2, $3, $4, $5)
 	`, log.Timestamp, log.Level, log.Service, log.Agent, log.Message)
+	return err
 }
 
 func (r *LogRepository) InsertBatch(ctx context.Context, logs []storage.LogEntry) error {
-	batch, err := r.Conn.PrepareBatch(ctx, "INSERT INTO logs (timestamp, level, service, agent, message)")
-	if err != nil {
-		return err
+	db := r.getDB()
+	if db == nil {
+		return nil
+	}
+	if len(logs) == 0 {
+		return nil
 	}
 
-	for _, log := range logs {
-		if err := batch.Append(log.Timestamp, log.Level, log.Service, log.Agent, log.Message); err != nil {
-			return err
+	// Build multi-row INSERT statement
+	query := "INSERT INTO logs (timestamp, level, service, agent, message) VALUES "
+	args := make([]interface{}, 0, len(logs)*5)
+	for i, log := range logs {
+		if i > 0 {
+			query += ", "
 		}
+		query += fmt.Sprintf("($%d, $%d, $%d, $%d, $%d)",
+			i*5+1, i*5+2, i*5+3, i*5+4, i*5+5)
+		args = append(args, log.Timestamp, log.Level, log.Service, log.Agent, log.Message)
 	}
 
-	return batch.Send()
+	_, err := db.ExecContext(ctx, query, args...)
+	return err
 }
 
 type LogFilter struct {
@@ -53,6 +94,11 @@ type LogFilter struct {
 }
 
 func (r *LogRepository) Search(ctx context.Context, filter LogFilter) ([]storage.LogEntry, error) {
+	db := r.getDB()
+	if db == nil {
+		return []storage.LogEntry{}, nil
+	}
+
 	query := "SELECT timestamp, level, service, agent, message FROM logs WHERE 1=1"
 	args := make([]interface{}, 0)
 	argIdx := 1
@@ -102,13 +148,13 @@ func (r *LogRepository) Search(ctx context.Context, filter LogFilter) ([]storage
 		args = append(args, filter.Offset)
 	}
 
-	rows, err := r.Conn.Query(ctx, query, args...)
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, err
 	}
 	defer rows.Close()
 
-	var logs []storage.LogEntry
+	logs := make([]storage.LogEntry, 0)
 	for rows.Next() {
 		var log storage.LogEntry
 		if err := rows.Scan(&log.Timestamp, &log.Level, &log.Service, &log.Agent, &log.Message); err != nil {
@@ -121,13 +167,17 @@ func (r *LogRepository) Search(ctx context.Context, filter LogFilter) ([]storage
 }
 
 func (r *LogRepository) GetDistinctServices(ctx context.Context) ([]string, error) {
-	rows, err := r.Conn.Query(ctx, "SELECT DISTINCT service FROM logs ORDER BY service")
+	db := r.getDB()
+	if db == nil {
+		return []string{}, nil
+	}
+	rows, err := db.QueryContext(ctx, "SELECT DISTINCT service FROM logs ORDER BY service")
 	if err != nil {
 		return nil, err
 	}
 	defer rows.Close()
 
-	var services []string
+	services := make([]string, 0)
 	for rows.Next() {
 		var service string
 		if err := rows.Scan(&service); err != nil {
@@ -140,13 +190,17 @@ func (r *LogRepository) GetDistinctServices(ctx context.Context) ([]string, erro
 }
 
 func (r *LogRepository) GetDistinctAgents(ctx context.Context) ([]string, error) {
-	rows, err := r.Conn.Query(ctx, "SELECT DISTINCT agent FROM logs ORDER BY agent")
+	db := r.getDB()
+	if db == nil {
+		return []string{}, nil
+	}
+	rows, err := db.QueryContext(ctx, "SELECT DISTINCT agent FROM logs ORDER BY agent")
 	if err != nil {
 		return nil, err
 	}
 	defer rows.Close()
 
-	var agents []string
+	agents := make([]string, 0)
 	for rows.Next() {
 		var agent string
 		if err := rows.Scan(&agent); err != nil {
@@ -159,13 +213,17 @@ func (r *LogRepository) GetDistinctAgents(ctx context.Context) ([]string, error)
 }
 
 func (r *LogRepository) GetDistinctLevels(ctx context.Context) ([]string, error) {
-	rows, err := r.Conn.Query(ctx, "SELECT DISTINCT level FROM logs ORDER BY level")
+	db := r.getDB()
+	if db == nil {
+		return []string{}, nil
+	}
+	rows, err := db.QueryContext(ctx, "SELECT DISTINCT level FROM logs ORDER BY level")
 	if err != nil {
 		return nil, err
 	}
 	defer rows.Close()
 
-	var levels []string
+	levels := make([]string, 0)
 	for rows.Next() {
 		var level string
 		if err := rows.Scan(&level); err != nil {