feat(config): standardize keepalive env names

This commit is contained in:
tiennm99 committed 2026-06-27 21:47:14 +07:00
1 parent f5218ea327
commit 655db8dcbf
12 files changed
+132 -72

No files matched your search

+6 -3
View File
@@ -13,11 +13,12 @@ func init() {
}
type postgresAdapter struct {
db *sql.DB
db *sql.DB
key string
}
func (a *postgresAdapter) Connect(ctx context.Context) error {
uri, err := envOrFail("SERVICE_URI")
uri, err := envOrFail("KEEPALIVE_POSTGRESQL_URL")
if err != nil {
return err
}
@@ -26,6 +27,7 @@ func (a *postgresAdapter) Connect(ctx context.Context) error {
return err
}
a.db = db
a.key = envOr("KEEPALIVE_COUNTER_KEY", "counter")
return a.db.PingContext(ctx)
}
@@ -36,7 +38,8 @@ func (a *postgresAdapter) Increment(ctx context.Context) (int64, error) {
}
var value int64
if err := tx.QueryRowContext(ctx,
`UPDATE keepalive SET value = value + 1 WHERE key = 'counter' RETURNING value`,
`UPDATE keepalive SET value = value + 1 WHERE key = $1 RETURNING value`,
a.key,
).Scan(&value); err != nil {
tx.Rollback()
return 0, err