diff --git a/dtmcli/dtmimp/utils.go b/dtmcli/dtmimp/utils.go index f2d77dd..f42ac72 100644 --- a/dtmcli/dtmimp/utils.go +++ b/dtmcli/dtmimp/utils.go @@ -18,6 +18,7 @@ import ( "strconv" "strings" "sync" + "time" "github.com/go-resty/resty/v2" "go.uber.org/zap" @@ -215,13 +216,15 @@ func DBExec(db DB, sql string, values ...interface{}) (affected int64, rerr erro if sql == "" { return 0, nil } + began := time.Now() sql = GetDBSpecial().GetPlaceHoldSQL(sql) r, rerr := db.Exec(sql, values...) + used := time.Since(began) / time.Millisecond if rerr == nil { affected, rerr = r.RowsAffected() - Logf("affected: %d for %s %v", affected, sql, values) + Logf("used: %d ms affected: %d for %s %v", used, affected, sql, values) } else { - LogRedf("exec error: %v for %s %v", rerr, sql, values) + LogRedf("used: %d ms exec error: %v for %s %v", used, rerr, sql, values) } return } diff --git a/examples/base_types.go b/examples/base_types.go index 400be7f..8ebfb5f 100644 --- a/examples/base_types.go +++ b/examples/base_types.go @@ -39,8 +39,8 @@ func (t *TransReq) String() string { func GenTransReq(amount int, outFailed bool, inFailed bool) *TransReq { return &TransReq{ Amount: amount, - TransOutResult: dtmimp.If(outFailed, dtmcli.ResultFailure, dtmcli.ResultSuccess).(string), - TransInResult: dtmimp.If(inFailed, dtmcli.ResultFailure, dtmcli.ResultSuccess).(string), + TransOutResult: dtmimp.If(outFailed, dtmcli.ResultFailure, "").(string), + TransInResult: dtmimp.If(inFailed, dtmcli.ResultFailure, "").(string), } } @@ -48,8 +48,8 @@ func GenTransReq(amount int, outFailed bool, inFailed bool) *TransReq { func GenBusiReq(amount int, outFailed bool, inFailed bool) *BusiReq { return &BusiReq{ Amount: int64(amount), - TransOutResult: dtmimp.If(outFailed, dtmcli.ResultFailure, dtmcli.ResultSuccess).(string), - TransInResult: dtmimp.If(inFailed, dtmcli.ResultFailure, dtmcli.ResultSuccess).(string), + TransOutResult: dtmimp.If(outFailed, dtmcli.ResultFailure, "").(string), + TransInResult: dtmimp.If(inFailed, dtmcli.ResultFailure, "").(string), } }