putting in base query and collector framework

2024-12-18 18:52:38 +00:00
parent b0040f0cbe
commit 2209b877f5
1739 changed files with 288187 additions and 288200 deletions
@@ -1,45 +1,45 @@
-package database
-
-import (
-	"context"
-	"fmt"
-	"gotemplate/internal/env"
-	"log"
-
-	"github.com/jackc/pgx/v5"
-	"github.com/jackc/pgx/v5/pgxpool"
-)
-
-func createConnectionString() string {
-	driver := "postgres"
-	dbUser := env.GetRequired("DB_USER")
-	dbPass := env.GetRequired("DB_PASS")
-	dbHost := env.GetRequired("DB_HOST")
-	dbPort := env.GetRequired("DB_PORT")
-	dbName := env.GetRequired("DB_NAME")
-	connStr := fmt.Sprintf("%s://%s:%s@%s:%s/%s?sslmode=disable", driver, dbUser, dbPass, dbHost, dbPort, dbName)
-
-	return connStr
-}
-
-func GetDBPool(ctx context.Context) *pgxpool.Pool {
-	connStr := createConnectionString()
-
-	pool, err := pgxpool.New(ctx, connStr)
-	if err != nil {
-		log.Fatalf("Unable to create database pool: %v\n", err)
-	}
-
-	return pool
-}
-
-func GetDBConn(ctx context.Context) *pgx.Conn {
-	connStr := createConnectionString()
-
-	conn, err := pgx.Connect(ctx, connStr)
-	if err != nil {
-		log.Fatalf("Unable to connect to database: %v\n", err)
-	}
-
-	return conn
-}
+package database
+
+import (
+	"context"
+	"fmt"
+	"gotemplate/internal/env"
+	"log"
+
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgxpool"
+)
+
+func createConnectionString() string {
+	driver := "postgres"
+	dbUser := env.GetRequired("DB_USER")
+	dbPass := env.GetRequired("DB_PASS")
+	dbHost := env.GetRequired("DB_HOST")
+	dbPort := env.GetRequired("DB_PORT")
+	dbName := env.GetRequired("DB_NAME")
+	connStr := fmt.Sprintf("%s://%s:%s@%s:%s/%s?sslmode=disable", driver, dbUser, dbPass, dbHost, dbPort, dbName)
+
+	return connStr
+}
+
+func GetDBPool(ctx context.Context) *pgxpool.Pool {
+	connStr := createConnectionString()
+
+	pool, err := pgxpool.New(ctx, connStr)
+	if err != nil {
+		log.Fatalf("Unable to create database pool: %v\n", err)
+	}
+
+	return pool
+}
+
+func GetDBConn(ctx context.Context) *pgx.Conn {
+	connStr := createConnectionString()
+
+	conn, err := pgx.Connect(ctx, connStr)
+	if err != nil {
+		log.Fatalf("Unable to connect to database: %v\n", err)
+	}
+
+	return conn
+}