33 lines
861 B
Go
33 lines
861 B
Go
package drivers
|
|
|
|
import (
|
|
"log/slog"
|
|
"nixcn-cms/config"
|
|
"nixcn-cms/logger"
|
|
"strings"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
"gorm.io/plugin/opentelemetry/tracing"
|
|
)
|
|
|
|
func SplitHostPort(url string) (host, port string) {
|
|
if !strings.Contains(url, ":") {
|
|
return url, "5432"
|
|
}
|
|
split := strings.Split(url, ":")
|
|
return split[0], split[1]
|
|
}
|
|
|
|
func Postgres(dsn ExternalDSN) (*gorm.DB, error) {
|
|
host, port := SplitHostPort(dsn.Host)
|
|
conn := "host=" + host + " user=" + dsn.Username + " password=" + dsn.Password + " dbname=" + dsn.Name + " port=" + port + " sslmode=disable TimeZone=" + config.TZ()
|
|
db, err := gorm.Open(postgres.Open(conn), &gorm.Config{Logger: logger.GormLogger()})
|
|
|
|
// Use otel gorm plugin
|
|
if err := db.Use(tracing.NewPlugin()); err != nil {
|
|
slog.Error("[Database] Error starting otel plugin!", "err", err)
|
|
}
|
|
return db, err
|
|
}
|