FEATURE/queries-via-UI

Author: serhii.zahuba

backend/internal/features/query/postgresql/di.go

@@ -12,22 +12,22 @@ var (
 )
 
 func init() {
-	// репозиторій та сервіс
+
 	sqlRepo = NewRepository()
 	sqlService = NewService(sqlRepo)
 
-	// залежності з сусідніх модулів через їх DI
+
 	userSvc := users.GetUserService()
 	dbSvc := databases.GetDatabaseService()
 
-	// контролер
+
 	sqlPgController = NewPostgresSqlQueryController(sqlService, userSvc, dbSvc)
 }
 
 func GetSqlQueryRepository() *Repository { return sqlRepo }
 func GetSqlQueryService() *Service       { return sqlService }
 
-// Головне: цей контролер реєструємо у router
+
 func GetPostgresSqlQueryController() *PostgresSqlQueryController {
 	return sqlPgController
 }

backend/internal/features/query/postgresql/repository.go

@@ -27,29 +27,24 @@ func (r *Repository) openPool(ctx context.Context, dsn string) (*pgxpool.Pool, e
 	return pgxpool.NewWithConfig(ctx, cfg)
 }
 
-func buildPostgresDSN(pg *postgresql.PostgresqlDatabase) (string, error) {
-	if pg == nil {
+func buildPostgresDSN(p *postgresql.PostgresqlDatabase) (string, error) {
+	if p == nil {
 		return "", fmt.Errorf("postgres config is nil")
 	}
-	if pg.Database == nil {
-		return "", fmt.Errorf("database name is nil")
-	}
 
-	db := *pg.Database
+	dbname := "postgres"
+	if p.Database != nil && strings.TrimSpace(*p.Database) != "" {
+		dbname = strings.TrimSpace(*p.Database)
+	}
 
-	ssl := "disable"
-	if pg.IsHttps {
-		ssl = "require"
+	sslMode := "disable"
+	if p.IsHttps {
+		sslMode = "require"
 	}
 
 	return fmt.Sprintf(
 		"host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
-		pg.Host,
-		pg.Port,
-		pg.Username,
-		pg.Password,
-		db,
-		ssl,
+		p.Host, p.Port, p.Username, p.Password, dbname, sslMode,
 	), nil
 }
 
@@ -62,9 +57,7 @@ type Result struct {
 }
 
 // ExecuteSelect — safe SELECT/CTE with forced LIMIT
-// ExecuteSQL — один стейтмент: SELECT/CTE виконуємо з лімітом, інші — через Exec
 func (r *Repository) ExecuteSQL(ctx context.Context, dbc *databases.Database, sql string, maxRows int) (*Result, error) {
-	// підтримуємо лише PostgreSQL (як і раніше)
 	t := strings.ToUpper(string(dbc.Type))
 	if t != "POSTGRES" && t != "POSTGRESQL" {
 		return nil, fmt.Errorf("only PostgreSQL type is supported, got %q", dbc.Type)
@@ -86,7 +79,7 @@ func (r *Repository) ExecuteSQL(ctx context.Context, dbc *databases.Database, sq
 
 	start := time.Now()
 	if isSelect {
-		// SELECT / WITH ... SELECT — обгортаємо лімітом
+
 		stmt = ensureLimit(stmt, maxRows)
 
 		rows, err := pool.Query(ctx, stmt)
@@ -120,37 +113,36 @@ func (r *Repository) ExecuteSQL(ctx context.Context, dbc *databases.Database, sq
 		return &Result{
 			Columns:   cols,
 			Rows:      outRows,
-			RowCount:  rowCount,        // кількість повернутих рядків
+			RowCount:  rowCount,
 			Truncated: rowCount >= maxRows,
 			Duration:  time.Since(start),
 		}, nil
 	}
 
-	// DML/DDL: UPDATE/INSERT/DELETE/CREATE/... — просто виконуємо
+
 	tag, err := pool.Exec(ctx, stmt)
 	if err != nil {
 		return nil, err
 	}
 
-	// фронту достатньо часу виконання; таблиця буде порожня
-	// (за бажанням можна додати RowsAffected в DTO пізніше)
+
 	return &Result{
 		Columns:   []string{},
 		Rows:      [][]any{},
-		RowCount:  int(tag.RowsAffected()), // для інфи: скільки рядків зачеплено
+		RowCount:  int(tag.RowsAffected()),
 		Truncated: false,
 		Duration:  time.Since(start),
 	}, nil
 }
 
-// дуже проста евристика: SELECT або WITH без DML-ключових слів
+
 func isSelectLike(sql string) bool {
 	up := strings.ToUpper(strings.TrimSpace(sql))
 	if strings.HasPrefix(up, "SELECT ") {
 		return true
 	}
 	if strings.HasPrefix(up, "WITH ") {
-		// якщо це WITH ... INSERT/UPDATE/DELETE/... — вважаємо не-SELECT
+		
 		for _, k := range []string{" INSERT ", " UPDATE ", " DELETE ", " CREATE ", " ALTER ", " DROP ", " TRUNCATE ", " GRANT ", " REVOKE "} {
 			if strings.Contains(up, k) {
 				return false

backend/internal/features/query/postgresql/service_test.go

@@ -15,9 +15,8 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-// ==== helpers (аналогічно до інших тестів у репо) ====
 
-// беремо повну модель користувача для доступу до DatabaseService
+
 func getTestUserModel(t *testing.T) *users_models.User {
 	t.Helper()
 
@@ -34,20 +33,19 @@ func getTestUserModel(t *testing.T) *users_models.User {
 	return u
 }
 
-// генеруємо унікальне ім'я тимчасової таблиці (не TEMP, бо різні конекшени)
+
 func tmpTableName() string {
 	return "tmp_sqlquery_" + strings.ReplaceAll(uuid.New().String(), "-", "_")
 }
 
-// виконує один SQL через сервіс
+
 func runSQL(t *testing.T, svc *Service, dbc *databases.Database, sql string) *ExecuteResponse {
 	t.Helper()
 	req := &ExecuteRequest{
-		// DatabaseId тут не використовується сервісом, бо ми передаємо готовий dbc,
-		// але залишимо для повноти
+
 		DatabaseID: dbc.ID,
 		SQL:        sql,
-		// для SELECT сервіс сам підставить maxRows/timeout за замовчуванням
+
 	}
 	resp, err := svc.Execute(dbc, req)
 	assert.NoError(t, err, "execute failed for SQL: %s", sql)
@@ -57,7 +55,7 @@ func runSQL(t *testing.T, svc *Service, dbc *databases.Database, sql string) *Ex
 // ==== tests ====
 
 func Test_Service_Execute_SelectUpdateDelete(t *testing.T) {
-	// arrange: користувач, storage, notifier, database, svc, dbc
+
 	testUser := getTestUserModel(t)
 
 	testUserResp := users.GetTestUser()
@@ -76,7 +74,7 @@ func Test_Service_Execute_SelectUpdateDelete(t *testing.T) {
 	assert.NoError(t, err)
 	assert.NotNil(t, dbc)
 
-	svc := GetSqlQueryService() // через DI (internal/features/sqlquery/di.go)
+	svc := GetSqlQueryService()
 
 	table := tmpTableName()
 
@@ -104,27 +102,27 @@ func Test_Service_Execute_SelectUpdateDelete(t *testing.T) {
 	assert.Equal(t, []string{"id", "name", "cnt"}, sel.Columns)
 	assert.Len(t, sel.Rows, 3)
 
-	// прості перевірки значень (без жорсткого касту типів)
+	// simple check
 	row1 := sel.Rows[0]
 	assert.Equal(t, "1", fmt.Sprint(row1[0]))
 	assert.Equal(t, "a", fmt.Sprint(row1[1]))
 	assert.Equal(t, "10", fmt.Sprint(row1[2]))
 
-	// 4) UPDATE двох рядків
+	// 4) UPDATE two rows
 	updateSQL := fmt.Sprintf(`UPDATE %s SET cnt = cnt + 5 WHERE id IN (1,3)`, table)
 	upd := runSQL(t, svc, dbc, updateSQL)
 	assert.Equal(t, 2, upd.RowCount)
 	assert.Empty(t, upd.Columns)
 	assert.Empty(t, upd.Rows)
 
-	// 5) DELETE одного рядка
+	// 5) DELETE one rows
 	deleteSQL := fmt.Sprintf(`DELETE FROM %s WHERE id = 2`, table)
 	del := runSQL(t, svc, dbc, deleteSQL)
 	assert.Equal(t, 1, del.RowCount)
 	assert.Empty(t, del.Columns)
 	assert.Empty(t, del.Rows)
 
-	// 6) повторний SELECT: лишилось 2 записи, cnt у 1й та 3й збільшився на 5
+	// 6) second SELECT
 	sel2 := runSQL(t, svc, dbc, selectSQL)
 	assert.Equal(t, 2, sel2.RowCount)
 	assert.Equal(t, []string{"id", "name", "cnt"}, sel2.Columns)

frontend/src/entity/query/api/sqlQueryApi.ts

@@ -1,4 +1,3 @@
-// src/entity/api/sqlQueryApi.ts
 import RequestOptions from '../../../shared/api/RequestOptions';
 import { apiHelper } from '../../../shared/api/apiHelper';
 import type { ExecuteResponse } from '../model/ExecuteResponse';

frontend/src/entity/query/index.ts

@@ -1,3 +1,2 @@
-// src/entity/query/index.ts
 export * from './api/sqlQueryApi';
 export * from './model/ExecuteResponse';

frontend/src/features/sqlquery/index.ts

@@ -1,2 +1 @@
-// src/features/sqlquery/index.ts
 export { SqlQueryComponent } from './ui/SqlQueryComponent';