mirror of https://github.com/dtm-labs/dtm.git
liulei
4 years ago
23 changed files with 354 additions and 138 deletions
@ -0,0 +1,110 @@ |
|||
package dtmsvr |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"fmt" |
|||
"net/http" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/gin-gonic/gin" |
|||
) |
|||
|
|||
type jsonRPCReq struct { |
|||
Method string `json:"method"` |
|||
Jsonrpc string `json:"jsonrpc"` |
|||
Params interface{} `json:"params"` |
|||
ID string `json:"id"` |
|||
} |
|||
|
|||
func addJSONRPCRouter(engine *gin.Engine) { |
|||
engine.POST("/", dispatcher) |
|||
} |
|||
|
|||
func dispatcher(c *gin.Context) { |
|||
req := new(jsonRPCReq) |
|||
err := c.BindJSON(req) |
|||
logger.Infof("request:%s\n", req) |
|||
if err != nil { |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": nil, "error": map[string]interface{}{"code": -32700, "message": "Parse error"}}) |
|||
return |
|||
} |
|||
if req.Method == "dtmserver.NewGid" { |
|||
res := jsonRPCNewGid() |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": res, "error": err}) |
|||
return |
|||
} |
|||
|
|||
if req.Method == "dtmserver.Prepare" { |
|||
res := jsonRPCPrepare(req.Params) |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": res, "error": nil}) |
|||
return |
|||
} |
|||
|
|||
if req.Method == "dtmserver.Submit" { |
|||
res := jsonRPCSubmit(req.Params) |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": res, "error": nil}) |
|||
return |
|||
} |
|||
|
|||
if req.Method == "dtmserver.Abort" { |
|||
res := jsonRPCAbort(req.Params) |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": res, "error": nil}) |
|||
return |
|||
} |
|||
|
|||
if req.Method == "dtmserver.RegisterBranch" { |
|||
res := jsonRPCRegisterBranch(req.Params) |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": res, "error": nil}) |
|||
return |
|||
} |
|||
c.JSON(http.StatusOK, gin.H{"id": req.ID, "result": nil, "error": map[string]interface{}{"code": -32601, "message": "Method not found"}}) |
|||
} |
|||
|
|||
func jsonRPCNewGid() interface{} { |
|||
return map[string]interface{}{"gid": GenGid(), "dtm_result": dtmcli.ResultSuccess} |
|||
} |
|||
|
|||
func jsonRPCPrepare(params interface{}) interface{} { |
|||
res := svcPrepare(TransFromJSONRPCContext(params)) |
|||
if res == nil { |
|||
return map[string]string{"dtm_result": "SUCCESS"} |
|||
} |
|||
return map[string]string{"dtm_result": "FAILURE", "message": fmt.Sprintf("%v", res)} |
|||
} |
|||
|
|||
func jsonRPCSubmit(params interface{}) interface{} { |
|||
res := svcSubmit(TransFromJSONRPCContext(params)) |
|||
if res == nil { |
|||
return map[string]string{"dtm_result": "SUCCESS"} |
|||
} |
|||
return map[string]string{"dtm_result": "FAILURE", "message": fmt.Sprintf("%v", res)} |
|||
} |
|||
|
|||
func jsonRPCAbort(params interface{}) interface{} { |
|||
res := svcAbort(TransFromJSONRPCContext(params)) |
|||
if res == nil { |
|||
return map[string]string{"dtm_result": "SUCCESS"} |
|||
} |
|||
return map[string]string{"dtm_result": "FAILURE", "message": fmt.Sprintf("%v", res)} |
|||
} |
|||
|
|||
func jsonRPCRegisterBranch(params interface{}) interface{} { |
|||
data := map[string]string{} |
|||
paramsJSON, _ := json.Marshal(params) |
|||
err := json.Unmarshal(paramsJSON, &data) |
|||
if err != nil { |
|||
return map[string]string{"dtm_result": "FAILURE", "message": err.Error()} |
|||
} |
|||
branch := TransBranch{ |
|||
Gid: data["gid"], |
|||
BranchID: data["branch_id"], |
|||
Status: dtmcli.StatusPrepared, |
|||
BinData: []byte(data["data"]), |
|||
} |
|||
res := svcRegisterBranch(data["trans_type"], &branch, data) |
|||
if res == nil { |
|||
return map[string]string{"dtm_result": "SUCCESS"} |
|||
} |
|||
return map[string]string{"dtm_result": "FAILURE", "message": res.Error()} |
|||
} |
|||
@ -0,0 +1,39 @@ |
|||
package test |
|||
|
|||
import ( |
|||
"testing" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmsvr" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
"github.com/dtm-labs/dtm/test/busi" |
|||
"github.com/stretchr/testify/assert" |
|||
) |
|||
|
|||
func genMsgDelay(gid string) *dtmcli.Msg { |
|||
req := busi.GenTransReq(30, false, false) |
|||
msg := dtmcli.NewMsg(dtmutil.DefaultHTTPServer, gid). |
|||
Add(busi.Busi+"/TransOut", &req). |
|||
Add(busi.Busi+"/TransIn", &req).SetDelay(10) |
|||
msg.QueryPrepared = busi.Busi + "/QueryPrepared" |
|||
return msg |
|||
} |
|||
|
|||
func TestMsgDelayNormal(t *testing.T) { |
|||
gid := dtmimp.GetFuncName() |
|||
msg := genMsgDelay(gid) |
|||
submitForwardCron(0, func() { |
|||
msg.Submit() |
|||
waitTransProcessed(msg.Gid) |
|||
}) |
|||
|
|||
dtmsvr.NowForwardDuration = 0 |
|||
assert.Equal(t, []string{StatusPrepared, StatusPrepared}, getBranchesStatus(msg.Gid)) |
|||
assert.Equal(t, StatusSubmitted, getTransStatus(msg.Gid)) |
|||
cronTransOnceForwardCron(t, "", 0) |
|||
cronTransOnceForwardCron(t, "", 8) |
|||
cronTransOnceForwardCron(t, gid, 12) |
|||
assert.Equal(t, []string{StatusSucceed, StatusSucceed}, getBranchesStatus(msg.Gid)) |
|||
assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid)) |
|||
} |
|||
Loading…
Reference in new issue