Merge pull request #34 from RiemaLabs/perf/cha-11

perf db connect poll

config/config.go

@@ -41,15 +41,18 @@ type Contract struct {
 
 // Database
 type Database struct {
- Path string `yaml:"path,omitempty"`
- Kind string `yaml:"kind" validate:"required,oneof=sqlite postgres mysql clickhouse elasticsearch"`
- Host string `yaml:"host" validate:"required_with=Port User Database"`
- Port int `yaml:"port" validate:"required_with=Host User Database,gt=-1,lt=65535"`
- User string `yaml:"user" validate:"required_with=Host Port Database"`
- Password string `yaml:"password"`
- Database string `yaml:"database" validate:"required_with=Host Port User"`
- SchemaName string `yaml:"schema_name"`
- ApplicationName string `yaml:"application_name"`
+ Path string `yaml:"path,omitempty"`
+ Kind string `yaml:"kind" validate:"required,oneof=sqlite postgres mysql clickhouse elasticsearch"`
+ Host string `yaml:"host" validate:"required_with=Port User Database"`
+ Port int `yaml:"port" validate:"required_with=Host User Database,gt=-1,lt=65535"`
+ User string `yaml:"user" validate:"required_with=Host Port Database"`
+ Password string `yaml:"password"`
+ Database string `yaml:"database" validate:"required_with=Host Port User"`
+ SchemaName string `yaml:"schema_name"`
+ ApplicationName string `yaml:"application_name"`
+ MaxOpenConnections int `yaml:"max_open_connections"`
+ MaxIdleConnections int `yaml:"max_idle_connections"`
+ MaxLifetimeConnections int `yaml:"max_lifetime_connections"`
 }
 
 // Hasura -

database/bun.go

@@ -6,6 +6,7 @@ import (
  "fmt"
  "net/url"
  "runtime"
+ "time"
 
  "github.com/dipdup-net/go-lib/config"
  "github.com/pkg/errors"
@@ -69,8 +70,23 @@ func (db *Bun) Connect(ctx context.Context, cfg config.Database) error {
  db.conn = bun.NewDB(db.sqldb, pgdialect.New())
  }
  maxOpenConns := 4 * runtime.GOMAXPROCS(0)
+ maxIdleConns := maxOpenConns
+ maxLifetime := time.Minute
+ if cfg.MaxOpenConnections > 0 {
+ maxOpenConns = cfg.MaxOpenConnections
+ }
+
+ if cfg.MaxIdleConnections > 0 {
+ maxIdleConns = cfg.MaxIdleConnections
+ }
+
+ if cfg.MaxLifetimeConnections > 0 {
+ maxLifetime = time.Duration(cfg.MaxLifetimeConnections) * time.Second
+ }
+
  db.sqldb.SetMaxOpenConns(maxOpenConns)
- db.sqldb.SetMaxIdleConns(maxOpenConns)
+ db.sqldb.SetMaxIdleConns(maxIdleConns)
+ db.sqldb.SetConnMaxLifetime(maxLifetime)
  return nil
 }