50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package drivers
|
|
|
|
import (
|
|
"log/slog"
|
|
"nixcn-cms/config"
|
|
"nixcn-cms/logger"
|
|
"strings"
|
|
|
|
"github.com/spf13/viper"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"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) {
|
|
serviceName := viper.GetString("database.service_name")
|
|
|
|
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(),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = db.Use(tracing.NewPlugin(
|
|
tracing.WithAttributes(
|
|
attribute.String("peer.service", serviceName),
|
|
attribute.String("db.instance", dsn.Name),
|
|
),
|
|
))
|
|
|
|
if err != nil {
|
|
slog.Error("[Database] Error starting otel plugin!", "name", serviceName, "err", err)
|
|
}
|
|
|
|
return db, err
|
|
}
|