mirror of https://github.com/dtm-labs/dtm.git
yedf2
5 years ago
15 changed files with 326 additions and 331 deletions
@ -0,0 +1,189 @@ |
|||
package dtmcli |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"fmt" |
|||
"os" |
|||
"path" |
|||
"path/filepath" |
|||
"runtime" |
|||
"strconv" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/go-resty/resty/v2" |
|||
) |
|||
|
|||
// P2E panic to error
|
|||
func P2E(perr *error) { |
|||
if x := recover(); x != nil { |
|||
if e, ok := x.(error); ok { |
|||
*perr = e |
|||
} else { |
|||
panic(x) |
|||
} |
|||
} |
|||
} |
|||
|
|||
// E2P error to panic
|
|||
func E2P(err error) { |
|||
if err != nil { |
|||
panic(err) |
|||
} |
|||
} |
|||
|
|||
// CatchP catch panic to error
|
|||
func CatchP(f func()) (rerr error) { |
|||
defer P2E(&rerr) |
|||
f() |
|||
return nil |
|||
} |
|||
|
|||
// PanicIf name is clear
|
|||
func PanicIf(cond bool, err error) { |
|||
if cond { |
|||
panic(err) |
|||
} |
|||
} |
|||
|
|||
// MustAtoi 走must逻辑
|
|||
func MustAtoi(s string) int { |
|||
r, err := strconv.Atoi(s) |
|||
if err != nil { |
|||
E2P(errors.New("convert to int error: " + s)) |
|||
} |
|||
return r |
|||
} |
|||
|
|||
// OrString return the first not empty string
|
|||
func OrString(ss ...string) string { |
|||
for _, s := range ss { |
|||
if s != "" { |
|||
return s |
|||
} |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
// If ternary operator
|
|||
func If(condition bool, trueObj interface{}, falseObj interface{}) interface{} { |
|||
if condition { |
|||
return trueObj |
|||
} |
|||
return falseObj |
|||
} |
|||
|
|||
// MustMarshal checked version for marshal
|
|||
func MustMarshal(v interface{}) []byte { |
|||
b, err := json.Marshal(v) |
|||
E2P(err) |
|||
return b |
|||
} |
|||
|
|||
// MustMarshalString string version of MustMarshal
|
|||
func MustMarshalString(v interface{}) string { |
|||
return string(MustMarshal(v)) |
|||
} |
|||
|
|||
// MustUnmarshal checked version for unmarshal
|
|||
func MustUnmarshal(b []byte, obj interface{}) { |
|||
err := json.Unmarshal(b, obj) |
|||
E2P(err) |
|||
} |
|||
|
|||
// MustUnmarshalString string version of MustUnmarshal
|
|||
func MustUnmarshalString(s string, obj interface{}) { |
|||
MustUnmarshal([]byte(s), obj) |
|||
} |
|||
|
|||
// MustRemarshal marshal and unmarshal, and check error
|
|||
func MustRemarshal(from interface{}, to interface{}) { |
|||
b, err := json.Marshal(from) |
|||
E2P(err) |
|||
err = json.Unmarshal(b, to) |
|||
E2P(err) |
|||
} |
|||
|
|||
// Logf 输出日志
|
|||
func Logf(format string, args ...interface{}) { |
|||
msg := fmt.Sprintf(format, args...) |
|||
n := time.Now() |
|||
ts := fmt.Sprintf("%d-%02d-%02d %02d:%02d:%02d.%03d", n.Year(), n.Month(), n.Day(), n.Hour(), n.Minute(), n.Second(), n.Nanosecond()/1000000) |
|||
var file string |
|||
var line int |
|||
for i := 1; ; i++ { |
|||
_, file, line, _ = runtime.Caller(i) |
|||
if strings.Contains(file, "dtm") { |
|||
break |
|||
} |
|||
} |
|||
fmt.Printf("%s %s:%d %s\n", ts, path.Base(file), line, msg) |
|||
} |
|||
|
|||
// LogRedf 采用红色打印错误类信息
|
|||
func LogRedf(fmt string, args ...interface{}) { |
|||
Logf("\x1b[31m\n"+fmt+"\x1b[0m\n", args...) |
|||
} |
|||
|
|||
// RestyClient the resty object
|
|||
var RestyClient = resty.New() |
|||
|
|||
func init() { |
|||
// RestyClient.SetTimeout(3 * time.Second)
|
|||
// RestyClient.SetRetryCount(2)
|
|||
// RestyClient.SetRetryWaitTime(1 * time.Second)
|
|||
RestyClient.OnBeforeRequest(func(c *resty.Client, r *resty.Request) error { |
|||
r.URL = MayReplaceLocalhost(r.URL) |
|||
Logf("requesting: %s %s %v %v", r.Method, r.URL, r.Body, r.QueryParam) |
|||
return nil |
|||
}) |
|||
RestyClient.OnAfterResponse(func(c *resty.Client, resp *resty.Response) error { |
|||
r := resp.Request |
|||
Logf("requested: %s %s %s", r.Method, r.URL, resp.String()) |
|||
return nil |
|||
}) |
|||
} |
|||
|
|||
// CheckRestySuccess panic if error or resp not success
|
|||
func CheckRestySuccess(resp *resty.Response, err error) { |
|||
E2P(err) |
|||
if !strings.Contains(resp.String(), "SUCCESS") { |
|||
panic(fmt.Errorf("resty response not success: %s", resp.String())) |
|||
} |
|||
} |
|||
|
|||
// MustGetwd must version of os.Getwd
|
|||
func MustGetwd() string { |
|||
wd, err := os.Getwd() |
|||
E2P(err) |
|||
return wd |
|||
} |
|||
|
|||
// GetCurrentCodeDir name is clear
|
|||
func GetCurrentCodeDir() string { |
|||
_, file, _, _ := runtime.Caller(1) |
|||
return filepath.Dir(file) |
|||
} |
|||
|
|||
// GetProjectDir name is clear
|
|||
func GetProjectDir() string { |
|||
_, file, _, _ := runtime.Caller(1) |
|||
for ; !strings.HasSuffix(file, "/dtm"); file = filepath.Dir(file) { |
|||
} |
|||
return file |
|||
} |
|||
|
|||
// GetFuncName get current call func name
|
|||
func GetFuncName() string { |
|||
pc, _, _, _ := runtime.Caller(1) |
|||
return runtime.FuncForPC(pc).Name() |
|||
} |
|||
|
|||
// MayReplaceLocalhost when run in docker compose, change localhost to host.docker.internal for accessing host network
|
|||
func MayReplaceLocalhost(host string) string { |
|||
if os.Getenv("IS_DOCKER_COMPOSE") != "" { |
|||
return strings.Replace(host, "localhost", "host.docker.internal", 1) |
|||
} |
|||
return host |
|||
} |
|||
Loading…
Reference in new issue