mirror of
https://github.com/zitadel/zitadel.git
synced 2025-03-01 13:47:33 +00:00
4645045987
# Which Problems Are Solved Zitadel currently uses 3 database pool, 1 for queries, 1 for pushing events and 1 for scheduled projection updates. This defeats the purpose of a connection pool which already handles multiple connections. During load tests we found that the current structure of connection pools consumes a lot of database resources. The resource usage dropped after we reduced the amount of database pools to 1 because existing connections can be used more efficiently. # How the Problems Are Solved Removed logic to handle multiple connection pools and use a single one. # Additional Changes none # Additional Context part of https://github.com/zitadel/zitadel/issues/8352
66 lines
2.0 KiB
Go
66 lines
2.0 KiB
Go
package dialect
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"reflect"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
var (
|
|
ErrIllegalMaxOpenConns = errors.New("MaxOpenConns of the database must be higher than 3 or 0 for unlimited")
|
|
ErrIllegalMaxIdleConns = errors.New("MaxIdleConns of the database must be higher than 3 or 0 for unlimited")
|
|
)
|
|
|
|
// ConnectionConfig defines the Max Open and Idle connections for a DB connection pool.
|
|
type ConnectionConfig struct {
|
|
MaxOpenConns,
|
|
MaxIdleConns uint32
|
|
AfterConnect []func(ctx context.Context, c *pgx.Conn) error
|
|
}
|
|
|
|
var afterConnectFuncs []func(ctx context.Context, c *pgx.Conn) error
|
|
|
|
func RegisterAfterConnect(f func(ctx context.Context, c *pgx.Conn) error) {
|
|
afterConnectFuncs = append(afterConnectFuncs, f)
|
|
}
|
|
|
|
func RegisterDefaultPgTypeVariants[T any](m *pgtype.Map, name, arrayName string) {
|
|
// T
|
|
var value T
|
|
m.RegisterDefaultPgType(value, name)
|
|
|
|
// *T
|
|
valueType := reflect.TypeOf(value)
|
|
m.RegisterDefaultPgType(reflect.New(valueType).Interface(), name)
|
|
|
|
// []T
|
|
sliceType := reflect.SliceOf(valueType)
|
|
m.RegisterDefaultPgType(reflect.MakeSlice(sliceType, 0, 0).Interface(), arrayName)
|
|
|
|
// *[]T
|
|
m.RegisterDefaultPgType(reflect.New(sliceType).Interface(), arrayName)
|
|
|
|
// []*T
|
|
sliceOfPointerType := reflect.SliceOf(reflect.TypeOf(reflect.New(valueType).Interface()))
|
|
m.RegisterDefaultPgType(reflect.MakeSlice(sliceOfPointerType, 0, 0).Interface(), arrayName)
|
|
|
|
// *[]*T
|
|
m.RegisterDefaultPgType(reflect.New(sliceOfPointerType).Interface(), arrayName)
|
|
}
|
|
|
|
// NewConnectionConfig calculates [ConnectionConfig] values from the passed ratios
|
|
// and returns the config applicable for the requested purpose.
|
|
//
|
|
// openConns and idleConns must be at least 3 or 0, which means no limit.
|
|
// The pusherRatio and spoolerRatio must be between 0 and 1.
|
|
func NewConnectionConfig(openConns, idleConns uint32) *ConnectionConfig {
|
|
return &ConnectionConfig{
|
|
MaxOpenConns: openConns,
|
|
MaxIdleConns: idleConns,
|
|
AfterConnect: afterConnectFuncs,
|
|
}
|
|
}
|