fix(config): validate DATABASE_URL required and parseable, log redis disabled (B15)
This commit is contained in:
@@ -2,11 +2,14 @@ package config
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/jackc/pgx/v5/pgxpool"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
@@ -52,6 +55,18 @@ func Load() (*Config, error) {
|
|||||||
if secret == "" {
|
if secret == "" {
|
||||||
return nil, fmt.Errorf("JWT_SECRET required")
|
return nil, fmt.Errorf("JWT_SECRET required")
|
||||||
}
|
}
|
||||||
|
// Validate required fields.
|
||||||
|
dbURL := env("DATABASE_URL", "")
|
||||||
|
if dbURL == "" {
|
||||||
|
return nil, fmt.Errorf("DATABASE_URL is required")
|
||||||
|
}
|
||||||
|
if _, perr := pgxpool.ParseConfig(dbURL); perr != nil {
|
||||||
|
return nil, fmt.Errorf("DATABASE_URL: %w", perr)
|
||||||
|
}
|
||||||
|
if env("REDIS_URL", "") == "" {
|
||||||
|
log.Printf("redis disabled: rate-limit/scan-lock/page-cache off")
|
||||||
|
}
|
||||||
|
|
||||||
// resolveDir: macOS 开发机上 /var、/tmp 是指向 /private 的软链,
|
// resolveDir: macOS 开发机上 /var、/tmp 是指向 /private 的软链,
|
||||||
// 启动时解析一次才能让 root_path 前缀校验对上真实路径;目录不存在/出错则保留原值(Docker 路径不受影响)
|
// 启动时解析一次才能让 root_path 前缀校验对上真实路径;目录不存在/出错则保留原值(Docker 路径不受影响)
|
||||||
resolveDir := func(dir string) string {
|
resolveDir := func(dir string) string {
|
||||||
|
|||||||
@@ -6,6 +6,22 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func TestLoadDatabaseURLRequired(t *testing.T) {
|
||||||
|
t.Setenv("JWT_SECRET", "x")
|
||||||
|
t.Setenv("DATABASE_URL", "")
|
||||||
|
if _, err := Load(); err == nil {
|
||||||
|
t.Fatal("empty DATABASE_URL must fail")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadDatabaseURLMalformed(t *testing.T) {
|
||||||
|
t.Setenv("JWT_SECRET", "x")
|
||||||
|
t.Setenv("DATABASE_URL", "not a url")
|
||||||
|
if _, err := Load(); err == nil {
|
||||||
|
t.Fatal("malformed DATABASE_URL must fail")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestLoad(t *testing.T) {
|
func TestLoad(t *testing.T) {
|
||||||
// 屏蔽外部 env,保证默认值断言自洽(Load 将空串视为未设置)
|
// 屏蔽外部 env,保证默认值断言自洽(Load 将空串视为未设置)
|
||||||
t.Setenv("BOOKS_DIR", "")
|
t.Setenv("BOOKS_DIR", "")
|
||||||
|
|||||||
Reference in New Issue
Block a user