apollo-backend/internal/cmdutil/cmdutil.go

99 lines
2.3 KiB
Go
Raw Normal View History

package cmdutil
import (
"context"
"fmt"
"os"
2022-07-13 15:09:06 +00:00
"time"
"github.com/DataDog/datadog-go/statsd"
2022-11-01 23:02:25 +00:00
"github.com/adjust/rmq/v5"
"github.com/go-redis/redis/extra/redisotel/v8"
"github.com/go-redis/redis/v8"
2023-03-24 17:12:09 +00:00
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
2022-05-23 18:17:25 +00:00
"go.uber.org/zap"
)
2022-05-23 18:29:15 +00:00
func NewLogger(service string) *zap.Logger {
env := os.Getenv("ENV")
logger, _ := zap.NewProduction(zap.Fields(
zap.String("env", env),
zap.String("service", service),
))
if env == "" || env == "development" {
2022-05-23 18:17:25 +00:00
logger, _ = zap.NewDevelopment()
}
return logger
}
func NewStatsdClient(tags ...string) (*statsd.Client, error) {
if env := os.Getenv("ENV"); env != "" {
tags = append(tags, fmt.Sprintf("env:%s", env))
}
2021-07-23 00:22:46 +00:00
return statsd.New(os.Getenv("STATSD_URL"), statsd.WithTags(tags))
}
func NewRedisLocksClient(ctx context.Context, maxConns int) (*redis.Client, error) {
return newRedisClient(ctx, "REDIS_LOCKS_URL", maxConns)
}
func NewRedisQueueClient(ctx context.Context, maxConns int) (*redis.Client, error) {
return newRedisClient(ctx, "REDIS_QUEUE_URL", maxConns)
}
func newRedisClient(ctx context.Context, env string, maxConns int) (*redis.Client, error) {
opt, err := redis.ParseURL(os.Getenv(env))
if err != nil {
return nil, err
}
2022-07-13 19:02:38 +00:00
opt.PoolSize = maxConns
client := redis.NewClient(opt)
if err := client.Ping(ctx).Err(); err != nil {
return nil, err
}
client.AddHook(redisotel.NewTracingHook())
return client, nil
}
2021-07-23 00:50:24 +00:00
func NewDatabasePool(ctx context.Context, maxConns int) (*pgxpool.Pool, error) {
2021-10-09 14:59:20 +00:00
if maxConns == 0 {
maxConns = 1
}
2021-07-15 17:47:22 +00:00
url := fmt.Sprintf(
2021-07-23 00:38:13 +00:00
"%s?pool_max_conns=%d&pool_min_conns=%d",
2021-07-15 17:47:22 +00:00
os.Getenv("DATABASE_CONNECTION_POOL_URL"),
2021-07-23 00:50:24 +00:00
maxConns,
2,
2021-07-15 17:47:22 +00:00
)
config, err := pgxpool.ParseConfig(url)
if err != nil {
return nil, err
}
// Setting the build statement cache to nil helps this work with pgbouncer
2023-03-24 17:12:09 +00:00
config.ConnConfig.DefaultQueryExecMode = pgx.QueryExecModeSimpleProtocol
2022-07-13 15:09:06 +00:00
config.MaxConnLifetime = 1 * time.Hour
config.MaxConnIdleTime = 30 * time.Second
2023-03-24 17:12:09 +00:00
return pgxpool.NewWithConfig(ctx, config)
}
2022-05-23 18:17:25 +00:00
func NewQueueClient(logger *zap.Logger, conn *redis.Client, identifier string) (rmq.Connection, error) {
errChan := make(chan error, 10)
go func() {
for err := range errChan {
2022-05-23 18:17:25 +00:00
logger.Error("error occurred within queue", zap.Error(err))
}
}()
return rmq.OpenConnectionWithRedisClient(identifier, conn, errChan)
}