mirror of https://github.com/dtm-labs/dtm.git
csharpjavadistributed-transactionsdtmgogolangmicroservicenodejsphpdatabasesagaseatatcctransactiontransactionsxapythondistributed
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.5 KiB
81 lines
2.5 KiB
/*
|
|
* 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 dtmsvr
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/dtm-labs/dtm/dtmcli"
|
|
"github.com/dtm-labs/dtm/dtmcli/dtmimp"
|
|
"github.com/dtm-labs/dtm/dtmutil"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
)
|
|
|
|
func addRoute(engine *gin.Engine) {
|
|
engine.GET("/api/dtmsvr/newGid", dtmutil.WrapHandler2(newGid))
|
|
engine.POST("/api/dtmsvr/prepare", dtmutil.WrapHandler2(prepare))
|
|
engine.POST("/api/dtmsvr/submit", dtmutil.WrapHandler2(submit))
|
|
engine.POST("/api/dtmsvr/abort", dtmutil.WrapHandler2(abort))
|
|
engine.POST("/api/dtmsvr/registerBranch", dtmutil.WrapHandler2(registerBranch))
|
|
engine.POST("/api/dtmsvr/registerXaBranch", dtmutil.WrapHandler2(registerBranch)) // compatible for old sdk
|
|
engine.POST("/api/dtmsvr/registerTccBranch", dtmutil.WrapHandler2(registerBranch)) // compatible for old sdk
|
|
engine.GET("/api/dtmsvr/query", dtmutil.WrapHandler2(query))
|
|
engine.GET("/api/dtmsvr/all", dtmutil.WrapHandler2(all))
|
|
|
|
// add prometheus exporter
|
|
h := promhttp.Handler()
|
|
engine.GET("/api/metrics", func(c *gin.Context) {
|
|
h.ServeHTTP(c.Writer, c.Request)
|
|
})
|
|
}
|
|
|
|
func newGid(c *gin.Context) interface{} {
|
|
return map[string]interface{}{"gid": GenGid(), "dtm_result": dtmcli.ResultSuccess}
|
|
}
|
|
|
|
func prepare(c *gin.Context) interface{} {
|
|
return svcPrepare(TransFromContext(c))
|
|
}
|
|
|
|
func submit(c *gin.Context) interface{} {
|
|
return svcSubmit(TransFromContext(c))
|
|
}
|
|
|
|
func abort(c *gin.Context) interface{} {
|
|
return svcAbort(TransFromContext(c))
|
|
}
|
|
|
|
func registerBranch(c *gin.Context) interface{} {
|
|
data := map[string]string{}
|
|
err := c.BindJSON(&data)
|
|
e2p(err)
|
|
branch := TransBranch{
|
|
Gid: data["gid"],
|
|
BranchID: data["branch_id"],
|
|
Status: dtmcli.StatusPrepared,
|
|
BinData: []byte(data["data"]),
|
|
}
|
|
return svcRegisterBranch(data["trans_type"], &branch, data)
|
|
}
|
|
|
|
func query(c *gin.Context) interface{} {
|
|
gid := c.Query("gid")
|
|
if gid == "" {
|
|
return errors.New("no gid specified")
|
|
}
|
|
trans := GetStore().FindTransGlobalStore(gid)
|
|
branches := GetStore().FindBranches(gid)
|
|
return map[string]interface{}{"transaction": trans, "branches": branches}
|
|
}
|
|
|
|
func all(c *gin.Context) interface{} {
|
|
position := c.Query("position")
|
|
slimit := dtmimp.OrString(c.Query("limit"), "100")
|
|
globals := GetStore().ScanTransGlobalStores(&position, int64(dtmimp.MustAtoi(slimit)))
|
|
return map[string]interface{}{"transactions": globals, "next_position": position}
|
|
}
|
|
|