1 Star 0 Fork 0

CNCF/devstatscode

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
error.go 3.16 KB
一键复制 编辑 原始数据 按行查看 历史
Łukasz Gryglicki 提交于 2024-05-08 14:38 +08:00 . Fix a reconnect bug
package devstatscode
import (
"fmt"
"os"
"runtime/debug"
"strings"
"time"
"github.com/lib/pq"
)
// FatalOnError displays error message (if error present) and exits program
func FatalOnError(err error) string {
if err != nil {
tm := time.Now()
switch e := err.(type) {
case *pq.Error:
errName := e.Code.Name()
if errName == "too_many_connections" {
fmt.Fprintf(os.Stderr, "PqError: code=%s, name=%s, detail=%s\n", e.Code, errName, e.Detail)
fmt.Fprintf(os.Stderr, "Warning: too many postgres connections: %+v: '%s'\n", tm, err.Error())
return Retry
} else if errName == "cannot_connect_now" {
fmt.Fprintf(os.Stderr, "PqError: code=%s, name=%s, detail=%s\n", e.Code, errName, e.Detail)
fmt.Fprintf(os.Stderr, "Warning: DB shutting down: %+v: '%s', sleeping 15 minutes to settle\n", tm, err.Error())
time.Sleep(time.Duration(900) * time.Second)
tm = time.Now()
fmt.Fprintf(os.Stderr, "Warning: DB shutting down: %+v: '%s', waited 15 minutes, retrying\n", tm, err.Error())
return Reconnect
}
Printf("PqError: code=%s, name=%s, detail=%s\n", e.Code, errName, e.Detail)
fmt.Fprintf(os.Stderr, "PqError: code=%s, name=%s, detail=%s\n", e.Code, errName, e.Detail)
if os.Getenv("DURABLE_PQ") != "" && os.Getenv("DURABLE_PQ") != "0" && os.Getenv("DURABLE_PQ") != "false" {
switch errName {
case "program_limit_exceeded", "undefined_column", "invalid_catalog_name", "character_not_in_repertoire":
Printf("%s error is not retryable, even with DURABLE_PQ\n", errName)
default:
fmt.Fprintf(os.Stderr, "retrying with DURABLE_PQ\n")
return Reconnect
}
}
default:
fmt.Fprintf(os.Stderr, "ErrorType: %T, error: %+v\n", e, e)
fmt.Fprintf(os.Stderr, "ErrorType: %T, error: %+v\n", e, e)
}
if strings.Contains(err.Error(), "driver: bad connection") {
fmt.Fprintf(os.Stderr, "Warning: bad driver, retrying\n")
return Reconnect
}
if strings.Contains(err.Error(), "cannot assign requested address") {
fmt.Fprintf(os.Stderr, "Warning: cannot assign requested address, retrying in 15 minutes\n")
time.Sleep(time.Duration(900) * time.Second)
fmt.Fprintf(os.Stderr, "Warning: cannot assign requested address - waited 15 minutes, retrying\n")
return Reconnect
}
/*
if strings.Contains(err.Error(), "database is closed") {
Printf("Warning: database is closed, retrying\n")
return Reconnect
}
*/
fmt.Fprintf(os.Stderr, "Error(time=%+v):\nError: '%s'\nStacktrace:\n%s\n", tm, err.Error(), string(debug.Stack()))
if os.Getenv("NO_FATAL_DELAY") == "" {
time.Sleep(time.Duration(60) * time.Second)
}
panic(fmt.Sprintf("stacktrace: %+v", err))
}
return OK
}
// Fatalf - it will call FatalOnError using fmt.Errorf with args provided
func Fatalf(f string, a ...interface{}) {
FatalOnError(fmt.Errorf(f, a...))
}
// FatalNoLog displays error message (if error present) and exits program, should be used for very early init state
func FatalNoLog(err error) string {
if err != nil {
tm := time.Now()
fmt.Fprintf(os.Stderr, "Error(time=%+v):\nError: '%s'\nStacktrace:\n", tm, err.Error())
time.Sleep(time.Duration(60) * time.Second)
panic(fmt.Sprintf("stacktrace: %+v", err))
}
return OK
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/cncf/devstatscode.git
git@gitee.com:cncf/devstatscode.git
cncf
devstatscode
devstatscode
master

搜索帮助