fix(pipeline): harden upstream resilience and production shutdown

DoH через DoWithRetry; CDN preview через UpstreamHTTPDo; частичный fail CDN (EVOBGP_CDN_PARTIAL_OK); безопасный доступ к Job.Meta; drain jobs при SIGTERM; ValidateProductionEnforce при EVOBGP_PRODUCTION=1.

Co-authored-by: Cursor <[email protected]>
This commit is contained in:
Denozordec
2026-07-31 12:11:52 +07:00
co-authored by Cursor
parent 1e04e91dd8
commit 53ce80c9ff
14 changed files with 228 additions and 16 deletions
+10
View File
@@ -29,6 +29,9 @@ func main() {
os.Exit(dbcli.Run(os.Args[2:]))
}
cfg := config.Load()
if err := config.ValidateProductionEnforce(); err != nil {
log.Fatal(err)
}
opts := httpapi.Options{
APIKeys: os.Getenv("EVOBGP_API_KEYS"),
DatabaseURL: cfg.DatabaseURL,
@@ -86,6 +89,13 @@ func main() {
<-ctx.Done()
shutdownCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
n := srv.Jobs().RequestCancelAll()
if n > 0 {
log.Printf("draining %d job(s)…", n)
if err := srv.Jobs().Drain(shutdownCtx); err != nil {
log.Printf("job drain: %v", err)
}
}
if err := httpSrv.Shutdown(shutdownCtx); err != nil {
log.Printf("HTTP shutdown: %v", err)
}
+10
View File
@@ -24,6 +24,9 @@ func main() {
os.Exit(dbcli.Run(os.Args[2:]))
}
cfg := config.Load()
if err := config.ValidateProductionEnforce(); err != nil {
log.Fatal(err)
}
seedDemo := os.Getenv("EVOBGP_SEED_DEMO") != "0"
opts := httpapi.Options{
APIKeys: os.Getenv("EVOBGP_API_KEYS"),
@@ -78,6 +81,13 @@ func main() {
<-ctx.Done()
shutdownCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
n := srv.Jobs().RequestCancelAll()
if n > 0 {
log.Printf("draining %d job(s)…", n)
if err := srv.Jobs().Drain(shutdownCtx); err != nil {
log.Printf("job drain: %v", err)
}
}
if err := httpSrv.Shutdown(shutdownCtx); err != nil {
log.Printf("HTTP shutdown: %v", err)
}
+2
View File
@@ -4,12 +4,14 @@
## Обязательно
- `EVOBGP_PRODUCTION=1` (или `EVOBGP_ENV=production`) — при старте API/`evobgp-all` процесс **откажется** стартовать, если нарушены жёсткие требования ниже (`config.ValidateProductionEnforce`).
- `EVOBGP_SEED_DEMO=0` — отключить demo-tenant и токен `Bearer dev`.
- `EVOBGP_DEV_INSECURE` не задавать или `0` — не использовать lab-флаги в prod.
- `EVOBGP_BUNDLE_SEED_HEX` — задать стабильный hex-ключ подписи бандлов; сохранить pubkey для нод.
- PostgreSQL с TLS (`sslmode` не `disable`) при доступе вне private network.
- `EVOBGP_CORS_ORIGINS` — явный whitelist origin веб-панели.
- `EVOBGP_STALE_ON_UPSTREAM_ERROR=1` (по умолчанию) — stale snapshot при сбоях CDN/ASN/DoH.
- Опционально `EVOBGP_CDN_PARTIAL_OK=1` — при сбое одного CDN source без stale cache продолжать refresh остальных (иначе fail модуля).
## Рекомендуется
+38
View File
@@ -0,0 +1,38 @@
package config
import (
"fmt"
"os"
"strings"
)
// ProductionMode reports whether EVOBGP_PRODUCTION / EVOBGP_ENV=production is set.
func ProductionMode() bool {
if v := strings.TrimSpace(os.Getenv("EVOBGP_PRODUCTION")); v == "1" || strings.EqualFold(v, "true") {
return true
}
env := strings.ToLower(strings.TrimSpace(os.Getenv("EVOBGP_ENV")))
return env == "production" || env == "prod"
}
// ValidateProductionEnforce enforces docs/production-checklist.md hard requirements
// when ProductionMode() is true. Returns an error that should abort process start.
func ValidateProductionEnforce() error {
if !ProductionMode() {
return nil
}
seedDemo := os.Getenv("EVOBGP_SEED_DEMO")
if seedDemo != "0" {
return fmt.Errorf("config: production requires EVOBGP_SEED_DEMO=0 (got %q)", seedDemo)
}
if strings.TrimSpace(os.Getenv("EVOBGP_DEV_INSECURE")) == "1" {
return fmt.Errorf("config: production forbids EVOBGP_DEV_INSECURE=1")
}
if strings.TrimSpace(os.Getenv("EVOBGP_BUNDLE_SEED_HEX")) == "" {
return fmt.Errorf("config: production requires EVOBGP_BUNDLE_SEED_HEX")
}
if strings.TrimSpace(os.Getenv("EVOBGP_CDN_ALLOW_PRIVATE")) == "1" {
return fmt.Errorf("config: production forbids EVOBGP_CDN_ALLOW_PRIVATE=1")
}
return nil
}
+31
View File
@@ -0,0 +1,31 @@
package config
import "testing"
func TestValidateProductionEnforce(t *testing.T) {
t.Setenv("EVOBGP_PRODUCTION", "")
t.Setenv("EVOBGP_ENV", "")
if err := ValidateProductionEnforce(); err != nil {
t.Fatalf("non-production: %v", err)
}
t.Setenv("EVOBGP_PRODUCTION", "1")
t.Setenv("EVOBGP_SEED_DEMO", "1")
t.Setenv("EVOBGP_BUNDLE_SEED_HEX", "abcd")
t.Setenv("EVOBGP_DEV_INSECURE", "")
t.Setenv("EVOBGP_CDN_ALLOW_PRIVATE", "")
if err := ValidateProductionEnforce(); err == nil {
t.Fatal("expected error for SEED_DEMO!=0")
}
t.Setenv("EVOBGP_SEED_DEMO", "0")
t.Setenv("EVOBGP_BUNDLE_SEED_HEX", "")
if err := ValidateProductionEnforce(); err == nil {
t.Fatal("expected error for missing BUNDLE_SEED_HEX")
}
t.Setenv("EVOBGP_BUNDLE_SEED_HEX", "bd8fbcd31545aacfdd228203beca8e945ab9a752f2ce5624cf42d9f316389a9d")
if err := ValidateProductionEnforce(); err != nil {
t.Fatalf("valid production: %v", err)
}
}
+1 -1
View File
@@ -314,7 +314,7 @@ func (s *Server) handlePreviewCDNSource(w http.ResponseWriter, r *http.Request)
writeProblem(w, http.StatusUnprocessableEntity, "Unprocessable Entity", "invalid url")
return
}
resp, err := s.cdnHTTP.Do(req)
resp, err := pipeline.UpstreamHTTPDo(r.Context(), s.cdnHTTP, req)
if err != nil {
writeBadGateway(w, "cdn preview fetch", err)
return
+99
View File
@@ -1,6 +1,7 @@
package jobs
import (
"context"
"fmt"
"os"
"sort"
@@ -111,6 +112,42 @@ func (j *Job) mergeMeta(kv map[string]any) {
}
}
// metaString returns a string meta field under lock (safe vs concurrent mergeMeta).
func (j *Job) metaString(key string) string {
j.mu.Lock()
defer j.mu.Unlock()
if j.Meta == nil {
return ""
}
s, _ := j.Meta[key].(string)
return s
}
// metaBool returns a bool meta field under lock.
func (j *Job) metaBool(key string) bool {
j.mu.Lock()
defer j.mu.Unlock()
if j.Meta == nil {
return false
}
b, _ := j.Meta[key].(bool)
return b
}
// metaCopy returns a shallow copy of job meta under lock.
func (j *Job) metaCopy() map[string]any {
j.mu.Lock()
defer j.mu.Unlock()
if j.Meta == nil {
return nil
}
out := make(map[string]any, len(j.Meta))
for k, v := range j.Meta {
out[k] = v
}
return out
}
// statusLocked is used by the worker defer for metrics (any stable terminal or in-flight status).
func (j *Job) statusLocked() string {
j.mu.Lock()
@@ -528,3 +565,65 @@ func (r *Registry) RequestCancel(tenantID, jobID string) (*Job, error) {
j.RequestCancel()
return j, nil
}
// RequestCancelAll requests cancellation of all non-terminal jobs (all tenants).
func (r *Registry) RequestCancelAll() int {
if r == nil {
return 0
}
r.mu.RLock()
defer r.mu.RUnlock()
n := 0
for _, j := range r.byID {
if j == nil {
continue
}
st := j.statusLocked()
if st == StatusSucceeded || st == StatusFailed || st == StatusCancelled {
continue
}
j.RequestCancel()
n++
}
return n
}
// ActiveCount returns the number of queued or running jobs.
func (r *Registry) ActiveCount() int {
if r == nil {
return 0
}
r.mu.RLock()
defer r.mu.RUnlock()
n := 0
for _, j := range r.byID {
if j == nil {
continue
}
st := j.statusLocked()
if st == StatusQueued || st == StatusRunning {
n++
}
}
return n
}
// Drain waits until no queued/running jobs remain or ctx is done.
// Call RequestCancelAll first for a cooperative shutdown.
func (r *Registry) Drain(ctx context.Context) error {
if r == nil {
return nil
}
ticker := time.NewTicker(50 * time.Millisecond)
defer ticker.Stop()
for {
if r.ActiveCount() == 0 {
return nil
}
select {
case <-ctx.Done():
return ctx.Err()
case <-ticker.C:
}
}
}
+3 -4
View File
@@ -21,14 +21,13 @@ func (w *Worker) runMaintenancePolicy(j *Job) {
j.Fail("postgresql not configured")
return
}
policyID, _ := j.Meta["policy_id"].(string)
policyID = strings.TrimSpace(policyID)
policyID := strings.TrimSpace(j.metaString("policy_id"))
if policyID == "" {
j.Fail("missing policy_id in job meta")
return
}
dryRun, _ := j.Meta["dry_run"].(bool)
actor, _ := j.Meta["actor_prefix"].(string)
dryRun := j.metaBool("dry_run")
actor := j.metaString("actor_prefix")
ctx, cancel := j.workContext()
defer cancel()
+3 -3
View File
@@ -93,9 +93,9 @@ func (w *Worker) runPostgresMaint(j *Job, kind string) {
j.Fail("postgresql not configured")
return
}
table, _ := j.Meta["table"].(string)
dryRun, _ := j.Meta["dry_run"].(bool)
actor, _ := j.Meta["actor_prefix"].(string)
table := j.metaString("table")
dryRun := j.metaBool("dry_run")
actor := j.metaString("actor_prefix")
ctx, cancel := j.workContext()
defer cancel()
auditID, _ := pgmonitor.InsertMaintenanceAudit(ctx, w.PgPool, j.TenantID, actor, kind, table, dryRun)
+7 -6
View File
@@ -291,8 +291,8 @@ func (w *Worker) runModuleRefresh(j *Job) {
}
}()
mid, _ := j.Meta["module_id"].(string)
if strings.TrimSpace(mid) == "" {
mid := strings.TrimSpace(j.metaString("module_id"))
if mid == "" {
j.Fail("missing module_id in job meta")
return
}
@@ -325,7 +325,7 @@ func (w *Worker) runTenantRefresh(j *Job) {
}
}()
moduleIDs := moduleIDsFromJobMeta(j.Meta)
moduleIDs := moduleIDsFromJobMeta(j.metaCopy())
if len(moduleIDs) == 0 {
j.Fail("missing module_ids in job meta")
return
@@ -449,8 +449,9 @@ func (w *Worker) enqueueDeployAllSpeakers(j *Job, tenantID, revID string) {
}
func (w *Worker) runDeployApply(j *Job) {
revID, _ := j.Meta["revision_id"].(string)
spk, hasSpeaker := j.Meta["speaker_id"].(string)
revID := strings.TrimSpace(j.metaString("revision_id"))
spk := strings.TrimSpace(j.metaString("speaker_id"))
hasSpeaker := spk != ""
if revID == "" {
j.Fail("missing revision_id in job meta")
return
@@ -571,7 +572,7 @@ func (w *Worker) runDeployApply(j *Job) {
}
func (w *Worker) runRollback(j *Job) {
src, _ := j.Meta["source_revision_id"].(string)
src := strings.TrimSpace(j.metaString("source_revision_id"))
if src == "" {
j.Fail("missing source_revision_id in job meta")
return
+10
View File
@@ -3,6 +3,7 @@ package pipeline
import (
"context"
"fmt"
"log"
"net/http"
"os"
"strings"
@@ -218,12 +219,21 @@ func collectCDNPrefixRows(ctx context.Context, st store.Backend, hc *http.Client
wg.Wait()
var out []store.PrefixRow
var skipped int
for _, r := range results {
if r.err != nil {
if cdnPartialOK() {
log.Printf("pipeline: CDN partial skip source error: %v", r.err)
skipped++
continue
}
return nil, r.err
}
out = append(out, r.rows...)
}
if skipped > 0 && len(out) == 0 && len(valid) > 0 {
return nil, fmt.Errorf("cdn: all %d source(s) failed (partial ok)", len(valid))
}
if len(valid) > 0 {
if err := mergeAllCDNSourcesIntoModuleSnapshot(st, tenantID, mod, priorSnapshot, out); err != nil {
return nil, err
+7
View File
@@ -20,6 +20,13 @@ func staleOnUpstreamError() bool {
return false
}
// cdnPartialOK reports whether a failed CDN source without stale cache should be skipped
// instead of failing the whole module refresh. Opt-in: EVOBGP_CDN_PARTIAL_OK=1.
func cdnPartialOK() bool {
v := strings.TrimSpace(os.Getenv("EVOBGP_CDN_PARTIAL_OK"))
return v == "1" || strings.EqualFold(v, "true")
}
func logStaleUpstream(kind, detail string) {
log.Printf("pipeline: stale upstream fallback (%s): %s", kind, detail)
}
+2 -2
View File
@@ -282,7 +282,7 @@ func resolveDomainWithDOHMessage(ctx context.Context, hc *http.Client, baseURL,
return nil, err
}
req.Header.Set("Accept", "application/dns-message")
resp, err := hc.Do(req)
resp, err := httpclient.DoWithRetry(ctx, hc, req, 3)
if err != nil {
return nil, err
}
@@ -354,7 +354,7 @@ func resolveDomainWithDOHJSON(ctx context.Context, hc *http.Client, baseURL, hos
}
req.Header.Set("Accept", "application/dns-json")
resp, err := hc.Do(req)
resp, err := httpclient.DoWithRetry(ctx, hc, req, 3)
if err != nil {
return nil, err
}
+5
View File
@@ -8,6 +8,11 @@ import (
"evobgp/internal/httpclient"
)
// UpstreamHTTPDo performs an outbound GET/POST with circuit breaker + retries (CDN, previews).
func UpstreamHTTPDo(ctx context.Context, hc *http.Client, req *http.Request) (*http.Response, error) {
return upstreamHTTPDo(ctx, hc, req)
}
func upstreamHTTPDo(ctx context.Context, hc *http.Client, req *http.Request) (*http.Response, error) {
if hc == nil {
hc = httpclient.New(httpclient.DefaultTimeout)