added cargo files
This commit is contained in:
290
PinePods-0.8.2/gpodder-api/internal/db/database.go
Normal file
290
PinePods-0.8.2/gpodder-api/internal/db/database.go
Normal file
@@ -0,0 +1,290 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/url"
|
||||
"os"
|
||||
"pinepods/gpodder-api/config"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
_ "github.com/go-sql-driver/mysql" // MySQL driver
|
||||
_ "github.com/lib/pq" // PostgreSQL driver
|
||||
)
|
||||
|
||||
// Database represents a database connection that can be either PostgreSQL or MySQL
|
||||
type Database struct {
|
||||
*sql.DB
|
||||
Type string // "postgresql" or "mysql"
|
||||
}
|
||||
|
||||
// NewDatabase creates a new database connection based on the DB_TYPE environment variable
|
||||
func NewDatabase(cfg config.DatabaseConfig) (*Database, error) {
|
||||
// Print connection details for debugging (hide password for security)
|
||||
fmt.Printf("Connecting to %s database: host=%s port=%d user=%s dbname=%s\n",
|
||||
cfg.Type, cfg.Host, cfg.Port, cfg.User, cfg.DBName)
|
||||
|
||||
var db *sql.DB
|
||||
var err error
|
||||
|
||||
switch cfg.Type {
|
||||
case "postgresql":
|
||||
db, err = connectPostgreSQL(cfg)
|
||||
case "mysql", "mariadb":
|
||||
db, err = connectMySQL(cfg)
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported database type: %s", cfg.Type)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// Test the connection
|
||||
if err := db.Ping(); err != nil {
|
||||
db.Close()
|
||||
if strings.Contains(err.Error(), "password authentication failed") {
|
||||
// Print environment variables (hide password)
|
||||
fmt.Println("Password authentication failed. Environment variables:")
|
||||
fmt.Printf("DB_HOST=%s\n", os.Getenv("DB_HOST"))
|
||||
fmt.Printf("DB_PORT=%s\n", os.Getenv("DB_PORT"))
|
||||
fmt.Printf("DB_USER=%s\n", os.Getenv("DB_USER"))
|
||||
fmt.Printf("DB_NAME=%s\n", os.Getenv("DB_NAME"))
|
||||
fmt.Printf("DB_PASSWORD=*** (length: %d)\n", len(os.Getenv("DB_PASSWORD")))
|
||||
}
|
||||
return nil, fmt.Errorf("failed to ping database: %w", err)
|
||||
}
|
||||
|
||||
fmt.Println("Successfully connected to the database")
|
||||
|
||||
// Migrations are now handled by the Python migration system
|
||||
// Skip Go migrations to avoid conflicts
|
||||
log.Println("Skipping Go migrations - now handled by Python migration system")
|
||||
|
||||
return &Database{DB: db, Type: cfg.Type}, nil
|
||||
}
|
||||
|
||||
// runMigrationsWithRetry - DISABLED: migrations now handled by Python system
|
||||
// func runMigrationsWithRetry(db *sql.DB, dbType string) error {
|
||||
// All migration logic has been moved to the Python migration system
|
||||
// to ensure consistency and centralized management
|
||||
// This function is kept for reference but is no longer used
|
||||
// }
|
||||
|
||||
// connectPostgreSQL connects to a PostgreSQL database
|
||||
func connectPostgreSQL(cfg config.DatabaseConfig) (*sql.DB, error) {
|
||||
// Escape special characters in password
|
||||
escapedPassword := url.QueryEscape(cfg.Password)
|
||||
|
||||
// Use a connection string without password for logging
|
||||
logConnStr := fmt.Sprintf(
|
||||
"host=%s port=%d user=%s dbname=%s sslmode=%s",
|
||||
cfg.Host, cfg.Port, cfg.User, cfg.DBName, cfg.SSLMode,
|
||||
)
|
||||
fmt.Printf("PostgreSQL connection string (without password): %s\n", logConnStr)
|
||||
|
||||
// Build the actual connection string with password
|
||||
connStr := fmt.Sprintf(
|
||||
"host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
|
||||
cfg.Host, cfg.Port, cfg.User, cfg.Password, cfg.DBName, cfg.SSLMode,
|
||||
)
|
||||
|
||||
// Try standard connection string first
|
||||
db, err := sql.Open("postgres", connStr)
|
||||
if err != nil {
|
||||
// Try URL format connection string
|
||||
urlConnStr := fmt.Sprintf(
|
||||
"postgres://%s:%s@%s:%d/%s?sslmode=%s",
|
||||
cfg.User, escapedPassword, cfg.Host, cfg.Port, cfg.DBName, cfg.SSLMode,
|
||||
)
|
||||
fmt.Println("First connection attempt failed, trying URL format...")
|
||||
db, err = sql.Open("postgres", urlConnStr)
|
||||
}
|
||||
|
||||
return db, err
|
||||
}
|
||||
|
||||
// Replace the existing connectMySQL function with this version
|
||||
func connectMySQL(cfg config.DatabaseConfig) (*sql.DB, error) {
|
||||
// Add needed parameters for MySQL authentication
|
||||
connStr := fmt.Sprintf(
|
||||
"%s:%s@tcp(%s:%d)/%s?parseTime=true&allowNativePasswords=true&multiStatements=true",
|
||||
cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DBName,
|
||||
)
|
||||
|
||||
fmt.Printf("Attempting MySQL connection to %s:%d as user '%s'\n",
|
||||
cfg.Host, cfg.Port, cfg.User)
|
||||
|
||||
// Open the connection
|
||||
db, err := sql.Open("mysql", connStr)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to open MySQL connection: %w", err)
|
||||
}
|
||||
|
||||
// Configure connection pool
|
||||
db.SetConnMaxLifetime(time.Minute * 3)
|
||||
db.SetMaxOpenConns(10)
|
||||
db.SetMaxIdleConns(5)
|
||||
|
||||
// Explicitly test the connection
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
|
||||
fmt.Println("Testing MySQL connection with ping...")
|
||||
if err := db.PingContext(ctx); err != nil {
|
||||
db.Close()
|
||||
fmt.Printf("MySQL connection failed: %v\n", err)
|
||||
return nil, fmt.Errorf("failed to ping MySQL database: %w", err)
|
||||
}
|
||||
|
||||
fmt.Println("MySQL connection successful!")
|
||||
return db, nil
|
||||
}
|
||||
|
||||
// Close closes the database connection
|
||||
func (db *Database) Close() error {
|
||||
return db.DB.Close()
|
||||
}
|
||||
|
||||
// IsMySQLDB returns true if the database is MySQL/MariaDB
|
||||
func (db *Database) IsMySQLDB() bool {
|
||||
return db.Type == "mysql"
|
||||
}
|
||||
|
||||
// IsPostgreSQLDB returns true if the database is PostgreSQL
|
||||
func (db *Database) IsPostgreSQLDB() bool {
|
||||
return db.Type == "postgresql"
|
||||
}
|
||||
|
||||
// FormatQuery formats a query for the specific database type
|
||||
func (db *Database) FormatQuery(query string) string {
|
||||
if db.Type == "postgresql" {
|
||||
return query // PostgreSQL queries already have correct format
|
||||
}
|
||||
|
||||
// For MySQL:
|
||||
result := query
|
||||
|
||||
// First, replace quoted table names
|
||||
knownTables := []string{
|
||||
"Users", "GpodderDevices", "GpodderSyncSettings",
|
||||
"GpodderSyncSubscriptions", "GpodderSyncEpisodeActions",
|
||||
"GpodderSyncPodcastLists", "GpodderSyncState", "GpodderSessions",
|
||||
"GpodderSyncMigrations", "Podcasts", "Episodes", "SavedEpisodes",
|
||||
"UserEpisodeHistory", "UserSettings", "APIKeys",
|
||||
}
|
||||
|
||||
for _, table := range knownTables {
|
||||
quoted := fmt.Sprintf("\"%s\"", table)
|
||||
result = strings.ReplaceAll(result, quoted, table)
|
||||
}
|
||||
|
||||
// Replace column quotes (double quotes to backticks)
|
||||
re := regexp.MustCompile(`"([^"]+)"`)
|
||||
result = re.ReplaceAllString(result, "`$1`")
|
||||
|
||||
// Then replace placeholders
|
||||
for i := 10; i > 0; i-- {
|
||||
old := fmt.Sprintf("$%d", i)
|
||||
result = strings.ReplaceAll(result, old, "?")
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
// Exec executes a query with the correct formatting for the database type
|
||||
func (db *Database) Exec(query string, args ...interface{}) (sql.Result, error) {
|
||||
formattedQuery := db.FormatQuery(query)
|
||||
return db.DB.Exec(formattedQuery, args...)
|
||||
}
|
||||
|
||||
// Query executes a query with the correct formatting for the database type
|
||||
func (db *Database) Query(query string, args ...interface{}) (*sql.Rows, error) {
|
||||
formattedQuery := db.FormatQuery(query)
|
||||
return db.DB.Query(formattedQuery, args...)
|
||||
}
|
||||
|
||||
// QueryRow executes a query with the correct formatting for the database type
|
||||
func (db *Database) QueryRow(query string, args ...interface{}) *sql.Row {
|
||||
formattedQuery := db.FormatQuery(query)
|
||||
return db.DB.QueryRow(formattedQuery, args...)
|
||||
}
|
||||
|
||||
// Begin starts a transaction with the correct formatting for the database type
|
||||
func (db *Database) Begin() (*Transaction, error) {
|
||||
tx, err := db.DB.Begin()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &Transaction{tx: tx, dbType: db.Type}, nil
|
||||
}
|
||||
|
||||
// Transaction is a wrapper around sql.Tx that formats queries correctly
|
||||
type Transaction struct {
|
||||
tx *sql.Tx
|
||||
dbType string
|
||||
}
|
||||
|
||||
// Commit commits the transaction
|
||||
func (tx *Transaction) Commit() error {
|
||||
return tx.tx.Commit()
|
||||
}
|
||||
|
||||
// Rollback rolls back the transaction
|
||||
func (tx *Transaction) Rollback() error {
|
||||
return tx.tx.Rollback()
|
||||
}
|
||||
|
||||
// Exec executes a query in the transaction with correct formatting
|
||||
func (tx *Transaction) Exec(query string, args ...interface{}) (sql.Result, error) {
|
||||
formattedQuery := formatQuery(query, tx.dbType)
|
||||
return tx.tx.Exec(formattedQuery, args...)
|
||||
}
|
||||
|
||||
// Query executes a query in the transaction with correct formatting
|
||||
func (tx *Transaction) Query(query string, args ...interface{}) (*sql.Rows, error) {
|
||||
formattedQuery := formatQuery(query, tx.dbType)
|
||||
return tx.tx.Query(formattedQuery, args...)
|
||||
}
|
||||
|
||||
// QueryRow executes a query in the transaction with correct formatting
|
||||
func (tx *Transaction) QueryRow(query string, args ...interface{}) *sql.Row {
|
||||
formattedQuery := formatQuery(query, tx.dbType)
|
||||
return tx.tx.QueryRow(formattedQuery, args...)
|
||||
}
|
||||
|
||||
// Helper function to format queries
|
||||
func formatQuery(query string, dbType string) string {
|
||||
if dbType == "postgresql" {
|
||||
return query
|
||||
}
|
||||
|
||||
// For MySQL:
|
||||
// Same logic as FormatQuery method
|
||||
result := query
|
||||
|
||||
knownTables := []string{
|
||||
"Users", "GpodderDevices", "GpodderSyncSettings",
|
||||
"GpodderSyncSubscriptions", "GpodderSyncEpisodeActions",
|
||||
"GpodderSyncPodcastLists", "GpodderSyncState", "GpodderSessions",
|
||||
"GpodderSyncMigrations", "Podcasts", "Episodes", "SavedEpisodes",
|
||||
"UserEpisodeHistory", "UserSettings", "APIKeys",
|
||||
}
|
||||
|
||||
for _, table := range knownTables {
|
||||
quoted := fmt.Sprintf("\"%s\"", table)
|
||||
result = strings.ReplaceAll(result, quoted, table)
|
||||
}
|
||||
|
||||
for i := 10; i > 0; i-- {
|
||||
old := fmt.Sprintf("$%d", i)
|
||||
result = strings.ReplaceAll(result, old, "?")
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
Reference in New Issue
Block a user