mirror of https://github.com/dtm-labs/dtm.git
wondarchan
4 years ago
committed by
GitHub
GitHub
121 changed files with 2364 additions and 2292 deletions
@ -1,33 +0,0 @@ |
|||
{ |
|||
// 使用 IntelliSense 了解相关属性。 |
|||
// 悬停以查看现有属性的描述。 |
|||
// 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387 |
|||
"version": "0.2.0", |
|||
"configurations": [ |
|||
{ |
|||
"name": "Launch", |
|||
"type": "go", |
|||
"request": "launch", |
|||
"mode": "debug", |
|||
"program": "${workspaceFolder}/app/main.go", |
|||
"cwd": "${workspaceFolder}", |
|||
"env": { |
|||
// "GIN_MODE": "release" |
|||
}, |
|||
"args": ["grpc_saga"] |
|||
}, |
|||
{ |
|||
"name": "Test", |
|||
"type": "go", |
|||
"request": "launch", |
|||
"mode": "test", |
|||
"port": 2345, |
|||
"host": "127.0.0.1", |
|||
"program": "${file}", |
|||
"env": { |
|||
// "GIN_MODE": "release" |
|||
}, |
|||
"args": [] |
|||
} |
|||
] |
|||
} |
|||
@ -1,3 +0,0 @@ |
|||
{ |
|||
"go.formatTool": "gofmt" |
|||
} |
|||
@ -1,82 +0,0 @@ |
|||
/* |
|||
* 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 main |
|||
|
|||
import ( |
|||
"fmt" |
|||
"os" |
|||
"strings" |
|||
|
|||
_ "go.uber.org/automaxprocs" |
|||
|
|||
"github.com/dtm-labs/dtm/common" |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/dtmsvr" |
|||
"github.com/dtm-labs/dtm/dtmsvr/storage/registry" |
|||
"github.com/dtm-labs/dtm/examples" |
|||
) |
|||
|
|||
var Version, Commit, Date string |
|||
|
|||
var usage = `dtm is a lightweight distributed transaction manager. |
|||
|
|||
usage: |
|||
dtm [command] |
|||
|
|||
Available commands: |
|||
version print dtm version |
|||
dtmsvr run dtm as a server |
|||
dev create all needed table and run dtm as a server |
|||
bench start bench server |
|||
|
|||
quick_start run quick start example (dtm will create needed table) |
|||
qs same as quick_start |
|||
` |
|||
|
|||
func main() { |
|||
if len(os.Args) == 1 { |
|||
fmt.Println(usage) |
|||
for name := range examples.Samples { |
|||
fmt.Printf("%4s%-18srun a sample includes %s\n", "", name, strings.ReplaceAll(name, "_", " ")) |
|||
} |
|||
return |
|||
} |
|||
if os.Args[1] == "version" { |
|||
fmt.Printf("version: %s commit: %s built at: %s\n", Version, Commit, Date) |
|||
return |
|||
} |
|||
logger.Infof("starting dtm....") |
|||
common.MustLoadConfig() |
|||
if common.Config.ExamplesDB.Driver != "" { |
|||
dtmcli.SetCurrentDBType(common.Config.ExamplesDB.Driver) |
|||
} |
|||
if os.Args[1] != "dtmsvr" { // 实际线上运行，只启动dtmsvr，不准备table相关的数据
|
|||
registry.WaitStoreUp() |
|||
dtmsvr.PopulateDB(true) |
|||
examples.PopulateDB(true) |
|||
} |
|||
dtmsvr.StartSvr() // 启动dtmsvr的api服务
|
|||
go dtmsvr.CronExpiredTrans(-1) // 启动dtmsvr的定时过期查询
|
|||
|
|||
switch os.Args[1] { |
|||
case "quick_start", "qs": |
|||
// quick_start 比较独立，单独作为一个例子运行，方便新人上手
|
|||
examples.QsStartSvr() |
|||
examples.QsFireRequest() |
|||
case "dev", "dtmsvr": // do nothing, not fallthrough
|
|||
default: |
|||
// 下面是各类的例子
|
|||
examples.GrpcStartup() |
|||
examples.BaseAppStartup() |
|||
|
|||
sample := examples.Samples[os.Args[1]] |
|||
logger.FatalfIf(sample == nil, "no sample name for %s", os.Args[1]) |
|||
sample.Action() |
|||
} |
|||
select {} |
|||
} |
|||
@ -1,128 +0,0 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"io/ioutil" |
|||
"path/filepath" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"gopkg.in/yaml.v2" |
|||
) |
|||
|
|||
const ( |
|||
DtmMetricsPort = 8889 |
|||
Mysql = "mysql" |
|||
Redis = "redis" |
|||
BoltDb = "boltdb" |
|||
) |
|||
|
|||
// MicroService config type for micro service
|
|||
type MicroService struct { |
|||
Driver string `yaml:"Driver" default:"default"` |
|||
Target string `yaml:"Target"` |
|||
EndPoint string `yaml:"EndPoint"` |
|||
} |
|||
|
|||
type Store struct { |
|||
Driver string `yaml:"Driver" default:"boltdb"` |
|||
Host string `yaml:"Host"` |
|||
Port int64 `yaml:"Port"` |
|||
User string `yaml:"User"` |
|||
Password string `yaml:"Password"` |
|||
MaxOpenConns int64 `yaml:"MaxOpenConns" default:"500"` |
|||
MaxIdleConns int64 `yaml:"MaxIdleConns" default:"500"` |
|||
ConnMaxLifeTime int64 `yaml:"ConnMaxLifeTime" default:"5"` |
|||
DataExpire int64 `yaml:"DataExpire" default:"604800"` // Trans data will expire in 7 days. only for redis/boltdb.
|
|||
RedisPrefix string `yaml:"RedisPrefix" default:"{a}"` // Redis storage prefix. store data to only one slot in cluster
|
|||
} |
|||
|
|||
func (s *Store) IsDB() bool { |
|||
return s.Driver == dtmcli.DBTypeMysql || s.Driver == dtmcli.DBTypePostgres |
|||
} |
|||
|
|||
func (s *Store) GetDBConf() dtmcli.DBConf { |
|||
return dtmcli.DBConf{ |
|||
Driver: s.Driver, |
|||
Host: s.Host, |
|||
Port: s.Port, |
|||
User: s.User, |
|||
Password: s.Password, |
|||
} |
|||
} |
|||
|
|||
type configType struct { |
|||
Store Store `yaml:"Store"` |
|||
TransCronInterval int64 `yaml:"TransCronInterval" default:"3"` |
|||
TimeoutToFail int64 `yaml:"TimeoutToFail" default:"35"` |
|||
RetryInterval int64 `yaml:"RetryInterval" default:"10"` |
|||
HttpPort int64 `yaml:"HttpPort" default:"36789"` |
|||
GrpcPort int64 `yaml:"GrpcPort" default:"36790"` |
|||
MicroService MicroService `yaml:"MicroService"` |
|||
UpdateBranchSync int64 `yaml:"UpdateBranchSync"` |
|||
LogLevel string `yaml:"LogLevel" default:"info"` |
|||
ExamplesDB dtmcli.DBConf `yaml:"ExamplesDB"` |
|||
} |
|||
|
|||
// Config 配置
|
|||
var Config = configType{} |
|||
|
|||
func MustLoadConfig() { |
|||
loadFromEnv("", &Config) |
|||
cont := []byte{} |
|||
for d := MustGetwd(); d != "" && d != "/"; d = filepath.Dir(d) { |
|||
cont1, err := ioutil.ReadFile(d + "/conf.yml") |
|||
if err != nil { |
|||
cont1, err = ioutil.ReadFile(d + "/conf.sample.yml") |
|||
} |
|||
if cont1 != nil { |
|||
cont = cont1 |
|||
break |
|||
} |
|||
} |
|||
if len(cont) != 0 { |
|||
err := yaml.UnmarshalStrict(cont, &Config) |
|||
logger.FatalIfError(err) |
|||
} |
|||
scont, err := json.MarshalIndent(&Config, "", " ") |
|||
logger.FatalIfError(err) |
|||
logger.Debugf("config is: \n%s", scont) |
|||
err = checkConfig() |
|||
logger.FatalfIf(err != nil, `config error: '%v'. |
|||
check you env, and conf.yml/conf.sample.yml in current and parent path: %s. |
|||
please visit http://d.dtm.pub to see the config document.
|
|||
loaded config is: |
|||
%v`, err, MustGetwd(), Config) |
|||
} |
|||
|
|||
func checkConfig() error { |
|||
if Config.RetryInterval < 10 { |
|||
return errors.New("RetryInterval should not be less than 10") |
|||
} |
|||
if Config.TimeoutToFail < Config.RetryInterval { |
|||
return errors.New("TimeoutToFail should not be less than RetryInterval") |
|||
} |
|||
switch Config.Store.Driver { |
|||
case BoltDb: |
|||
return nil |
|||
case Mysql: |
|||
if Config.Store.Host == "" { |
|||
return errors.New("Db host not valid ") |
|||
} |
|||
if Config.Store.Port == 0 { |
|||
return errors.New("Db port not valid ") |
|||
} |
|||
if Config.Store.User == "" { |
|||
return errors.New("Db user not valid ") |
|||
} |
|||
case Redis: |
|||
if Config.Store.Host == "" { |
|||
return errors.New("Redis host not valid ") |
|||
} |
|||
if Config.Store.Port == 0 { |
|||
return errors.New("Redis port not valid ") |
|||
} |
|||
} |
|||
return nil |
|||
} |
|||
@ -1,49 +0,0 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"errors" |
|||
"os" |
|||
"testing" |
|||
|
|||
"github.com/go-playground/assert/v2" |
|||
) |
|||
|
|||
func TestLoadFromEnv(t *testing.T) { |
|||
assert.Equal(t, "MICRO_SERVICE_DRIVER", toUnderscoreUpper("MicroService_Driver")) |
|||
|
|||
ms := MicroService{} |
|||
os.Setenv("T_DRIVER", "d1") |
|||
loadFromEnv("T", &ms) |
|||
assert.Equal(t, "d1", ms.Driver) |
|||
} |
|||
|
|||
func TestCheckConfig(t *testing.T) { |
|||
config := &Config |
|||
retryIntervalErr := checkConfig() |
|||
retryIntervalExpect := errors.New("RetryInterval should not be less than 10") |
|||
assert.Equal(t, retryIntervalErr, retryIntervalExpect) |
|||
|
|||
config.RetryInterval = 10 |
|||
timeoutToFailErr := checkConfig() |
|||
timeoutToFailExpect := errors.New("TimeoutToFail should not be less than RetryInterval") |
|||
assert.Equal(t, timeoutToFailErr, timeoutToFailExpect) |
|||
|
|||
config.TimeoutToFail = 20 |
|||
driverErr := checkConfig() |
|||
assert.Equal(t, driverErr, nil) |
|||
|
|||
config.Store = Store{Driver: Mysql} |
|||
hostErr := checkConfig() |
|||
hostExpect := errors.New("Db host not valid ") |
|||
assert.Equal(t, hostErr, hostExpect) |
|||
|
|||
config.Store = Store{Driver: Mysql, Host: "127.0.0.1"} |
|||
portErr := checkConfig() |
|||
portExpect := errors.New("Db port not valid ") |
|||
assert.Equal(t, portErr, portExpect) |
|||
|
|||
config.Store = Store{Driver: Mysql, Host: "127.0.0.1", Port: 8686} |
|||
userErr := checkConfig() |
|||
userExpect := errors.New("Db user not valid ") |
|||
assert.Equal(t, userErr, userExpect) |
|||
} |
|||
@ -1,67 +0,0 @@ |
|||
/* |
|||
* 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 common |
|||
|
|||
import ( |
|||
"testing" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/stretchr/testify/assert" |
|||
) |
|||
|
|||
func TestGeneralDB(t *testing.T) { |
|||
MustLoadConfig() |
|||
if Config.Store.IsDB() { |
|||
testSql(t) |
|||
testDbAlone(t) |
|||
} |
|||
} |
|||
|
|||
func testSql(t *testing.T) { |
|||
db := DbGet(Config.Store.GetDBConf()) |
|||
err := func() (rerr error) { |
|||
defer dtmimp.P2E(&rerr) |
|||
db.Must().Exec("select a") |
|||
return nil |
|||
}() |
|||
assert.NotEqual(t, nil, err) |
|||
} |
|||
|
|||
func testDbAlone(t *testing.T) { |
|||
db, err := dtmimp.StandaloneDB(Config.Store.GetDBConf()) |
|||
assert.Nil(t, err) |
|||
_, err = dtmimp.DBExec(db, "select 1") |
|||
assert.Equal(t, nil, err) |
|||
_, err = dtmimp.DBExec(db, "") |
|||
assert.Equal(t, nil, err) |
|||
db.Close() |
|||
_, err = dtmimp.DBExec(db, "select 1") |
|||
assert.NotEqual(t, nil, err) |
|||
} |
|||
|
|||
func TestConfig(t *testing.T) { |
|||
testConfigStringField(&Config.Store.Driver, "", t) |
|||
testConfigStringField(&Config.Store.User, "", t) |
|||
testConfigIntField(&Config.RetryInterval, 9, t) |
|||
testConfigIntField(&Config.TimeoutToFail, 9, t) |
|||
} |
|||
|
|||
func testConfigStringField(fd *string, val string, t *testing.T) { |
|||
old := *fd |
|||
*fd = val |
|||
str := checkConfig() |
|||
assert.NotEqual(t, "", str) |
|||
*fd = old |
|||
} |
|||
|
|||
func testConfigIntField(fd *int64, val int64, t *testing.T) { |
|||
old := *fd |
|||
*fd = val |
|||
str := checkConfig() |
|||
assert.NotEqual(t, "", str) |
|||
*fd = old |
|||
} |
|||
@ -0,0 +1,85 @@ |
|||
package config |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"io/ioutil" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"gopkg.in/yaml.v2" |
|||
) |
|||
|
|||
const ( |
|||
DtmMetricsPort = 8889 |
|||
Mysql = "mysql" |
|||
Redis = "redis" |
|||
BoltDb = "boltdb" |
|||
) |
|||
|
|||
// MicroService config type for micro service
|
|||
type MicroService struct { |
|||
Driver string `yaml:"Driver" default:"default"` |
|||
Target string `yaml:"Target"` |
|||
EndPoint string `yaml:"EndPoint"` |
|||
} |
|||
|
|||
type Store struct { |
|||
Driver string `yaml:"Driver" default:"boltdb"` |
|||
Host string `yaml:"Host"` |
|||
Port int64 `yaml:"Port"` |
|||
User string `yaml:"User"` |
|||
Password string `yaml:"Password"` |
|||
MaxOpenConns int64 `yaml:"MaxOpenConns" default:"500"` |
|||
MaxIdleConns int64 `yaml:"MaxIdleConns" default:"500"` |
|||
ConnMaxLifeTime int64 `yaml:"ConnMaxLifeTime" default:"5"` |
|||
DataExpire int64 `yaml:"DataExpire" default:"604800"` // Trans data will expire in 7 days. only for redis/boltdb.
|
|||
RedisPrefix string `yaml:"RedisPrefix" default:"{a}"` // Redis storage prefix. store data to only one slot in cluster
|
|||
TransGlobalTable string `yaml:"TransGlobalTable" default:"dtm.trans_global"` |
|||
TransBranchOpTable string `yaml:"BranchTransOpTable" default:"dtm.trans_branch_op"` |
|||
} |
|||
|
|||
func (s *Store) IsDB() bool { |
|||
return s.Driver == dtmcli.DBTypeMysql || s.Driver == dtmcli.DBTypePostgres |
|||
} |
|||
|
|||
func (s *Store) GetDBConf() dtmcli.DBConf { |
|||
return dtmcli.DBConf{ |
|||
Driver: s.Driver, |
|||
Host: s.Host, |
|||
Port: s.Port, |
|||
User: s.User, |
|||
Password: s.Password, |
|||
} |
|||
} |
|||
|
|||
type configType struct { |
|||
Store Store `yaml:"Store"` |
|||
TransCronInterval int64 `yaml:"TransCronInterval" default:"3"` |
|||
TimeoutToFail int64 `yaml:"TimeoutToFail" default:"35"` |
|||
RetryInterval int64 `yaml:"RetryInterval" default:"10"` |
|||
HttpPort int64 `yaml:"HttpPort" default:"36789"` |
|||
GrpcPort int64 `yaml:"GrpcPort" default:"36790"` |
|||
MicroService MicroService `yaml:"MicroService"` |
|||
UpdateBranchSync int64 `yaml:"UpdateBranchSync"` |
|||
LogLevel string `yaml:"LogLevel" default:"info"` |
|||
} |
|||
|
|||
// Config 配置
|
|||
var Config = configType{} |
|||
|
|||
// MustLoadConfig load config from env and file
|
|||
func MustLoadConfig(confFile string) { |
|||
loadFromEnv("", &Config) |
|||
if confFile != "" { |
|||
cont, err := ioutil.ReadFile(confFile) |
|||
logger.FatalIfError(err) |
|||
err = yaml.UnmarshalStrict(cont, &Config) |
|||
logger.FatalIfError(err) |
|||
} |
|||
scont, err := json.MarshalIndent(&Config, "", " ") |
|||
logger.FatalIfError(err) |
|||
logger.Infof("config file: %s loaded config is: \n%s", confFile, scont) |
|||
err = checkConfig(&Config) |
|||
logger.FatalfIf(err != nil, `config error: '%v'. |
|||
please visit http://d.dtm.pub to see the config document.`, err)
|
|||
} |
|||
@ -0,0 +1,84 @@ |
|||
package config |
|||
|
|||
import ( |
|||
"errors" |
|||
"os" |
|||
"testing" |
|||
|
|||
"github.com/stretchr/testify/assert" |
|||
) |
|||
|
|||
func TestLoadFromEnv(t *testing.T) { |
|||
assert.Equal(t, "MICRO_SERVICE_DRIVER", toUnderscoreUpper("MicroService_Driver")) |
|||
|
|||
ms := MicroService{} |
|||
os.Setenv("T_DRIVER", "d1") |
|||
loadFromEnv("T", &ms) |
|||
assert.Equal(t, "d1", ms.Driver) |
|||
} |
|||
|
|||
func TestLoadConfig(t *testing.T) { |
|||
MustLoadConfig("../../conf.sample.yml") |
|||
} |
|||
func TestCheckConfig(t *testing.T) { |
|||
conf := Config |
|||
conf.RetryInterval = 1 |
|||
retryIntervalErr := checkConfig(&conf) |
|||
retryIntervalExpect := errors.New("RetryInterval should not be less than 10") |
|||
assert.Equal(t, retryIntervalErr, retryIntervalExpect) |
|||
|
|||
conf.RetryInterval = 10 |
|||
conf.TimeoutToFail = 5 |
|||
timeoutToFailErr := checkConfig(&conf) |
|||
timeoutToFailExpect := errors.New("TimeoutToFail should not be less than RetryInterval") |
|||
assert.Equal(t, timeoutToFailErr, timeoutToFailExpect) |
|||
|
|||
conf.TimeoutToFail = 20 |
|||
driverErr := checkConfig(&conf) |
|||
assert.Equal(t, driverErr, nil) |
|||
|
|||
conf.Store = Store{Driver: Mysql} |
|||
hostErr := checkConfig(&conf) |
|||
hostExpect := errors.New("Db host not valid ") |
|||
assert.Equal(t, hostErr, hostExpect) |
|||
|
|||
conf.Store = Store{Driver: Mysql, Host: "127.0.0.1"} |
|||
portErr := checkConfig(&conf) |
|||
portExpect := errors.New("Db port not valid ") |
|||
assert.Equal(t, portErr, portExpect) |
|||
|
|||
conf.Store = Store{Driver: Mysql, Host: "127.0.0.1", Port: 8686} |
|||
userErr := checkConfig(&conf) |
|||
userExpect := errors.New("Db user not valid ") |
|||
assert.Equal(t, userErr, userExpect) |
|||
|
|||
conf.Store = Store{Driver: Redis, Host: "", Port: 8686} |
|||
assert.Equal(t, errors.New("Redis host not valid"), checkConfig(&conf)) |
|||
|
|||
conf.Store = Store{Driver: Redis, Host: "127.0.0.1", Port: 0} |
|||
assert.Equal(t, errors.New("Redis port not valid"), checkConfig(&conf)) |
|||
|
|||
} |
|||
|
|||
func TestConfig(t *testing.T) { |
|||
testConfigStringField(&Config.Store.Driver, "", t) |
|||
testConfigStringField(&Config.Store.User, "", t) |
|||
testConfigIntField(&Config.RetryInterval, 9, t) |
|||
testConfigIntField(&Config.TimeoutToFail, 9, t) |
|||
} |
|||
|
|||
func testConfigStringField(fd *string, val string, t *testing.T) { |
|||
old := *fd |
|||
*fd = val |
|||
str := checkConfig(&Config) |
|||
assert.NotEqual(t, "", str) |
|||
*fd = old |
|||
} |
|||
|
|||
func testConfigIntField(fd *int64, val int64, t *testing.T) { |
|||
old := *fd |
|||
*fd = val |
|||
str := checkConfig(&Config) |
|||
assert.NotEqual(t, "", str) |
|||
*fd = old |
|||
} |
|||
@ -1,7 +0,0 @@ |
|||
package dtmsvr |
|||
|
|||
import ( |
|||
_ "github.com/dtm-labs/dtmdriver-gozero" |
|||
_ "github.com/dtm-labs/dtmdriver-polaris" |
|||
_ "github.com/dtm-labs/dtmdriver-protocol1" |
|||
) |
|||
@ -0,0 +1,14 @@ |
|||
/* |
|||
* 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 dtmutil |
|||
|
|||
const ( |
|||
// DefaultHttpServer default url for http server. used by test and examples
|
|||
DefaultHttpServer = "http://localhost:36789/api/dtmsvr" |
|||
// DefaultGrpcServer default url for grpc server. used by test and examples
|
|||
DefaultGrpcServer = "localhost:36790" |
|||
) |
|||
@ -1,14 +0,0 @@ |
|||
package examples |
|||
|
|||
import "fmt" |
|||
|
|||
func Startup() { |
|||
InitConfig() |
|||
GrpcStartup() |
|||
BaseAppStartup() |
|||
} |
|||
|
|||
func InitConfig() { |
|||
DtmHttpServer = fmt.Sprintf("http://localhost:%d/api/dtmsvr", config.HttpPort) |
|||
DtmGrpcServer = fmt.Sprintf("localhost:%d", config.GrpcPort) |
|||
} |
|||
@ -1,330 +0,0 @@ |
|||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|||
// versions:
|
|||
// protoc-gen-go v1.27.1
|
|||
// protoc v3.17.3
|
|||
// source: examples/busi.proto
|
|||
|
|||
package examples |
|||
|
|||
import ( |
|||
protoreflect "google.golang.org/protobuf/reflect/protoreflect" |
|||
protoimpl "google.golang.org/protobuf/runtime/protoimpl" |
|||
emptypb "google.golang.org/protobuf/types/known/emptypb" |
|||
reflect "reflect" |
|||
sync "sync" |
|||
) |
|||
|
|||
const ( |
|||
// Verify that this generated code is sufficiently up-to-date.
|
|||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) |
|||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
|||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) |
|||
) |
|||
|
|||
// DtmRequest request sent to dtm server
|
|||
type BusiReq struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Amount int64 `protobuf:"varint,1,opt,name=Amount,proto3" json:"Amount,omitempty"` |
|||
TransOutResult string `protobuf:"bytes,2,opt,name=TransOutResult,proto3" json:"TransOutResult,omitempty"` |
|||
TransInResult string `protobuf:"bytes,3,opt,name=TransInResult,proto3" json:"TransInResult,omitempty"` |
|||
} |
|||
|
|||
func (x *BusiReq) Reset() { |
|||
*x = BusiReq{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_examples_busi_proto_msgTypes[0] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *BusiReq) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*BusiReq) ProtoMessage() {} |
|||
|
|||
func (x *BusiReq) ProtoReflect() protoreflect.Message { |
|||
mi := &file_examples_busi_proto_msgTypes[0] |
|||
if protoimpl.UnsafeEnabled && x != nil { |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
if ms.LoadMessageInfo() == nil { |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
return ms |
|||
} |
|||
return mi.MessageOf(x) |
|||
} |
|||
|
|||
// Deprecated: Use BusiReq.ProtoReflect.Descriptor instead.
|
|||
func (*BusiReq) Descriptor() ([]byte, []int) { |
|||
return file_examples_busi_proto_rawDescGZIP(), []int{0} |
|||
} |
|||
|
|||
func (x *BusiReq) GetAmount() int64 { |
|||
if x != nil { |
|||
return x.Amount |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func (x *BusiReq) GetTransOutResult() string { |
|||
if x != nil { |
|||
return x.TransOutResult |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *BusiReq) GetTransInResult() string { |
|||
if x != nil { |
|||
return x.TransInResult |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
type BusiReply struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Message string `protobuf:"bytes,1,opt,name=message,proto3" json:"message,omitempty"` |
|||
} |
|||
|
|||
func (x *BusiReply) Reset() { |
|||
*x = BusiReply{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_examples_busi_proto_msgTypes[1] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *BusiReply) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*BusiReply) ProtoMessage() {} |
|||
|
|||
func (x *BusiReply) ProtoReflect() protoreflect.Message { |
|||
mi := &file_examples_busi_proto_msgTypes[1] |
|||
if protoimpl.UnsafeEnabled && x != nil { |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
if ms.LoadMessageInfo() == nil { |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
return ms |
|||
} |
|||
return mi.MessageOf(x) |
|||
} |
|||
|
|||
// Deprecated: Use BusiReply.ProtoReflect.Descriptor instead.
|
|||
func (*BusiReply) Descriptor() ([]byte, []int) { |
|||
return file_examples_busi_proto_rawDescGZIP(), []int{1} |
|||
} |
|||
|
|||
func (x *BusiReply) GetMessage() string { |
|||
if x != nil { |
|||
return x.Message |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
var File_examples_busi_proto protoreflect.FileDescriptor |
|||
|
|||
var file_examples_busi_proto_rawDesc = []byte{ |
|||
0x0a, 0x13, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2f, 0x62, 0x75, 0x73, 0x69, 0x2e, |
|||
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x08, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x1a, |
|||
0x1b, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, |
|||
0x2f, 0x65, 0x6d, 0x70, 0x74, 0x79, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x6f, 0x0a, 0x07, |
|||
0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x12, 0x16, 0x0a, 0x06, 0x41, 0x6d, 0x6f, 0x75, 0x6e, |
|||
0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x03, 0x52, 0x06, 0x41, 0x6d, 0x6f, 0x75, 0x6e, 0x74, 0x12, |
|||
0x26, 0x0a, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, 0x73, 0x75, 0x6c, |
|||
0x74, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, |
|||
0x74, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x12, 0x24, 0x0a, 0x0d, 0x54, 0x72, 0x61, 0x6e, 0x73, |
|||
0x49, 0x6e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x22, 0x25, 0x0a, |
|||
0x09, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, |
|||
0x73, 0x73, 0x61, 0x67, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, |
|||
0x73, 0x61, 0x67, 0x65, 0x32, 0x97, 0x08, 0x0a, 0x04, 0x42, 0x75, 0x73, 0x69, 0x12, 0x35, 0x0a, |
|||
0x09, 0x43, 0x61, 0x6e, 0x53, 0x75, 0x62, 0x6d, 0x69, 0x74, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, |
|||
0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x13, 0x2e, |
|||
0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x70, |
|||
0x6c, 0x79, 0x22, 0x00, 0x12, 0x36, 0x0a, 0x07, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x12, |
|||
0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, |
|||
0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, |
|||
0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x37, 0x0a, 0x08, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, |
|||
0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, |
|||
0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, |
|||
0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3c, 0x0a, 0x0d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, |
|||
0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, |
|||
0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, |
|||
0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, |
|||
0x79, 0x22, 0x00, 0x12, 0x3d, 0x0a, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, |
|||
0x65, 0x76, 0x65, 0x72, 0x74, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, |
|||
0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
|||
0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, |
|||
0x22, 0x00, 0x12, 0x3d, 0x0a, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x43, 0x6f, 0x6e, |
|||
0x66, 0x69, 0x72, 0x6d, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, |
|||
0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, |
|||
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, |
|||
0x00, 0x12, 0x3e, 0x0a, 0x0f, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x43, 0x6f, 0x6e, |
|||
0x66, 0x69, 0x72, 0x6d, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, |
|||
0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, |
|||
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, |
|||
0x00, 0x12, 0x3c, 0x0a, 0x08, 0x58, 0x61, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x79, 0x12, 0x16, 0x2e, |
|||
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, |
|||
0x45, 0x6d, 0x70, 0x74, 0x79, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, |
|||
0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, |
|||
0x38, 0x0a, 0x09, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x58, 0x61, 0x12, 0x11, 0x2e, 0x65, |
|||
0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, |
|||
0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, |
|||
0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x39, 0x0a, 0x0a, 0x54, 0x72, 0x61, |
|||
0x6e, 0x73, 0x4f, 0x75, 0x74, 0x58, 0x61, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, |
|||
0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, |
|||
0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, |
|||
0x74, 0x79, 0x22, 0x00, 0x12, 0x39, 0x0a, 0x0a, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x54, |
|||
0x63, 0x63, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, |
|||
0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, |
|||
0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, |
|||
0x3a, 0x0a, 0x0b, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x54, 0x63, 0x63, 0x12, 0x11, |
|||
0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, |
|||
0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, |
|||
0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3f, 0x0a, 0x10, 0x54, |
|||
0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x54, 0x63, 0x63, 0x4e, 0x65, 0x73, 0x74, 0x65, 0x64, 0x12, |
|||
0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, |
|||
0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, |
|||
0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3b, 0x0a, 0x0c, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x42, 0x53, 0x61, 0x67, 0x61, 0x12, 0x11, 0x2e, 0x65, |
|||
0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, |
|||
0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, |
|||
0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3c, 0x0a, 0x0d, 0x54, 0x72, 0x61, |
|||
0x6e, 0x73, 0x4f, 0x75, 0x74, 0x42, 0x53, 0x61, 0x67, 0x61, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, |
|||
0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, |
|||
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, |
|||
0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x41, 0x0a, 0x12, 0x54, 0x72, 0x61, 0x6e, 0x73, |
|||
0x49, 0x6e, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x42, 0x53, 0x61, 0x67, 0x61, 0x12, 0x11, 0x2e, |
|||
0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, |
|||
0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, |
|||
0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x42, 0x0a, 0x13, 0x54, 0x72, |
|||
0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x42, 0x53, 0x61, 0x67, |
|||
0x61, 0x12, 0x11, 0x2e, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2e, 0x42, 0x75, 0x73, |
|||
0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, |
|||
0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x42, 0x0c, |
|||
0x5a, 0x0a, 0x2e, 0x2f, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x62, 0x06, 0x70, 0x72, |
|||
0x6f, 0x74, 0x6f, 0x33, |
|||
} |
|||
|
|||
var ( |
|||
file_examples_busi_proto_rawDescOnce sync.Once |
|||
file_examples_busi_proto_rawDescData = file_examples_busi_proto_rawDesc |
|||
) |
|||
|
|||
func file_examples_busi_proto_rawDescGZIP() []byte { |
|||
file_examples_busi_proto_rawDescOnce.Do(func() { |
|||
file_examples_busi_proto_rawDescData = protoimpl.X.CompressGZIP(file_examples_busi_proto_rawDescData) |
|||
}) |
|||
return file_examples_busi_proto_rawDescData |
|||
} |
|||
|
|||
var file_examples_busi_proto_msgTypes = make([]protoimpl.MessageInfo, 2) |
|||
var file_examples_busi_proto_goTypes = []interface{}{ |
|||
(*BusiReq)(nil), // 0: examples.BusiReq
|
|||
(*BusiReply)(nil), // 1: examples.BusiReply
|
|||
(*emptypb.Empty)(nil), // 2: google.protobuf.Empty
|
|||
} |
|||
var file_examples_busi_proto_depIdxs = []int32{ |
|||
0, // 0: examples.Busi.CanSubmit:input_type -> examples.BusiReq
|
|||
0, // 1: examples.Busi.TransIn:input_type -> examples.BusiReq
|
|||
0, // 2: examples.Busi.TransOut:input_type -> examples.BusiReq
|
|||
0, // 3: examples.Busi.TransInRevert:input_type -> examples.BusiReq
|
|||
0, // 4: examples.Busi.TransOutRevert:input_type -> examples.BusiReq
|
|||
0, // 5: examples.Busi.TransInConfirm:input_type -> examples.BusiReq
|
|||
0, // 6: examples.Busi.TransOutConfirm:input_type -> examples.BusiReq
|
|||
2, // 7: examples.Busi.XaNotify:input_type -> google.protobuf.Empty
|
|||
0, // 8: examples.Busi.TransInXa:input_type -> examples.BusiReq
|
|||
0, // 9: examples.Busi.TransOutXa:input_type -> examples.BusiReq
|
|||
0, // 10: examples.Busi.TransInTcc:input_type -> examples.BusiReq
|
|||
0, // 11: examples.Busi.TransOutTcc:input_type -> examples.BusiReq
|
|||
0, // 12: examples.Busi.TransInTccNested:input_type -> examples.BusiReq
|
|||
0, // 13: examples.Busi.TransInBSaga:input_type -> examples.BusiReq
|
|||
0, // 14: examples.Busi.TransOutBSaga:input_type -> examples.BusiReq
|
|||
0, // 15: examples.Busi.TransInRevertBSaga:input_type -> examples.BusiReq
|
|||
0, // 16: examples.Busi.TransOutRevertBSaga:input_type -> examples.BusiReq
|
|||
1, // 17: examples.Busi.CanSubmit:output_type -> examples.BusiReply
|
|||
2, // 18: examples.Busi.TransIn:output_type -> google.protobuf.Empty
|
|||
2, // 19: examples.Busi.TransOut:output_type -> google.protobuf.Empty
|
|||
2, // 20: examples.Busi.TransInRevert:output_type -> google.protobuf.Empty
|
|||
2, // 21: examples.Busi.TransOutRevert:output_type -> google.protobuf.Empty
|
|||
2, // 22: examples.Busi.TransInConfirm:output_type -> google.protobuf.Empty
|
|||
2, // 23: examples.Busi.TransOutConfirm:output_type -> google.protobuf.Empty
|
|||
2, // 24: examples.Busi.XaNotify:output_type -> google.protobuf.Empty
|
|||
2, // 25: examples.Busi.TransInXa:output_type -> google.protobuf.Empty
|
|||
2, // 26: examples.Busi.TransOutXa:output_type -> google.protobuf.Empty
|
|||
2, // 27: examples.Busi.TransInTcc:output_type -> google.protobuf.Empty
|
|||
2, // 28: examples.Busi.TransOutTcc:output_type -> google.protobuf.Empty
|
|||
2, // 29: examples.Busi.TransInTccNested:output_type -> google.protobuf.Empty
|
|||
2, // 30: examples.Busi.TransInBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 31: examples.Busi.TransOutBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 32: examples.Busi.TransInRevertBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 33: examples.Busi.TransOutRevertBSaga:output_type -> google.protobuf.Empty
|
|||
17, // [17:34] is the sub-list for method output_type
|
|||
0, // [0:17] is the sub-list for method input_type
|
|||
0, // [0:0] is the sub-list for extension type_name
|
|||
0, // [0:0] is the sub-list for extension extendee
|
|||
0, // [0:0] is the sub-list for field type_name
|
|||
} |
|||
|
|||
func init() { file_examples_busi_proto_init() } |
|||
func file_examples_busi_proto_init() { |
|||
if File_examples_busi_proto != nil { |
|||
return |
|||
} |
|||
if !protoimpl.UnsafeEnabled { |
|||
file_examples_busi_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*BusiReq); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_examples_busi_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*BusiReply); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
} |
|||
type x struct{} |
|||
out := protoimpl.TypeBuilder{ |
|||
File: protoimpl.DescBuilder{ |
|||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(), |
|||
RawDescriptor: file_examples_busi_proto_rawDesc, |
|||
NumEnums: 0, |
|||
NumMessages: 2, |
|||
NumExtensions: 0, |
|||
NumServices: 1, |
|||
}, |
|||
GoTypes: file_examples_busi_proto_goTypes, |
|||
DependencyIndexes: file_examples_busi_proto_depIdxs, |
|||
MessageInfos: file_examples_busi_proto_msgTypes, |
|||
}.Build() |
|||
File_examples_busi_proto = out.File |
|||
file_examples_busi_proto_rawDesc = nil |
|||
file_examples_busi_proto_goTypes = nil |
|||
file_examples_busi_proto_depIdxs = nil |
|||
} |
|||
@ -1,55 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"fmt" |
|||
|
|||
"github.com/dtm-labs/dtm/common" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
) |
|||
|
|||
var config = &common.Config |
|||
|
|||
func resetXaData() { |
|||
if config.ExamplesDB.Driver != "mysql" { |
|||
return |
|||
} |
|||
|
|||
db := dbGet() |
|||
type XaRow struct { |
|||
Data string |
|||
} |
|||
xas := []XaRow{} |
|||
db.Must().Raw("xa recover").Scan(&xas) |
|||
for _, xa := range xas { |
|||
db.Must().Exec(fmt.Sprintf("xa rollback '%s'", xa.Data)) |
|||
} |
|||
} |
|||
|
|||
// PopulateDB populate example mysql data
|
|||
func PopulateDB(skipDrop bool) { |
|||
resetXaData() |
|||
file := fmt.Sprintf("%s/examples.%s.sql", common.GetSqlDir(), config.ExamplesDB.Driver) |
|||
common.RunSQLScript(config.ExamplesDB, file, skipDrop) |
|||
file = fmt.Sprintf("%s/dtmcli.barrier.%s.sql", common.GetSqlDir(), config.ExamplesDB.Driver) |
|||
common.RunSQLScript(config.ExamplesDB, file, skipDrop) |
|||
} |
|||
|
|||
type sampleInfo struct { |
|||
Arg string |
|||
Action func() string |
|||
Desc string |
|||
} |
|||
|
|||
// Samples 所有的示例都会注册到这里
|
|||
var Samples = map[string]*sampleInfo{} |
|||
|
|||
func addSample(name string, fn func() string) { |
|||
logger.FatalfIf(Samples[name] != nil, "%s already exists", name) |
|||
Samples[name] = &sampleInfo{Arg: name, Action: fn} |
|||
} |
|||
@ -1,25 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
dtmgrpc "github.com/dtm-labs/dtm/dtmgrpc" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("grpc_msg", func() string { |
|||
req := &BusiReq{Amount: 30} |
|||
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
|||
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer, gid). |
|||
Add(BusiGrpc+"/examples.Busi/TransOut", req). |
|||
Add(BusiGrpc+"/examples.Busi/TransIn", req) |
|||
err := msg.Submit() |
|||
logger.FatalIfError(err) |
|||
return msg.Gid |
|||
}) |
|||
} |
|||
@ -1,36 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
dtmgrpc "github.com/dtm-labs/dtm/dtmgrpc" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("grpc_saga", func() string { |
|||
req := &BusiReq{Amount: 30} |
|||
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
|||
saga := dtmgrpc.NewSagaGrpc(DtmGrpcServer, gid). |
|||
Add(BusiGrpc+"/examples.Busi/TransOut", BusiGrpc+"/examples.Busi/TransOutRevert", req). |
|||
Add(BusiGrpc+"/examples.Busi/TransIn", BusiGrpc+"/examples.Busi/TransInRevert", req) |
|||
err := saga.Submit() |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
addSample("grpc_saga_wait", func() string { |
|||
req := &BusiReq{Amount: 30} |
|||
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
|||
saga := dtmgrpc.NewSagaGrpc(DtmGrpcServer, gid). |
|||
Add(BusiGrpc+"/examples.Busi/TransOut", BusiGrpc+"/examples.Busi/TransOutRevert", req). |
|||
Add(BusiGrpc+"/examples.Busi/TransIn", BusiGrpc+"/examples.Busi/TransInRevert", req) |
|||
saga.WaitResult = true |
|||
err := saga.Submit() |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
} |
|||
@ -1,70 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/dtmgrpc" |
|||
"google.golang.org/grpc/codes" |
|||
"google.golang.org/grpc/status" |
|||
emptypb "google.golang.org/protobuf/types/known/emptypb" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("grpc_saga_barrier", func() string { |
|||
req := &BusiReq{Amount: 30} |
|||
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
|||
saga := dtmgrpc.NewSagaGrpc(DtmGrpcServer, gid). |
|||
Add(BusiGrpc+"/examples.Busi/TransOutBSaga", BusiGrpc+"/examples.Busi/TransOutRevertBSaga", req). |
|||
Add(BusiGrpc+"/examples.Busi/TransInBSaga", BusiGrpc+"/examples.Busi/TransInRevertBSaga", req) |
|||
err := saga.Submit() |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
} |
|||
|
|||
func sagaGrpcBarrierAdjustBalance(db dtmcli.DB, uid int, amount int64, result string) error { |
|||
if result == dtmcli.ResultFailure { |
|||
return status.New(codes.Aborted, dtmcli.ResultFailure).Err() |
|||
} |
|||
_, err := dtmimp.DBExec(db, "update dtm_busi.user_account set balance = balance + ? where user_id = ?", amount, uid) |
|||
return err |
|||
|
|||
} |
|||
|
|||
func (s *busiServer) TransInBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcBarrierAdjustBalance(tx, 2, in.Amount, in.TransInResult) |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransOutBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcBarrierAdjustBalance(tx, 1, -in.Amount, in.TransOutResult) |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransInRevertBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcBarrierAdjustBalance(tx, 2, -in.Amount, "") |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransOutRevertBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcBarrierAdjustBalance(tx, 1, in.Amount, "") |
|||
}) |
|||
} |
|||
@ -1,32 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
dtmgrpc "github.com/dtm-labs/dtm/dtmgrpc" |
|||
emptypb "google.golang.org/protobuf/types/known/emptypb" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("grpc_tcc", func() string { |
|||
logger.Debugf("tcc simple transaction begin") |
|||
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
|||
err := dtmgrpc.TccGlobalTransaction(DtmGrpcServer, gid, func(tcc *dtmgrpc.TccGrpc) error { |
|||
data := &BusiReq{Amount: 30} |
|||
r := &emptypb.Empty{} |
|||
err := tcc.CallBranch(data, BusiGrpc+"/examples.Busi/TransOutTcc", BusiGrpc+"/examples.Busi/TransOutConfirm", BusiGrpc+"/examples.Busi/TransOutRevert", r) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
err = tcc.CallBranch(data, BusiGrpc+"/examples.Busi/TransInTcc", BusiGrpc+"/examples.Busi/TransInConfirm", BusiGrpc+"/examples.Busi/TransInRevert", r) |
|||
return err |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
} |
|||
@ -1,37 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
context "context" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/dtmgrpc" |
|||
"google.golang.org/protobuf/types/known/emptypb" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("grpc_xa", func() string { |
|||
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
|||
req := &BusiReq{Amount: 30} |
|||
err := XaGrpcClient.XaGlobalTransaction(gid, func(xa *dtmgrpc.XaGrpc) error { |
|||
r := &emptypb.Empty{} |
|||
err := xa.CallBranch(req, BusiGrpc+"/examples.Busi/TransOutXa", r) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
err = xa.CallBranch(req, BusiGrpc+"/examples.Busi/TransInXa", r) |
|||
return err |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) XaNotify(ctx context.Context, in *emptypb.Empty) (*emptypb.Empty, error) { |
|||
return XaGrpcClient.HandleCallback(ctx) |
|||
} |
|||
@ -1,29 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/go-resty/resty/v2" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("xa_gorm", func() string { |
|||
gid := dtmcli.MustGenGid(DtmHttpServer) |
|||
err := XaClient.XaGlobalTransaction(gid, func(xa *dtmcli.Xa) (*resty.Response, error) { |
|||
resp, err := xa.CallBranch(&TransReq{Amount: 30}, Busi+"/TransOutXaGorm") |
|||
if err != nil { |
|||
return resp, err |
|||
} |
|||
return xa.CallBranch(&TransReq{Amount: 30}, Busi+"/TransInXa") |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
|
|||
} |
|||
@ -1,28 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("msg", func() string { |
|||
logger.Debugf("a busi transaction begin") |
|||
req := &TransReq{Amount: 30} |
|||
msg := dtmcli.NewMsg(DtmHttpServer, dtmcli.MustGenGid(DtmHttpServer)). |
|||
Add(Busi+"/TransOut", req). |
|||
Add(Busi+"/TransIn", req) |
|||
err := msg.Prepare(Busi + "/query") |
|||
logger.FatalIfError(err) |
|||
logger.Debugf("busi trans submit") |
|||
err = msg.Submit() |
|||
logger.FatalIfError(err) |
|||
return msg.Gid |
|||
}) |
|||
} |
|||
@ -1,56 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
) |
|||
|
|||
func init() { |
|||
addSample("saga", func() string { |
|||
logger.Debugf("a saga busi transaction begin") |
|||
req := &TransReq{Amount: 30} |
|||
saga := dtmcli.NewSaga(DtmHttpServer, dtmcli.MustGenGid(DtmHttpServer)). |
|||
Add(Busi+"/TransOut", Busi+"/TransOutRevert", req). |
|||
Add(Busi+"/TransIn", Busi+"/TransInRevert", req) |
|||
logger.Debugf("saga busi trans submit") |
|||
err := saga.Submit() |
|||
logger.Debugf("result gid is: %s", saga.Gid) |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
addSample("saga_wait", func() string { |
|||
logger.Debugf("a saga busi transaction begin") |
|||
req := &TransReq{Amount: 30} |
|||
saga := dtmcli.NewSaga(DtmHttpServer, dtmcli.MustGenGid(DtmHttpServer)). |
|||
Add(Busi+"/TransOut", Busi+"/TransOutRevert", req). |
|||
Add(Busi+"/TransIn", Busi+"/TransInRevert", req) |
|||
saga.SetOptions(&dtmcli.TransOptions{WaitResult: true}) |
|||
err := saga.Submit() |
|||
logger.Debugf("result gid is: %s", saga.Gid) |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
addSample("concurrent_saga", func() string { |
|||
logger.Debugf("a concurrent saga busi transaction begin") |
|||
req := &TransReq{Amount: 30} |
|||
csaga := dtmcli.NewSaga(DtmHttpServer, dtmcli.MustGenGid(DtmHttpServer)). |
|||
Add(Busi+"/TransOut", Busi+"/TransOutRevert", req). |
|||
Add(Busi+"/TransOut", Busi+"/TransOutRevert", req). |
|||
Add(Busi+"/TransIn", Busi+"/TransInRevert", req). |
|||
Add(Busi+"/TransIn", Busi+"/TransInRevert", req). |
|||
EnableConcurrent(). |
|||
AddBranchOrder(2, []int{0, 1}). |
|||
AddBranchOrder(3, []int{0, 1}) |
|||
logger.Debugf("concurrent saga busi trans submit") |
|||
err := csaga.Submit() |
|||
logger.Debugf("result gid is: %s", csaga.Gid) |
|||
logger.FatalIfError(err) |
|||
return csaga.Gid |
|||
}) |
|||
} |
|||
@ -1,79 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"database/sql" |
|||
|
|||
"github.com/dtm-labs/dtm/common" |
|||
"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" |
|||
) |
|||
|
|||
func init() { |
|||
setupFuncs["SagaBarrierSetup"] = func(app *gin.Engine) { |
|||
app.POST(BusiAPI+"/SagaBTransIn", common.WrapHandler(sagaBarrierTransIn)) |
|||
app.POST(BusiAPI+"/SagaBTransInCompensate", common.WrapHandler(sagaBarrierTransInCompensate)) |
|||
app.POST(BusiAPI+"/SagaBTransOut", common.WrapHandler(sagaBarrierTransOut)) |
|||
app.POST(BusiAPI+"/SagaBTransOutCompensate", common.WrapHandler(sagaBarrierTransOutCompensate)) |
|||
} |
|||
addSample("saga_barrier", func() string { |
|||
logger.Debugf("a busi transaction begin") |
|||
req := &TransReq{Amount: 30} |
|||
saga := dtmcli.NewSaga(DtmHttpServer, dtmcli.MustGenGid(DtmHttpServer)). |
|||
Add(Busi+"/SagaBTransOut", Busi+"/SagaBTransOutCompensate", req). |
|||
Add(Busi+"/SagaBTransIn", Busi+"/SagaBTransInCompensate", req) |
|||
logger.Debugf("busi trans submit") |
|||
err := saga.Submit() |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
} |
|||
|
|||
func sagaBarrierAdjustBalance(db dtmcli.DB, uid int, amount int) error { |
|||
_, err := dtmimp.DBExec(db, "update dtm_busi.user_account set balance = balance + ? where user_id = ?", amount, uid) |
|||
return err |
|||
|
|||
} |
|||
|
|||
func sagaBarrierTransIn(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) |
|||
if req.TransInResult != "" { |
|||
return req.TransInResult, nil |
|||
} |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaBarrierAdjustBalance(tx, 1, req.Amount) |
|||
}) |
|||
} |
|||
|
|||
func sagaBarrierTransInCompensate(c *gin.Context) (interface{}, error) { |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaBarrierAdjustBalance(tx, 1, -reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
|
|||
func sagaBarrierTransOut(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) |
|||
if req.TransOutResult != "" { |
|||
return req.TransOutResult, nil |
|||
} |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaBarrierAdjustBalance(tx, 2, -req.Amount) |
|||
}) |
|||
} |
|||
|
|||
func sagaBarrierTransOutCompensate(c *gin.Context) (interface{}, error) { |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaBarrierAdjustBalance(tx, 2, reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
@ -1,43 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"database/sql" |
|||
|
|||
"github.com/dtm-labs/dtm/common" |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/gin-gonic/gin" |
|||
) |
|||
|
|||
func init() { |
|||
setupFuncs["SagaGormBarrierSetup"] = func(app *gin.Engine) { |
|||
app.POST(BusiAPI+"/SagaBTransOutGorm", common.WrapHandler(sagaGormBarrierTransOut)) |
|||
} |
|||
addSample("saga_gorm_barrier", func() string { |
|||
logger.Debugf("a busi transaction begin") |
|||
req := &TransReq{Amount: 30} |
|||
saga := dtmcli.NewSaga(DtmHttpServer, dtmcli.MustGenGid(DtmHttpServer)). |
|||
Add(Busi+"/SagaBTransOutGorm", Busi+"/SagaBTransOutCompensate", req). |
|||
Add(Busi+"/SagaBTransIn", Busi+"/SagaBTransInCompensate", req) |
|||
logger.Debugf("busi trans submit") |
|||
err := saga.Submit() |
|||
logger.FatalIfError(err) |
|||
return saga.Gid |
|||
}) |
|||
|
|||
} |
|||
|
|||
func sagaGormBarrierTransOut(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) |
|||
barrier := MustBarrierFromGin(c) |
|||
tx := dbGet().DB.Begin() |
|||
return dtmcli.MapSuccess, barrier.Call(tx.Statement.ConnPool.(*sql.Tx), func(tx1 *sql.Tx) error { |
|||
return tx.Exec("update dtm_busi.user_account set balance = balance + ? where user_id = ?", -req.Amount, 2).Error |
|||
}) |
|||
} |
|||
@ -1,51 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/common" |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/gin-gonic/gin" |
|||
"github.com/go-resty/resty/v2" |
|||
) |
|||
|
|||
func init() { |
|||
setupFuncs["TccSetupSetup"] = func(app *gin.Engine) { |
|||
app.POST(BusiAPI+"/TransInTccParent", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
tcc, err := dtmcli.TccFromQuery(c.Request.URL.Query()) |
|||
logger.FatalIfError(err) |
|||
logger.Debugf("TransInTccParent ") |
|||
return tcc.CallBranch(&TransReq{Amount: reqFrom(c).Amount}, Busi+"/TransIn", Busi+"/TransInConfirm", Busi+"/TransInRevert") |
|||
})) |
|||
} |
|||
addSample("tcc_nested", func() string { |
|||
gid := dtmcli.MustGenGid(DtmHttpServer) |
|||
err := dtmcli.TccGlobalTransaction(DtmHttpServer, gid, func(tcc *dtmcli.Tcc) (*resty.Response, error) { |
|||
resp, err := tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TransOut", Busi+"/TransOutConfirm", Busi+"/TransOutRevert") |
|||
if err != nil { |
|||
return resp, err |
|||
} |
|||
return tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TransInTccParent", Busi+"/TransInConfirm", Busi+"/TransInRevert") |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
addSample("tcc", func() string { |
|||
logger.Debugf("tcc simple transaction begin") |
|||
gid := dtmcli.MustGenGid(DtmHttpServer) |
|||
err := dtmcli.TccGlobalTransaction(DtmHttpServer, gid, func(tcc *dtmcli.Tcc) (*resty.Response, error) { |
|||
resp, err := tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TransOut", Busi+"/TransOutConfirm", Busi+"/TransOutRevert") |
|||
if err != nil { |
|||
return resp, err |
|||
} |
|||
return tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TransIn", Busi+"/TransInConfirm", Busi+"/TransInRevert") |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
} |
|||
@ -1,111 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"database/sql" |
|||
"fmt" |
|||
|
|||
"github.com/dtm-labs/dtm/common" |
|||
"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" |
|||
"github.com/go-resty/resty/v2" |
|||
) |
|||
|
|||
func init() { |
|||
setupFuncs["TccBarrierSetup"] = func(app *gin.Engine) { |
|||
app.POST(BusiAPI+"/TccBTransInTry", common.WrapHandler(tccBarrierTransInTry)) |
|||
app.POST(BusiAPI+"/TccBTransInConfirm", common.WrapHandler(tccBarrierTransInConfirm)) |
|||
app.POST(BusiAPI+"/TccBTransInCancel", common.WrapHandler(tccBarrierTransInCancel)) |
|||
app.POST(BusiAPI+"/TccBTransOutTry", common.WrapHandler(tccBarrierTransOutTry)) |
|||
app.POST(BusiAPI+"/TccBTransOutConfirm", common.WrapHandler(tccBarrierTransOutConfirm)) |
|||
app.POST(BusiAPI+"/TccBTransOutCancel", common.WrapHandler(TccBarrierTransOutCancel)) |
|||
} |
|||
addSample("tcc_barrier", func() string { |
|||
logger.Debugf("tcc transaction begin") |
|||
gid := dtmcli.MustGenGid(DtmHttpServer) |
|||
err := dtmcli.TccGlobalTransaction(DtmHttpServer, gid, func(tcc *dtmcli.Tcc) (*resty.Response, error) { |
|||
resp, err := tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TccBTransOutTry", |
|||
Busi+"/TccBTransOutConfirm", Busi+"/TccBTransOutCancel") |
|||
if err != nil { |
|||
return resp, err |
|||
} |
|||
return tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TccBTransInTry", Busi+"/TccBTransInConfirm", Busi+"/TccBTransInCancel") |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
} |
|||
|
|||
const transInUID = 1 |
|||
const transOutUID = 2 |
|||
|
|||
func adjustTrading(db dtmcli.DB, uid int, amount int) error { |
|||
affected, err := dtmimp.DBExec(db, `update dtm_busi.user_account set trading_balance=trading_balance+? |
|||
where user_id=? and trading_balance + ? + balance >= 0`, amount, uid, amount) |
|||
if err == nil && affected == 0 { |
|||
return fmt.Errorf("update error, maybe balance not enough") |
|||
} |
|||
return err |
|||
} |
|||
|
|||
func adjustBalance(db dtmcli.DB, uid int, amount int) error { |
|||
affected, err := dtmimp.DBExec(db, `update dtm_busi.user_account set trading_balance=trading_balance-?, |
|||
balance=balance+? where user_id=?`, amount, amount, uid) |
|||
if err == nil && affected == 0 { |
|||
return fmt.Errorf("update user_account 0 rows") |
|||
} |
|||
return err |
|||
} |
|||
|
|||
// TCC下，转入
|
|||
func tccBarrierTransInTry(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) // 去重构一下，改成可以重复使用的输入
|
|||
if req.TransInResult != "" { |
|||
return req.TransInResult, nil |
|||
} |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return adjustTrading(tx, transInUID, req.Amount) |
|||
}) |
|||
} |
|||
|
|||
func tccBarrierTransInConfirm(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return adjustBalance(tx, transInUID, reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
|
|||
func tccBarrierTransInCancel(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return adjustTrading(tx, transInUID, -reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
|
|||
func tccBarrierTransOutTry(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) |
|||
if req.TransOutResult != "" { |
|||
return req.TransOutResult, nil |
|||
} |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return adjustTrading(tx, transOutUID, -req.Amount) |
|||
}) |
|||
} |
|||
|
|||
func tccBarrierTransOutConfirm(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return adjustBalance(tx, transOutUID, -reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
|
|||
// TccBarrierTransOutCancel will be use in test
|
|||
func TccBarrierTransOutCancel(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return adjustTrading(tx, transOutUID, reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
@ -1,42 +0,0 @@ |
|||
/* |
|||
* 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 examples |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/common" |
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/gin-gonic/gin" |
|||
"github.com/go-resty/resty/v2" |
|||
) |
|||
|
|||
// XaClient XA client connection
|
|||
var XaClient *dtmcli.XaClient = nil |
|||
|
|||
func init() { |
|||
setupFuncs["XaSetup"] = func(app *gin.Engine) { |
|||
var err error |
|||
XaClient, err = dtmcli.NewXaClient(DtmHttpServer, config.ExamplesDB, Busi+"/xa", func(path string, xa *dtmcli.XaClient) { |
|||
app.POST(path, common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return xa.HandleCallback(c.Query("gid"), c.Query("branch_id"), c.Query("op")) |
|||
})) |
|||
}) |
|||
logger.FatalIfError(err) |
|||
} |
|||
addSample("xa", func() string { |
|||
gid := dtmcli.MustGenGid(DtmHttpServer) |
|||
err := XaClient.XaGlobalTransaction(gid, func(xa *dtmcli.Xa) (*resty.Response, error) { |
|||
resp, err := xa.CallBranch(&TransReq{Amount: 30}, Busi+"/TransOutXa") |
|||
if err != nil { |
|||
return resp, err |
|||
} |
|||
return xa.CallBranch(&TransReq{Amount: 30}, Busi+"/TransInXa") |
|||
}) |
|||
logger.FatalIfError(err) |
|||
return gid |
|||
}) |
|||
} |
|||
@ -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 main |
|||
|
|||
import ( |
|||
"flag" |
|||
"fmt" |
|||
"os" |
|||
"path/filepath" |
|||
|
|||
"go.uber.org/automaxprocs/maxprocs" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/dtmsvr" |
|||
"github.com/dtm-labs/dtm/dtmsvr/config" |
|||
"github.com/dtm-labs/dtm/dtmsvr/storage/registry" |
|||
|
|||
// load the microserver driver
|
|||
_ "github.com/dtm-labs/dtmdriver-gozero" |
|||
_ "github.com/dtm-labs/dtmdriver-polaris" |
|||
_ "github.com/dtm-labs/dtmdriver-protocol1" |
|||
) |
|||
|
|||
var Version string |
|||
|
|||
func version() { |
|||
if Version == "" { |
|||
Version = "0.0.0-dev" |
|||
} |
|||
fmt.Printf("dtm version: %s\n", Version) |
|||
} |
|||
|
|||
func usage() { |
|||
cmd := filepath.Base(os.Args[0]) |
|||
s := "Usage: %s [options]\n\n" |
|||
fmt.Fprintf(os.Stderr, s, cmd) |
|||
flag.PrintDefaults() |
|||
} |
|||
|
|||
var isVersion = flag.Bool("v", false, "Show the version of dtm.") |
|||
var isDebug = flag.Bool("d", false, "Set log level to debug.") |
|||
var isHelp = flag.Bool("h", false, "Show the help information about etcd.") |
|||
var isReset = flag.Bool("r", false, "Reset dtm server data.") |
|||
var confFile = flag.String("c", "", "Path to the server configuration file.") |
|||
|
|||
func main() { |
|||
flag.Parse() |
|||
if flag.NArg() > 0 || *isHelp { |
|||
usage() |
|||
return |
|||
} else if *isVersion { |
|||
version() |
|||
return |
|||
} |
|||
config.MustLoadConfig(*confFile) |
|||
if *isDebug { |
|||
config.Config.LogLevel = "debug" |
|||
} |
|||
if *isReset { |
|||
dtmsvr.PopulateDB(false) |
|||
} |
|||
maxprocs.Set(maxprocs.Logger(logger.Infof)) |
|||
registry.WaitStoreUp() |
|||
dtmsvr.StartSvr() // 启动dtmsvr的api服务
|
|||
go dtmsvr.CronExpiredTrans(-1) // 启动dtmsvr的定时过期查询
|
|||
select {} |
|||
} |
|||
@ -0,0 +1,18 @@ |
|||
/* |
|||
* 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 main |
|||
|
|||
import ( |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/test/busi" |
|||
) |
|||
|
|||
func main() { |
|||
busi.QsStartSvr() |
|||
gid := busi.QsFireRequest() |
|||
logger.Infof("transaction: %s succeed", gid) |
|||
} |
|||
@ -0,0 +1,46 @@ |
|||
CREATE DATABASE IF NOT EXISTS dtm |
|||
/*!40100 DEFAULT CHARACTER SET utf8mb4 */ |
|||
; |
|||
drop table IF EXISTS dtm.trans_global; |
|||
CREATE TABLE if not EXISTS dtm.trans_global ( |
|||
`id` bigint(22) NOT NULL AUTO_INCREMENT, |
|||
`gid` varchar(128) NOT NULL COMMENT '事务全局id', |
|||
`trans_type` varchar(45) not null COMMENT '事务类型: saga | xa | tcc | msg', |
|||
-- `data` TEXT COMMENT '事务携带的数据', -- 影响性能，不必要存储 |
|||
`status` varchar(12) NOT NULL COMMENT '全局事务的状态 prepared | submitted | aborting | finished | rollbacked', |
|||
`query_prepared` varchar(128) NOT NULL COMMENT 'prepared状态事务的查询api', |
|||
`protocol` varchar(45) not null comment '通信协议 http | grpc', |
|||
`create_time` datetime DEFAULT NULL, |
|||
`update_time` datetime DEFAULT NULL, |
|||
`commit_time` datetime DEFAULT NULL, |
|||
`finish_time` datetime DEFAULT NULL, |
|||
`rollback_time` datetime DEFAULT NULL, |
|||
`options` varchar(256) DEFAULT '', |
|||
`custom_data` varchar(256) DEFAULT '', |
|||
`next_cron_interval` int(11) default null comment '下次定时处理的间隔', |
|||
`next_cron_time` datetime default null comment '下次定时处理的时间', |
|||
`owner` varchar(128) not null default '' comment '正在处理全局事务的锁定者', |
|||
PRIMARY KEY (`id`,`gid`), |
|||
UNIQUE KEY `id` (`id`,`gid`), |
|||
UNIQUE KEY `gid` (`gid`), |
|||
key `owner`(`owner`), |
|||
key `status_next_cron_time` (`status`, `next_cron_time`) comment '这个索引用于查询超时的全局事务，能够合理的走索引' |
|||
) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 shardkey=gid; |
|||
drop table IF EXISTS dtm.trans_branch_op; |
|||
CREATE TABLE IF NOT EXISTS dtm.trans_branch_op ( |
|||
`id` bigint(22) NOT NULL AUTO_INCREMENT, |
|||
`gid` varchar(128) NOT NULL COMMENT '事务全局id', |
|||
`url` varchar(128) NOT NULL COMMENT '动作关联的url', |
|||
`data` TEXT COMMENT '请求所携带的数据', |
|||
`bin_data` BLOB COMMENT 'grpc的二进制数据', |
|||
`branch_id` VARCHAR(128) NOT NULL COMMENT '事务分支ID', |
|||
`op` varchar(45) NOT NULL COMMENT '事务分支类型 saga_action | saga_compensate | xa', |
|||
`status` varchar(45) NOT NULL COMMENT '步骤的状态 submitted | finished | rollbacked', |
|||
`finish_time` datetime DEFAULT NULL, |
|||
`rollback_time` datetime DEFAULT NULL, |
|||
`create_time` datetime DEFAULT NULL, |
|||
`update_time` datetime DEFAULT NULL, |
|||
PRIMARY KEY (`id`,`gid`), |
|||
UNIQUE KEY `id` (`id`,`gid`), |
|||
UNIQUE KEY `gid_uniq` (`gid`, `branch_id`, `op`) |
|||
) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 shardkey=gid; |
|||
@ -0,0 +1,124 @@ |
|||
/* |
|||
* 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 busi |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
"github.com/gin-gonic/gin" |
|||
emptypb "google.golang.org/protobuf/types/known/emptypb" |
|||
) |
|||
|
|||
func init() { |
|||
setupFuncs["TccBarrierSetup"] = func(app *gin.Engine) { |
|||
app.POST(BusiAPI+"/SagaBTransIn", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaAdjustBalance(tx, transInUID, reqFrom(c).Amount, reqFrom(c).TransInResult) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/SagaBTransInCompensate", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaAdjustBalance(tx, transInUID, -reqFrom(c).Amount, "") |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/SagaBTransOut", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaAdjustBalance(tx, transOutUID, -reqFrom(c).Amount, reqFrom(c).TransOutResult) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/SagaBTransOutCompensate", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
barrier := MustBarrierFromGin(c) |
|||
return dtmcli.MapSuccess, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaAdjustBalance(tx, transOutUID, reqFrom(c).Amount, "") |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/SagaBTransOutGorm", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) |
|||
barrier := MustBarrierFromGin(c) |
|||
tx := dbGet().DB.Begin() |
|||
return dtmcli.MapSuccess, barrier.Call(tx.Statement.ConnPool.(*sql.Tx), func(tx1 *sql.Tx) error { |
|||
return tx.Exec("update dtm_busi.user_account set balance = balance + ? where user_id = ?", -req.Amount, transOutUID).Error |
|||
}) |
|||
})) |
|||
|
|||
app.POST(BusiAPI+"/TccBTransInTry", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) // 去重构一下，改成可以重复使用的输入
|
|||
if req.TransInResult != "" { |
|||
return req.TransInResult, nil |
|||
} |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return tccAdjustTrading(tx, transInUID, req.Amount) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/TccBTransInConfirm", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return tccAdjustBalance(tx, transInUID, reqFrom(c).Amount) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/TccBTransInCancel", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return tccAdjustTrading(tx, transInUID, -reqFrom(c).Amount) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/TccBTransOutTry", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
req := reqFrom(c) |
|||
if req.TransOutResult != "" { |
|||
return req.TransOutResult, nil |
|||
} |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return tccAdjustTrading(tx, transOutUID, -req.Amount) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/TccBTransOutConfirm", dtmutil.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return tccAdjustBalance(tx, transOutUID, -reqFrom(c).Amount) |
|||
}) |
|||
})) |
|||
app.POST(BusiAPI+"/TccBTransOutCancel", dtmutil.WrapHandler(TccBarrierTransOutCancel)) |
|||
} |
|||
} |
|||
|
|||
// TccBarrierTransOutCancel will be use in test
|
|||
func TccBarrierTransOutCancel(c *gin.Context) (interface{}, error) { |
|||
return dtmcli.MapSuccess, MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { |
|||
return tccAdjustTrading(tx, transOutUID, reqFrom(c).Amount) |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransInBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcAdjustBalance(tx, transInUID, in.Amount, in.TransInResult) |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransOutBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcAdjustBalance(tx, transOutUID, -in.Amount, in.TransOutResult) |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransInRevertBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcAdjustBalance(tx, transInUID, -in.Amount, "") |
|||
}) |
|||
} |
|||
|
|||
func (s *busiServer) TransOutRevertBSaga(ctx context.Context, in *BusiReq) (*emptypb.Empty, error) { |
|||
barrier := MustBarrierFromGrpc(ctx) |
|||
return &emptypb.Empty{}, barrier.Call(txGet(), func(tx *sql.Tx) error { |
|||
return sagaGrpcAdjustBalance(tx, transOutUID, in.Amount, "") |
|||
}) |
|||
} |
|||
@ -0,0 +1,85 @@ |
|||
package busi |
|||
|
|||
import ( |
|||
"errors" |
|||
"fmt" |
|||
"strings" |
|||
|
|||
"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" |
|||
codes "google.golang.org/grpc/codes" |
|||
status "google.golang.org/grpc/status" |
|||
) |
|||
|
|||
const transOutUID = 1 |
|||
const transInUID = 2 |
|||
|
|||
func handleGrpcBusiness(in *BusiReq, result1 string, result2 string, busi string) error { |
|||
res := dtmimp.OrString(result1, result2, dtmcli.ResultSuccess) |
|||
logger.Debugf("grpc busi %s %v %s %s result: %s", busi, in, result1, result2, res) |
|||
if res == dtmcli.ResultSuccess { |
|||
return nil |
|||
} else if res == dtmcli.ResultFailure { |
|||
return status.New(codes.Aborted, dtmcli.ResultFailure).Err() |
|||
} else if res == dtmcli.ResultOngoing { |
|||
return status.New(codes.Aborted, dtmcli.ResultOngoing).Err() |
|||
} |
|||
return status.New(codes.Internal, fmt.Sprintf("unknow result %s", res)).Err() |
|||
} |
|||
|
|||
func handleGeneralBusiness(c *gin.Context, result1 string, result2 string, busi string) (interface{}, error) { |
|||
info := infoFromContext(c) |
|||
res := dtmimp.OrString(result1, result2, dtmcli.ResultSuccess) |
|||
logger.Debugf("%s %s result: %s", busi, info.String(), res) |
|||
if res == "ERROR" { |
|||
return nil, errors.New("ERROR from user") |
|||
} |
|||
return map[string]interface{}{"dtm_result": res}, nil |
|||
} |
|||
|
|||
func error2Resp(err error) (interface{}, error) { |
|||
if err != nil { |
|||
s := err.Error() |
|||
if strings.Contains(s, dtmcli.ResultFailure) || strings.Contains(s, dtmcli.ResultOngoing) { |
|||
return gin.H{"dtm_result": s}, nil |
|||
} |
|||
return nil, err |
|||
} |
|||
return gin.H{"dtm_result": dtmcli.ResultSuccess}, nil |
|||
} |
|||
|
|||
func sagaGrpcAdjustBalance(db dtmcli.DB, uid int, amount int64, result string) error { |
|||
if result == dtmcli.ResultFailure { |
|||
return status.New(codes.Aborted, dtmcli.ResultFailure).Err() |
|||
} |
|||
_, err := dtmimp.DBExec(db, "update dtm_busi.user_account set balance = balance + ? where user_id = ?", amount, uid) |
|||
return err |
|||
|
|||
} |
|||
|
|||
func sagaAdjustBalance(db dtmcli.DB, uid int, amount int, result string) error { |
|||
if strings.Contains(result, dtmcli.ResultFailure) { |
|||
return dtmcli.ErrFailure |
|||
} |
|||
_, err := dtmimp.DBExec(db, "update dtm_busi.user_account set balance = balance + ? where user_id = ?", amount, uid) |
|||
return err |
|||
} |
|||
func tccAdjustTrading(db dtmcli.DB, uid int, amount int) error { |
|||
affected, err := dtmimp.DBExec(db, `update dtm_busi.user_account set trading_balance=trading_balance+? |
|||
where user_id=? and trading_balance + ? + balance >= 0`, amount, uid, amount) |
|||
if err == nil && affected == 0 { |
|||
return fmt.Errorf("update error, maybe balance not enough") |
|||
} |
|||
return err |
|||
} |
|||
|
|||
func tccAdjustBalance(db dtmcli.DB, uid int, amount int) error { |
|||
affected, err := dtmimp.DBExec(db, `update dtm_busi.user_account set trading_balance=trading_balance-?, |
|||
balance=balance+? where user_id=?`, amount, amount, uid) |
|||
if err == nil && affected == 0 { |
|||
return fmt.Errorf("update user_account 0 rows") |
|||
} |
|||
return err |
|||
} |
|||
@ -0,0 +1,337 @@ |
|||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|||
// versions:
|
|||
// protoc-gen-go v1.27.1
|
|||
// protoc v3.17.3
|
|||
// source: test/busi/busi.proto
|
|||
|
|||
package busi |
|||
|
|||
import ( |
|||
protoreflect "google.golang.org/protobuf/reflect/protoreflect" |
|||
protoimpl "google.golang.org/protobuf/runtime/protoimpl" |
|||
emptypb "google.golang.org/protobuf/types/known/emptypb" |
|||
reflect "reflect" |
|||
sync "sync" |
|||
) |
|||
|
|||
const ( |
|||
// Verify that this generated code is sufficiently up-to-date.
|
|||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) |
|||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
|||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) |
|||
) |
|||
|
|||
// DtmRequest request sent to dtm server
|
|||
type BusiReq struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Amount int64 `protobuf:"varint,1,opt,name=Amount,proto3" json:"Amount,omitempty"` |
|||
TransOutResult string `protobuf:"bytes,2,opt,name=TransOutResult,proto3" json:"TransOutResult,omitempty"` |
|||
TransInResult string `protobuf:"bytes,3,opt,name=TransInResult,proto3" json:"TransInResult,omitempty"` |
|||
} |
|||
|
|||
func (x *BusiReq) Reset() { |
|||
*x = BusiReq{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_test_busi_busi_proto_msgTypes[0] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *BusiReq) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*BusiReq) ProtoMessage() {} |
|||
|
|||
func (x *BusiReq) ProtoReflect() protoreflect.Message { |
|||
mi := &file_test_busi_busi_proto_msgTypes[0] |
|||
if protoimpl.UnsafeEnabled && x != nil { |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
if ms.LoadMessageInfo() == nil { |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
return ms |
|||
} |
|||
return mi.MessageOf(x) |
|||
} |
|||
|
|||
// Deprecated: Use BusiReq.ProtoReflect.Descriptor instead.
|
|||
func (*BusiReq) Descriptor() ([]byte, []int) { |
|||
return file_test_busi_busi_proto_rawDescGZIP(), []int{0} |
|||
} |
|||
|
|||
func (x *BusiReq) GetAmount() int64 { |
|||
if x != nil { |
|||
return x.Amount |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func (x *BusiReq) GetTransOutResult() string { |
|||
if x != nil { |
|||
return x.TransOutResult |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *BusiReq) GetTransInResult() string { |
|||
if x != nil { |
|||
return x.TransInResult |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
type BusiReply struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Message string `protobuf:"bytes,1,opt,name=message,proto3" json:"message,omitempty"` |
|||
} |
|||
|
|||
func (x *BusiReply) Reset() { |
|||
*x = BusiReply{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_test_busi_busi_proto_msgTypes[1] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *BusiReply) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*BusiReply) ProtoMessage() {} |
|||
|
|||
func (x *BusiReply) ProtoReflect() protoreflect.Message { |
|||
mi := &file_test_busi_busi_proto_msgTypes[1] |
|||
if protoimpl.UnsafeEnabled && x != nil { |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
if ms.LoadMessageInfo() == nil { |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
return ms |
|||
} |
|||
return mi.MessageOf(x) |
|||
} |
|||
|
|||
// Deprecated: Use BusiReply.ProtoReflect.Descriptor instead.
|
|||
func (*BusiReply) Descriptor() ([]byte, []int) { |
|||
return file_test_busi_busi_proto_rawDescGZIP(), []int{1} |
|||
} |
|||
|
|||
func (x *BusiReply) GetMessage() string { |
|||
if x != nil { |
|||
return x.Message |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
var File_test_busi_busi_proto protoreflect.FileDescriptor |
|||
|
|||
var file_test_busi_busi_proto_rawDesc = []byte{ |
|||
0x0a, 0x14, 0x74, 0x65, 0x73, 0x74, 0x2f, 0x62, 0x75, 0x73, 0x69, 0x2f, 0x62, 0x75, 0x73, 0x69, |
|||
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x04, 0x62, 0x75, 0x73, 0x69, 0x1a, 0x1b, 0x67, 0x6f, |
|||
0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2f, 0x65, 0x6d, |
|||
0x70, 0x74, 0x79, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x6f, 0x0a, 0x07, 0x42, 0x75, 0x73, |
|||
0x69, 0x52, 0x65, 0x71, 0x12, 0x16, 0x0a, 0x06, 0x41, 0x6d, 0x6f, 0x75, 0x6e, 0x74, 0x18, 0x01, |
|||
0x20, 0x01, 0x28, 0x03, 0x52, 0x06, 0x41, 0x6d, 0x6f, 0x75, 0x6e, 0x74, 0x12, 0x26, 0x0a, 0x0e, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x18, 0x02, |
|||
0x20, 0x01, 0x28, 0x09, 0x52, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, |
|||
0x73, 0x75, 0x6c, 0x74, 0x12, 0x24, 0x0a, 0x0d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x52, |
|||
0x65, 0x73, 0x75, 0x6c, 0x74, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x54, 0x72, 0x61, |
|||
0x6e, 0x73, 0x49, 0x6e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x22, 0x25, 0x0a, 0x09, 0x42, 0x75, |
|||
0x73, 0x69, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, |
|||
0x67, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, |
|||
0x65, 0x32, 0xce, 0x08, 0x0a, 0x04, 0x42, 0x75, 0x73, 0x69, 0x12, 0x2d, 0x0a, 0x09, 0x43, 0x61, |
|||
0x6e, 0x53, 0x75, 0x62, 0x6d, 0x69, 0x74, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, |
|||
0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x0f, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, |
|||
0x73, 0x69, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x22, 0x00, 0x12, 0x32, 0x0a, 0x07, 0x54, 0x72, 0x61, |
|||
0x6e, 0x73, 0x49, 0x6e, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, |
|||
0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, |
|||
0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x33, 0x0a, |
|||
0x08, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, |
|||
0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
|||
0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, |
|||
0x22, 0x00, 0x12, 0x38, 0x0a, 0x0d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x52, 0x65, 0x76, |
|||
0x65, 0x72, 0x74, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, |
|||
0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, |
|||
0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x39, 0x0a, 0x0e, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x12, 0x0d, |
|||
0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, |
|||
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, |
|||
0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x39, 0x0a, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, |
|||
0x49, 0x6e, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x72, 0x6d, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, |
|||
0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
|||
0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, |
|||
0x22, 0x00, 0x12, 0x3a, 0x0a, 0x0f, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x43, 0x6f, |
|||
0x6e, 0x66, 0x69, 0x72, 0x6d, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, |
|||
0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, |
|||
0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3c, |
|||
0x0a, 0x08, 0x58, 0x61, 0x4e, 0x6f, 0x74, 0x69, 0x66, 0x79, 0x12, 0x16, 0x2e, 0x67, 0x6f, 0x6f, |
|||
0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, |
|||
0x74, 0x79, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, |
|||
0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x34, 0x0a, 0x09, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x58, 0x61, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, |
|||
0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
|||
0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, |
|||
0x22, 0x00, 0x12, 0x35, 0x0a, 0x0a, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x58, 0x61, |
|||
0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, |
|||
0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, |
|||
0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x35, 0x0a, 0x0a, 0x54, 0x72, 0x61, |
|||
0x6e, 0x73, 0x49, 0x6e, 0x54, 0x63, 0x63, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, |
|||
0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, |
|||
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, |
|||
0x12, 0x36, 0x0a, 0x0b, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x54, 0x63, 0x63, 0x12, |
|||
0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, |
|||
0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, |
|||
0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3b, 0x0a, 0x10, 0x54, 0x72, 0x61, 0x6e, |
|||
0x73, 0x49, 0x6e, 0x54, 0x63, 0x63, 0x4e, 0x65, 0x73, 0x74, 0x65, 0x64, 0x12, 0x0d, 0x2e, 0x62, |
|||
0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, |
|||
0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, |
|||
0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x37, 0x0a, 0x0c, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, |
|||
0x42, 0x53, 0x61, 0x67, 0x61, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, |
|||
0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, |
|||
0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x38, |
|||
0x0a, 0x0d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x42, 0x53, 0x61, 0x67, 0x61, 0x12, |
|||
0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, |
|||
0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, |
|||
0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3d, 0x0a, 0x12, 0x54, 0x72, 0x61, 0x6e, |
|||
0x73, 0x49, 0x6e, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x42, 0x53, 0x61, 0x67, 0x61, 0x12, 0x0d, |
|||
0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, |
|||
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, |
|||
0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3e, 0x0a, 0x13, 0x54, 0x72, 0x61, 0x6e, 0x73, |
|||
0x4f, 0x75, 0x74, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x42, 0x53, 0x61, 0x67, 0x61, 0x12, 0x0d, |
|||
0x2e, 0x62, 0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, |
|||
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, |
|||
0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3c, 0x0a, 0x11, 0x54, 0x72, 0x61, 0x6e, 0x73, |
|||
0x4f, 0x75, 0x74, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x59, 0x65, 0x73, 0x12, 0x0d, 0x2e, 0x62, |
|||
0x75, 0x73, 0x69, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, |
|||
0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, |
|||
0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3b, 0x0a, 0x10, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, |
|||
0x74, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x4e, 0x6f, 0x12, 0x0d, 0x2e, 0x62, 0x75, 0x73, 0x69, |
|||
0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
|||
0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, |
|||
0x22, 0x00, 0x42, 0x08, 0x5a, 0x06, 0x2e, 0x2f, 0x62, 0x75, 0x73, 0x69, 0x62, 0x06, 0x70, 0x72, |
|||
0x6f, 0x74, 0x6f, 0x33, |
|||
} |
|||
|
|||
var ( |
|||
file_test_busi_busi_proto_rawDescOnce sync.Once |
|||
file_test_busi_busi_proto_rawDescData = file_test_busi_busi_proto_rawDesc |
|||
) |
|||
|
|||
func file_test_busi_busi_proto_rawDescGZIP() []byte { |
|||
file_test_busi_busi_proto_rawDescOnce.Do(func() { |
|||
file_test_busi_busi_proto_rawDescData = protoimpl.X.CompressGZIP(file_test_busi_busi_proto_rawDescData) |
|||
}) |
|||
return file_test_busi_busi_proto_rawDescData |
|||
} |
|||
|
|||
var file_test_busi_busi_proto_msgTypes = make([]protoimpl.MessageInfo, 2) |
|||
var file_test_busi_busi_proto_goTypes = []interface{}{ |
|||
(*BusiReq)(nil), // 0: busi.BusiReq
|
|||
(*BusiReply)(nil), // 1: busi.BusiReply
|
|||
(*emptypb.Empty)(nil), // 2: google.protobuf.Empty
|
|||
} |
|||
var file_test_busi_busi_proto_depIdxs = []int32{ |
|||
0, // 0: busi.Busi.CanSubmit:input_type -> busi.BusiReq
|
|||
0, // 1: busi.Busi.TransIn:input_type -> busi.BusiReq
|
|||
0, // 2: busi.Busi.TransOut:input_type -> busi.BusiReq
|
|||
0, // 3: busi.Busi.TransInRevert:input_type -> busi.BusiReq
|
|||
0, // 4: busi.Busi.TransOutRevert:input_type -> busi.BusiReq
|
|||
0, // 5: busi.Busi.TransInConfirm:input_type -> busi.BusiReq
|
|||
0, // 6: busi.Busi.TransOutConfirm:input_type -> busi.BusiReq
|
|||
2, // 7: busi.Busi.XaNotify:input_type -> google.protobuf.Empty
|
|||
0, // 8: busi.Busi.TransInXa:input_type -> busi.BusiReq
|
|||
0, // 9: busi.Busi.TransOutXa:input_type -> busi.BusiReq
|
|||
0, // 10: busi.Busi.TransInTcc:input_type -> busi.BusiReq
|
|||
0, // 11: busi.Busi.TransOutTcc:input_type -> busi.BusiReq
|
|||
0, // 12: busi.Busi.TransInTccNested:input_type -> busi.BusiReq
|
|||
0, // 13: busi.Busi.TransInBSaga:input_type -> busi.BusiReq
|
|||
0, // 14: busi.Busi.TransOutBSaga:input_type -> busi.BusiReq
|
|||
0, // 15: busi.Busi.TransInRevertBSaga:input_type -> busi.BusiReq
|
|||
0, // 16: busi.Busi.TransOutRevertBSaga:input_type -> busi.BusiReq
|
|||
0, // 17: busi.Busi.TransOutHeaderYes:input_type -> busi.BusiReq
|
|||
0, // 18: busi.Busi.TransOutHeaderNo:input_type -> busi.BusiReq
|
|||
1, // 19: busi.Busi.CanSubmit:output_type -> busi.BusiReply
|
|||
2, // 20: busi.Busi.TransIn:output_type -> google.protobuf.Empty
|
|||
2, // 21: busi.Busi.TransOut:output_type -> google.protobuf.Empty
|
|||
2, // 22: busi.Busi.TransInRevert:output_type -> google.protobuf.Empty
|
|||
2, // 23: busi.Busi.TransOutRevert:output_type -> google.protobuf.Empty
|
|||
2, // 24: busi.Busi.TransInConfirm:output_type -> google.protobuf.Empty
|
|||
2, // 25: busi.Busi.TransOutConfirm:output_type -> google.protobuf.Empty
|
|||
2, // 26: busi.Busi.XaNotify:output_type -> google.protobuf.Empty
|
|||
2, // 27: busi.Busi.TransInXa:output_type -> google.protobuf.Empty
|
|||
2, // 28: busi.Busi.TransOutXa:output_type -> google.protobuf.Empty
|
|||
2, // 29: busi.Busi.TransInTcc:output_type -> google.protobuf.Empty
|
|||
2, // 30: busi.Busi.TransOutTcc:output_type -> google.protobuf.Empty
|
|||
2, // 31: busi.Busi.TransInTccNested:output_type -> google.protobuf.Empty
|
|||
2, // 32: busi.Busi.TransInBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 33: busi.Busi.TransOutBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 34: busi.Busi.TransInRevertBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 35: busi.Busi.TransOutRevertBSaga:output_type -> google.protobuf.Empty
|
|||
2, // 36: busi.Busi.TransOutHeaderYes:output_type -> google.protobuf.Empty
|
|||
2, // 37: busi.Busi.TransOutHeaderNo:output_type -> google.protobuf.Empty
|
|||
19, // [19:38] is the sub-list for method output_type
|
|||
0, // [0:19] is the sub-list for method input_type
|
|||
0, // [0:0] is the sub-list for extension type_name
|
|||
0, // [0:0] is the sub-list for extension extendee
|
|||
0, // [0:0] is the sub-list for field type_name
|
|||
} |
|||
|
|||
func init() { file_test_busi_busi_proto_init() } |
|||
func file_test_busi_busi_proto_init() { |
|||
if File_test_busi_busi_proto != nil { |
|||
return |
|||
} |
|||
if !protoimpl.UnsafeEnabled { |
|||
file_test_busi_busi_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*BusiReq); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_test_busi_busi_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*BusiReply); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
} |
|||
type x struct{} |
|||
out := protoimpl.TypeBuilder{ |
|||
File: protoimpl.DescBuilder{ |
|||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(), |
|||
RawDescriptor: file_test_busi_busi_proto_rawDesc, |
|||
NumEnums: 0, |
|||
NumMessages: 2, |
|||
NumExtensions: 0, |
|||
NumServices: 1, |
|||
}, |
|||
GoTypes: file_test_busi_busi_proto_goTypes, |
|||
DependencyIndexes: file_test_busi_busi_proto_depIdxs, |
|||
MessageInfos: file_test_busi_busi_proto_msgTypes, |
|||
}.Build() |
|||
File_test_busi_busi_proto = out.File |
|||
file_test_busi_busi_proto_rawDesc = nil |
|||
file_test_busi_busi_proto_goTypes = nil |
|||
file_test_busi_busi_proto_depIdxs = nil |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
package busi |
|||
|
|||
import ( |
|||
"fmt" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
"github.com/gin-gonic/gin" |
|||
) |
|||
|
|||
// Startup startup the busi's grpc and http service
|
|||
func Startup() *gin.Engine { |
|||
GrpcStartup() |
|||
return BaseAppStartup() |
|||
} |
|||
|
|||
// PopulateDB populate example mysql data
|
|||
func PopulateDB(skipDrop bool) { |
|||
resetXaData() |
|||
file := fmt.Sprintf("%s/busi.%s.sql", dtmutil.GetSqlDir(), BusiConf.Driver) |
|||
dtmutil.RunSQLScript(BusiConf, file, skipDrop) |
|||
file = fmt.Sprintf("%s/dtmcli.barrier.%s.sql", dtmutil.GetSqlDir(), BusiConf.Driver) |
|||
dtmutil.RunSQLScript(BusiConf, file, skipDrop) |
|||
} |
|||
@ -0,0 +1,85 @@ |
|||
package busi |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
"fmt" |
|||
"strings" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/dtmgrpc" |
|||
"github.com/dtm-labs/dtm/dtmgrpc/dtmgpb" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
"github.com/gin-gonic/gin" |
|||
"github.com/go-resty/resty/v2" |
|||
grpc "google.golang.org/grpc" |
|||
"google.golang.org/grpc/metadata" |
|||
) |
|||
|
|||
func dbGet() *dtmutil.DB { |
|||
return dtmutil.DbGet(BusiConf) |
|||
} |
|||
|
|||
func pdbGet() *sql.DB { |
|||
db, err := dtmimp.PooledDB(BusiConf) |
|||
logger.FatalIfError(err) |
|||
return db |
|||
} |
|||
|
|||
func txGet() *sql.Tx { |
|||
db := pdbGet() |
|||
tx, err := db.Begin() |
|||
logger.FatalIfError(err) |
|||
return tx |
|||
} |
|||
|
|||
func resetXaData() { |
|||
if BusiConf.Driver != "mysql" { |
|||
return |
|||
} |
|||
|
|||
db := dbGet() |
|||
type XaRow struct { |
|||
Data string |
|||
} |
|||
xas := []XaRow{} |
|||
db.Must().Raw("xa recover").Scan(&xas) |
|||
for _, xa := range xas { |
|||
db.Must().Exec(fmt.Sprintf("xa rollback '%s'", xa.Data)) |
|||
} |
|||
} |
|||
|
|||
// MustBarrierFromGin 1
|
|||
func MustBarrierFromGin(c *gin.Context) *dtmcli.BranchBarrier { |
|||
ti, err := dtmcli.BarrierFromQuery(c.Request.URL.Query()) |
|||
logger.FatalIfError(err) |
|||
return ti |
|||
} |
|||
|
|||
// MustBarrierFromGrpc 1
|
|||
func MustBarrierFromGrpc(ctx context.Context) *dtmcli.BranchBarrier { |
|||
ti, err := dtmgrpc.BarrierFromGrpc(ctx) |
|||
logger.FatalIfError(err) |
|||
return ti |
|||
} |
|||
|
|||
// SetGrpcHeaderForHeadersYes interceptor to set head for HeadersYes
|
|||
func SetGrpcHeaderForHeadersYes(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error { |
|||
if r, ok := req.(*dtmgpb.DtmRequest); ok && strings.HasSuffix(r.Gid, "HeadersYes") { |
|||
logger.Debugf("writing test_header:test to ctx") |
|||
md := metadata.New(map[string]string{"test_header": "test"}) |
|||
ctx = metadata.NewOutgoingContext(ctx, md) |
|||
} |
|||
return invoker(ctx, method, req, reply, cc, opts...) |
|||
} |
|||
|
|||
// SetHttpHeaderForHeadersYes interceptor to set head for HeadersYes
|
|||
func SetHttpHeaderForHeadersYes(c *resty.Client, r *resty.Request) error { |
|||
if b, ok := r.Body.(*dtmcli.Saga); ok && strings.HasSuffix(b.Gid, "HeadersYes") { |
|||
logger.Debugf("set test_header for url: %s", r.URL) |
|||
r.SetHeader("test_header", "yes") |
|||
} |
|||
return nil |
|||
} |
|||
@ -0,0 +1,48 @@ |
|||
package test |
|||
|
|||
import ( |
|||
"testing" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmgrpc" |
|||
"github.com/dtm-labs/dtm/dtmsvr/storage/sql" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
"github.com/stretchr/testify/assert" |
|||
) |
|||
|
|||
func TestGeneralDB(t *testing.T) { |
|||
if conf.Store.IsDB() { |
|||
testSql(t) |
|||
testDbAlone(t) |
|||
} |
|||
} |
|||
|
|||
func testSql(t *testing.T) { |
|||
conf := conf.Store.GetDBConf() |
|||
conf.Host = "127.0.0.1" // use a new host to trigger SetDBConn called
|
|||
db := dtmutil.DbGet(conf, sql.SetDBConn) |
|||
err := func() (rerr error) { |
|||
defer dtmimp.P2E(&rerr) |
|||
db.Must().Exec("select a") |
|||
return nil |
|||
}() |
|||
assert.NotEqual(t, nil, err) |
|||
} |
|||
|
|||
func testDbAlone(t *testing.T) { |
|||
db, err := dtmimp.StandaloneDB(conf.Store.GetDBConf()) |
|||
assert.Nil(t, err) |
|||
_, err = dtmimp.DBExec(db, "select 1") |
|||
assert.Equal(t, nil, err) |
|||
_, err = dtmimp.DBExec(db, "") |
|||
assert.Equal(t, nil, err) |
|||
db.Close() |
|||
_, err = dtmimp.DBExec(db, "select 1") |
|||
assert.NotEqual(t, nil, err) |
|||
} |
|||
|
|||
func TestMustGenGid(t *testing.T) { |
|||
dtmgrpc.MustGenGid(dtmutil.DefaultGrpcServer) |
|||
dtmcli.MustGenGid(dtmutil.DefaultHttpServer) |
|||
} |
|||
@ -1,20 +0,0 @@ |
|||
/* |
|||
* 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/examples" |
|||
) |
|||
|
|||
func TestExamples(t *testing.T) { |
|||
examples.QsStartSvr() |
|||
for _, s := range examples.Samples { |
|||
assertSucceed(t, s.Action()) |
|||
} |
|||
} |
|||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue