mirror of
https://github.com/grafana/grafana.git
synced 2025-08-01 06:21:50 +08:00
28f2a4078d
* records grafana database metrics for unified storage * update type to tracing.Tracer * use nil for tracer in tests
117 lines
3.4 KiB
Go
117 lines
3.4 KiB
Go
package dbimpl
|
|
|
|
import (
|
|
"cmp"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/go-sql-driver/mysql"
|
|
"github.com/grafana/grafana/pkg/infra/tracing"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
"github.com/grafana/grafana/pkg/storage/unified/sql/db"
|
|
"xorm.io/xorm"
|
|
)
|
|
|
|
func getEngineMySQL(getter *sectionGetter, tracer tracing.Tracer) (*xorm.Engine, error) {
|
|
config := mysql.NewConfig()
|
|
config.User = getter.String("db_user")
|
|
config.Passwd = getter.String("db_pass")
|
|
config.Net = "tcp"
|
|
config.Addr = getter.String("db_host")
|
|
config.DBName = getter.String("db_name")
|
|
config.Params = map[string]string{
|
|
// See: https://dev.mysql.com/doc/refman/en/sql-mode.html
|
|
"@@SESSION.sql_mode": "ANSI",
|
|
}
|
|
tls := getter.String("db_tls")
|
|
if tls != "" {
|
|
config.Params["tls"] = tls
|
|
}
|
|
config.Collation = "utf8mb4_unicode_ci"
|
|
config.Loc = time.UTC
|
|
config.AllowNativePasswords = true
|
|
config.ClientFoundRows = true
|
|
|
|
// allow executing multiple SQL statements in a single roundtrip, and also
|
|
// enable executing the CALL statement to run stored procedures that execute
|
|
// multiple SQL statements.
|
|
//config.MultiStatements = true
|
|
|
|
// TODO: do we want to support these?
|
|
// config.ServerPubKey = getter.String("db_server_pub_key")
|
|
// config.TLSConfig = getter.String("db_tls_config_name")
|
|
|
|
if err := getter.Err(); err != nil {
|
|
return nil, fmt.Errorf("config error: %w", err)
|
|
}
|
|
|
|
if strings.HasPrefix(config.Addr, "/") {
|
|
config.Net = "unix"
|
|
}
|
|
|
|
// FIXME: get rid of xorm
|
|
driverName := sqlstore.WrapDatabaseDriverWithHooks(db.DriverMySQL, tracer)
|
|
engine, err := xorm.NewEngine(driverName, config.FormatDSN())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("open database: %w", err)
|
|
}
|
|
|
|
engine.SetMaxOpenConns(0)
|
|
engine.SetMaxIdleConns(2)
|
|
engine.SetConnMaxLifetime(4 * time.Hour)
|
|
|
|
return engine, nil
|
|
}
|
|
|
|
func getEnginePostgres(getter *sectionGetter, tracer tracing.Tracer) (*xorm.Engine, error) {
|
|
dsnKV := map[string]string{
|
|
"user": getter.String("db_user"),
|
|
"password": getter.String("db_pass"),
|
|
"dbname": getter.String("db_name"),
|
|
"sslmode": cmp.Or(getter.String("db_sslmode"), "disable"),
|
|
}
|
|
|
|
// TODO: probably interesting:
|
|
// "passfile", "statement_timeout", "lock_timeout", "connect_timeout"
|
|
|
|
// TODO: for CockroachDB, we probably need to use the following:
|
|
// dsnKV["options"] = "-c enable_experimental_alter_column_type_general=true"
|
|
// Or otherwise specify it as:
|
|
// dsnKV["enable_experimental_alter_column_type_general"] = "true"
|
|
|
|
// TODO: do we want to support these options in the DSN as well?
|
|
// "sslkey", "sslcert", "sslrootcert", "sslpassword", "sslsni", "krbspn",
|
|
// "krbsrvname", "target_session_attrs", "service", "servicefile"
|
|
|
|
// More on Postgres connection string parameters:
|
|
// https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNSTRING
|
|
|
|
hostport := getter.String("db_host")
|
|
|
|
if err := getter.Err(); err != nil {
|
|
return nil, fmt.Errorf("config error: %w", err)
|
|
}
|
|
|
|
host, port, err := splitHostPortDefault(hostport, "127.0.0.1", "5432")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid db_host: %w", err)
|
|
}
|
|
dsnKV["host"] = host
|
|
dsnKV["port"] = port
|
|
|
|
dsn, err := MakeDSN(dsnKV)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error building DSN: %w", err)
|
|
}
|
|
|
|
// FIXME: get rid of xorm
|
|
driverName := sqlstore.WrapDatabaseDriverWithHooks(db.DriverPostgres, tracer)
|
|
engine, err := xorm.NewEngine(driverName, dsn)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("open database: %w", err)
|
|
}
|
|
|
|
return engine, nil
|
|
}
|