liz/PluralKit
liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(stats): use pooled connections to postgres

Browse Source
This commit is contained in:
spiral 2022-12-03 11:58:30 +00:00
parent 2602763e25
commit 939a727969
No known key found for this signature in database
GPG Key ID: 244A11E4B0BCF40E
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
services/scheduled_tasks/db.go
View File

@ -4,15 +4,15 @@ import (
"context" "context"
redis "github.com/go-redis/redis/v8" redis "github.com/go-redis/redis/v8"
pgx "github.com/jackc/pgx/v4" "github.com/jackc/pgx/v4/pgxpool"
) )
var data_db *pgx.Conn var data_db *pgxpool.Pool
var messages_db *pgx.Conn var messages_db *pgxpool.Pool
var stats_db *pgx.Conn var stats_db *pgxpool.Pool
var rdb *redis.Client var rdb *redis.Client
func run_simple_pg_query(c *pgx.Conn, sql string) { func run_simple_pg_query(c *pgxpool.Pool, sql string) {
_, err := c.Exec(context.Background(), sql) _, err := c.Exec(context.Background(), sql)
if err != nil { if err != nil {
panic(err) panic(err)
@ -26,8 +26,8 @@ func connect_dbs() {
rdb = redis_connect(get_env_var("REDIS_ADDR")) rdb = redis_connect(get_env_var("REDIS_ADDR"))
} }
func pg_connect(url string) *pgx.Conn { func pg_connect(url string) *pgxpool.Pool {
conn, err := pgx.Connect(context.Background(), url) conn, err := pgxpool.Connect(context.Background(), url)
if err != nil { if err != nil {
panic(err) panic(err)
} }