mirror of https://github.com/dtm-labs/dtm.git
xyctruth
4 years ago
19 changed files with 388 additions and 86 deletions
@ -0,0 +1,138 @@ |
|||
package dtmsvr |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"fmt" |
|||
"time" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/gin-gonic/gin" |
|||
) |
|||
|
|||
const jrpcCodeFailure = -32901 |
|||
const jrpcCodeOngoing = -32902 |
|||
|
|||
type jrpcReq struct { |
|||
Method string `json:"method"` |
|||
Jsonrpc string `json:"jsonrpc"` |
|||
Params interface{} `json:"params"` |
|||
ID string `json:"id"` |
|||
} |
|||
|
|||
func addJrpcRouter(engine *gin.Engine) { |
|||
type jrpcFunc = func(interface{}) interface{} |
|||
handlers := map[string]jrpcFunc{ |
|||
"newGid": jrpcNewGid, |
|||
"prepare": jrpcPrepare, |
|||
"submit": jrpcSubmit, |
|||
"abort": jrpcAbort, |
|||
"registerBranch": jrpcRegisterBranch, |
|||
} |
|||
engine.POST("/api/json-rpc", func(c *gin.Context) { |
|||
began := time.Now() |
|||
var err error |
|||
var req jrpcReq |
|||
var jerr map[string]interface{} |
|||
r := func() interface{} { |
|||
defer dtmimp.P2E(&err) |
|||
err2 := c.BindJSON(&req) |
|||
if err2 != nil { |
|||
jerr = map[string]interface{}{ |
|||
"code": -32700, |
|||
"message": fmt.Sprintf("Parse json error: %s", err2.Error()), |
|||
} |
|||
} else if req.ID == "" || req.Jsonrpc != "2.0" { |
|||
jerr = map[string]interface{}{ |
|||
"code": -32600, |
|||
"message": fmt.Sprintf("Bad json request: %s", dtmimp.MustMarshalString(req)), |
|||
} |
|||
} else if handlers[req.Method] == nil { |
|||
jerr = map[string]interface{}{ |
|||
"code": -32601, |
|||
"message": fmt.Sprintf("Method not found: %s", req.Method), |
|||
} |
|||
} else if handlers[req.Method] != nil { |
|||
return handlers[req.Method](req.Params) |
|||
} |
|||
return nil |
|||
}() |
|||
|
|||
// error maybe returned in r, assign it to err
|
|||
if ne, ok := r.(error); ok && err == nil { |
|||
err = ne |
|||
} |
|||
|
|||
if err != nil { |
|||
if errors.Is(err, dtmcli.ErrFailure) { |
|||
jerr = map[string]interface{}{ |
|||
"code": jrpcCodeFailure, |
|||
"message": err.Error(), |
|||
} |
|||
} else if errors.Is(err, dtmcli.ErrOngoing) { |
|||
jerr = map[string]interface{}{ |
|||
"code": jrpcCodeOngoing, |
|||
"message": err.Error(), |
|||
} |
|||
} else if jerr == nil { |
|||
jerr = map[string]interface{}{ |
|||
"code": -32603, |
|||
"message": err.Error(), |
|||
} |
|||
} |
|||
} |
|||
|
|||
result := map[string]interface{}{ |
|||
"jsonrpc": "2.0", |
|||
"id": req.ID, |
|||
"error": jerr, |
|||
"result": r, |
|||
} |
|||
b, _ := json.Marshal(result) |
|||
cont := string(b) |
|||
if jerr == nil || jerr["code"] == jrpcCodeOngoing { |
|||
logger.Infof("%2dms %d %s %s %s", time.Since(began).Milliseconds(), 200, c.Request.Method, c.Request.RequestURI, cont) |
|||
} else { |
|||
logger.Errorf("%2dms %d %s %s %s", time.Since(began).Milliseconds(), 200, c.Request.Method, c.Request.RequestURI, cont) |
|||
} |
|||
c.JSON(200, result) |
|||
}) |
|||
} |
|||
|
|||
// TransFromJrpcParams construct TransGlobal from jrpc params
|
|||
func TransFromJrpcParams(params interface{}) *TransGlobal { |
|||
t := TransGlobal{} |
|||
dtmimp.MustRemarshal(params, &t) |
|||
t.setupPayloads() |
|||
return &t |
|||
} |
|||
|
|||
func jrpcNewGid(interface{}) interface{} { |
|||
return map[string]interface{}{"gid": GenGid()} |
|||
} |
|||
|
|||
func jrpcPrepare(params interface{}) interface{} { |
|||
return svcPrepare(TransFromJrpcParams(params)) |
|||
} |
|||
|
|||
func jrpcSubmit(params interface{}) interface{} { |
|||
return svcSubmit(TransFromJrpcParams(params)) |
|||
} |
|||
|
|||
func jrpcAbort(params interface{}) interface{} { |
|||
return svcAbort(TransFromJrpcParams(params)) |
|||
} |
|||
|
|||
func jrpcRegisterBranch(params interface{}) interface{} { |
|||
data := map[string]string{} |
|||
dtmimp.MustRemarshal(params, &data) |
|||
branch := TransBranch{ |
|||
Gid: data["gid"], |
|||
BranchID: data["branch_id"], |
|||
Status: dtmcli.StatusPrepared, |
|||
BinData: []byte(data["data"]), |
|||
} |
|||
return svcRegisterBranch(data["trans_type"], &branch, data) |
|||
} |
|||
@ -0,0 +1,71 @@ |
|||
/* |
|||
* Copyright (c) 2021 yedf. All rights reserved. |
|||
* Use of this source code is governed by a BSD-style |
|||
* license that can be found in the LICENSE file. |
|||
*/ |
|||
|
|||
package test |
|||
|
|||
import ( |
|||
"testing" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
"github.com/dtm-labs/dtm/test/busi" |
|||
"github.com/stretchr/testify/assert" |
|||
) |
|||
|
|||
func TestMsgJrpcNormal(t *testing.T) { |
|||
msg := genJrpcMsg(dtmimp.GetFuncName()) |
|||
msg.Submit() |
|||
assert.Equal(t, StatusSubmitted, getTransStatus(msg.Gid)) |
|||
waitTransProcessed(msg.Gid) |
|||
assert.Equal(t, []string{StatusSucceed, StatusSucceed}, getBranchesStatus(msg.Gid)) |
|||
assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid)) |
|||
} |
|||
|
|||
func TestMsgJrpcRepeated(t *testing.T) { |
|||
msg := genJrpcMsg(dtmimp.GetFuncName()) |
|||
msg.Submit() |
|||
assert.Equal(t, StatusSubmitted, getTransStatus(msg.Gid)) |
|||
waitTransProcessed(msg.Gid) |
|||
assert.Equal(t, []string{StatusSucceed, StatusSucceed}, getBranchesStatus(msg.Gid)) |
|||
assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid)) |
|||
err := msg.Submit() |
|||
assert.Error(t, err) |
|||
} |
|||
func TestMsgJprcAbnormal(t *testing.T) { |
|||
id := "no-use" |
|||
resp, err := dtmcli.GetRestyClient().R().SetBody("hello").Post(dtmutil.DefaultJrpcServer) |
|||
assert.Nil(t, err) |
|||
assert.Contains(t, resp.String(), "-32700") |
|||
|
|||
resp, err = dtmcli.GetRestyClient().R().SetBody(map[string]string{ |
|||
"jsonrpc": "1.0", |
|||
"method": "newGid", |
|||
"params": "", |
|||
"id": id, |
|||
}).Post(dtmutil.DefaultJrpcServer) |
|||
assert.Nil(t, err) |
|||
assert.Contains(t, resp.String(), "-32600") |
|||
|
|||
resp, err = dtmcli.GetRestyClient().R().SetBody(map[string]string{ |
|||
"jsonrpc": "2.0", |
|||
"method": "not-exists", |
|||
"params": "", |
|||
"id": id, |
|||
}).Post(dtmutil.DefaultJrpcServer) |
|||
assert.Nil(t, err) |
|||
assert.Contains(t, resp.String(), "-32601") |
|||
} |
|||
|
|||
func genJrpcMsg(gid string) *dtmcli.Msg { |
|||
req := busi.GenTransReq(30, false, false) |
|||
msg := dtmcli.NewMsg(dtmutil.DefaultJrpcServer, gid). |
|||
Add(busi.Busi+"/TransOut", &req). |
|||
Add(busi.Busi+"/TransIn", &req) |
|||
msg.QueryPrepared = busi.Busi + "/QueryPrepared" |
|||
msg.Protocol = "json-rpc" |
|||
return msg |
|||
} |
|||
Loading…
Reference in new issue