mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
127 lines
2.5 KiB
Go
127 lines
2.5 KiB
Go
package session
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/iancoleman/strcase"
|
|
"github.com/jmoiron/sqlx"
|
|
"go.temporal.io/server/common/config"
|
|
"go.temporal.io/server/common/persistence/sql/sqlplugin/postgresql/driver"
|
|
"go.temporal.io/server/common/resolver"
|
|
)
|
|
|
|
const (
|
|
dsnFmt = "postgres://%v:%v@%v/%v?%v"
|
|
)
|
|
|
|
const (
|
|
sslMode = "sslmode"
|
|
sslModeNoop = "disable"
|
|
sslModeRequire = "require"
|
|
sslModeFull = "verify-full"
|
|
|
|
sslCA = "sslrootcert"
|
|
sslKey = "sslkey"
|
|
sslCert = "sslcert"
|
|
)
|
|
|
|
type Session struct {
|
|
*sqlx.DB
|
|
}
|
|
|
|
func NewSession(
|
|
cfg *config.SQL,
|
|
d driver.Driver,
|
|
resolver resolver.ServiceResolver,
|
|
) (*Session, error) {
|
|
db, err := createConnection(cfg, d, resolver)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Session{DB: db}, nil
|
|
}
|
|
|
|
func (s *Session) Close() {
|
|
if s.DB != nil {
|
|
_ = s.DB.Close()
|
|
}
|
|
}
|
|
|
|
func createConnection(
|
|
cfg *config.SQL,
|
|
d driver.Driver,
|
|
resolver resolver.ServiceResolver,
|
|
) (*sqlx.DB, error) {
|
|
db, err := d.CreateConnection(buildDSN(cfg, resolver))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if cfg.MaxConns > 0 {
|
|
db.SetMaxOpenConns(cfg.MaxConns)
|
|
}
|
|
if cfg.MaxIdleConns > 0 {
|
|
db.SetMaxIdleConns(cfg.MaxIdleConns)
|
|
}
|
|
if cfg.MaxConnLifetime > 0 {
|
|
db.SetConnMaxLifetime(cfg.MaxConnLifetime)
|
|
}
|
|
|
|
// Maps struct names in CamelCase to snake without need for db struct tags.
|
|
db.MapperFunc(strcase.ToSnake)
|
|
return db, nil
|
|
}
|
|
|
|
func buildDSN(
|
|
cfg *config.SQL,
|
|
r resolver.ServiceResolver,
|
|
) string {
|
|
tlsAttrs := buildDSNAttr(cfg).Encode()
|
|
resolvedAddr := r.Resolve(cfg.ConnectAddr)[0]
|
|
dsn := fmt.Sprintf(
|
|
dsnFmt,
|
|
cfg.User,
|
|
url.QueryEscape(cfg.Password),
|
|
resolvedAddr,
|
|
cfg.DatabaseName,
|
|
tlsAttrs,
|
|
)
|
|
return dsn
|
|
}
|
|
|
|
func buildDSNAttr(cfg *config.SQL) url.Values {
|
|
parameters := url.Values{}
|
|
if cfg.TLS != nil && cfg.TLS.Enabled {
|
|
if !cfg.TLS.EnableHostVerification {
|
|
parameters.Set(sslMode, sslModeRequire)
|
|
} else {
|
|
parameters.Set(sslMode, sslModeFull)
|
|
}
|
|
|
|
if cfg.TLS.CaFile != "" {
|
|
parameters.Set(sslCA, cfg.TLS.CaFile)
|
|
}
|
|
if cfg.TLS.KeyFile != "" && cfg.TLS.CertFile != "" {
|
|
parameters.Set(sslKey, cfg.TLS.KeyFile)
|
|
parameters.Set(sslCert, cfg.TLS.CertFile)
|
|
}
|
|
} else {
|
|
parameters.Set(sslMode, sslModeNoop)
|
|
}
|
|
|
|
for k, v := range cfg.ConnectAttributes {
|
|
key := strings.TrimSpace(k)
|
|
value := strings.TrimSpace(v)
|
|
if parameters.Get(key) != "" {
|
|
panic(fmt.Sprintf("duplicate connection attr: %v:%v, %v:%v",
|
|
key,
|
|
parameters.Get(key),
|
|
key, value,
|
|
))
|
|
}
|
|
parameters.Set(key, value)
|
|
}
|
|
return parameters
|
|
}
|