micro-wrapper-sql/README.md

75 lines
1.7 KiB
Markdown
Raw Permalink Normal View History

2022-12-07 10:20:08 +03:00
# micro-wrapper-sql
Example for For postgres
```go
package storage
2022-12-07 10:20:08 +03:00
import (
"fmt"
"net/url"
"time"
"github.com/jackc/pgx/v4"
"github.com/jackc/pgx/v4/stdlib"
"github.com/jmoiron/sqlx"
2022-12-07 10:20:56 +03:00
wrapper "go.unistack.org/micro-wrapper-sql/v3"
)
func Connect(cfg *PostgresConfig) (*sqlx.DB, error) {
// format connection string
cstr := fmt.Sprintf(
"postgres://%s:%s@%s/%s?sslmode=disable&statement_cache_mode=describe",
cfg.Login,
url.QueryEscape(cfg.Passw),
cfg.Addr,
cfg.DBName,
)
// parse connection string
dbConf, err := pgx.ParseConfig(cstr)
if err != nil {
return nil, err
}
// needed for pgbouncer
dbConf.RuntimeParams = map[string]string{
"standard_conforming_strings": "on",
"application_name": cfg.AppName,
}
// may be needed for pbbouncer, needs to check
// dbConf.PreferSimpleProtocol = true
// register pgx conn
dsn := stdlib.RegisterConnConfig(dbConf)
wrapper.DefaultMeterStatsInterval = 1 * time.Second
logger.DefaultLogger = logger.NewLogger(logger.WithLevel(logger.DebugLevel))
if err := logger.DefaultLogger.Init(); err != nil {
t.Fatal(err)
}
// for postgres user stdlib.GetDefaultDriver() or it fails
sql.Register("micro-wrapper-sql", wrapper.NewWrapper(&sqlite.Driver{},
wrapper.DatabaseHost("localhost"),
wrapper.DatabaseName("mydb"),
wrapper.LoggerLevel(logger.DebugLevel),
wrapper.LoggerEnabled(true),
))
wdb, err := sql.Open("micro-wrapper-sql", dsn)
if err != nil {
return nil, err
}
db := sqlx.NewDb(wdb, "pgx")
db.SetMaxOpenConns(int(cfg.ConnMax))
db.SetMaxIdleConns(int(cfg.ConnMaxIdle))
db.SetConnMaxLifetime(time.Duration(cfg.ConnLifetime) * time.Second)
db.SetConnMaxIdleTime(time.Duration(cfg.ConnMaxIdleTime) * time.Second)
return db, nil
}
```