mirror of https://github.com/dtm-labs/dtm.git
yedf2
4 years ago
committed by
GitHub
GitHub
167 changed files with 5615 additions and 4232 deletions
@ -0,0 +1,70 @@ |
|||
# For most projects, this workflow file will not need changing; you simply need |
|||
# to commit it to your repository. |
|||
# |
|||
# You may wish to alter this file to override the set of languages analyzed, |
|||
# or to provide custom queries or build logic. |
|||
# |
|||
# ******** NOTE ******** |
|||
# We have attempted to detect the languages in your repository. Please check |
|||
# the `language` matrix defined below to confirm you have the correct set of |
|||
# supported CodeQL languages. |
|||
# |
|||
name: "CodeQL" |
|||
|
|||
on: |
|||
push: |
|||
branches: [ main ] |
|||
pull_request: |
|||
# The branches below must be a subset of the branches above |
|||
branches: [ main ] |
|||
schedule: |
|||
- cron: '25 19 * * 2' |
|||
|
|||
jobs: |
|||
analyze: |
|||
name: Analyze |
|||
runs-on: ubuntu-latest |
|||
permissions: |
|||
actions: read |
|||
contents: read |
|||
security-events: write |
|||
|
|||
strategy: |
|||
fail-fast: false |
|||
matrix: |
|||
language: [ 'go' ] |
|||
# CodeQL supports [ 'cpp', 'csharp', 'go', 'java', 'javascript', 'python', 'ruby' ] |
|||
# Learn more about CodeQL language support at https://git.io/codeql-language-support |
|||
|
|||
steps: |
|||
- name: Checkout repository |
|||
uses: actions/checkout@v2 |
|||
|
|||
# Initializes the CodeQL tools for scanning. |
|||
- name: Initialize CodeQL |
|||
uses: github/codeql-action/init@v1 |
|||
with: |
|||
languages: ${{ matrix.language }} |
|||
# If you wish to specify custom queries, you can do so here or in a config file. |
|||
# By default, queries listed here will override any specified in a config file. |
|||
# Prefix the list here with "+" to use these queries and those in the config file. |
|||
# queries: ./path/to/local/query, your-org/your-repo/queries@main |
|||
|
|||
# Autobuild attempts to build any compiled languages (C/C++, C#, or Java). |
|||
# If this step fails, then you should remove it and run the build manually (see below) |
|||
- name: Autobuild |
|||
uses: github/codeql-action/autobuild@v1 |
|||
|
|||
# ℹ️ Command-line programs to run using the OS shell. |
|||
# 📚 https://git.io/JvXDl |
|||
|
|||
# ✏️ If the Autobuild fails above, remove it and uncomment the following three lines |
|||
# and modify them (or add more) to build your code if your project |
|||
# uses a compiled language |
|||
|
|||
#- run: | |
|||
# make bootstrap |
|||
# make release |
|||
|
|||
- name: Perform CodeQL Analysis |
|||
uses: github/codeql-action/analyze@v1 |
|||
@ -0,0 +1,48 @@ |
|||
name: Tests |
|||
on: |
|||
push: |
|||
branches-ignore: |
|||
- 'tmp-*' |
|||
pull_request: |
|||
branches-ignore: |
|||
- 'tmp-*' |
|||
|
|||
jobs: |
|||
tests: |
|||
name: CI |
|||
runs-on: ubuntu-latest |
|||
services: |
|||
mysql: |
|||
image: 'mysql:5.7' |
|||
env: |
|||
MYSQL_ALLOW_EMPTY_PASSWORD: 1 |
|||
volumes: |
|||
- /etc/localtime:/etc/localtime:ro |
|||
- /etc/timezone:/etc/timezone:ro |
|||
ports: |
|||
- 3306:3306 |
|||
reids: |
|||
image: 'redis' |
|||
volumes: |
|||
- /etc/localtime:/etc/localtime:ro |
|||
- /etc/timezone:/etc/timezone:ro |
|||
ports: |
|||
- 6379:6379 |
|||
steps: |
|||
- name: Set up Go 1.16 |
|||
uses: actions/setup-go@v2 |
|||
with: |
|||
go-version: '1.16' |
|||
|
|||
- name: Check out code |
|||
uses: actions/checkout@v2 |
|||
|
|||
- name: Install dependencies |
|||
run: | |
|||
go mod download |
|||
|
|||
- name: Run CI lint |
|||
run: make lint |
|||
|
|||
- name: Run test cover |
|||
run: sh helper/test-cover.sh |
|||
@ -1,18 +0,0 @@ |
|||
language: go |
|||
go: |
|||
- 1.15.x |
|||
env: |
|||
- GO111MODULE=on |
|||
branches: |
|||
only: |
|||
- master |
|||
- main |
|||
- alpha |
|||
services: |
|||
- mysql |
|||
- redis-server |
|||
before_install: |
|||
- go get -t -v ./... |
|||
- go get github.com/yedf2/goveralls |
|||
script: |
|||
- $GOPATH/bin/goveralls -envs=TEST_STORE=redis,TEST_STORE=mysql,TEST_STORE=boltdb -flags '-gcflags=-l' -service=travis-ci -ignore="examples/*,dtmgrpc/dtmgimp/*.pb.go,bench/*,test/*" |
|||
@ -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/yedf/dtm/common" |
|||
"github.com/yedf/dtm/dtmcli" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
"github.com/yedf/dtm/dtmsvr" |
|||
"github.com/yedf/dtm/dtmsvr/storage/registry" |
|||
"github.com/yedf/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 |
|||
} |
|||
dtmimp.Logf("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": |
|||
default: |
|||
// 下面是各类的例子
|
|||
examples.GrpcStartup() |
|||
examples.BaseAppStartup() |
|||
|
|||
sample := examples.Samples[os.Args[1]] |
|||
dtmimp.LogIfFatalf(sample == nil, "no sample name for %s", os.Args[1]) |
|||
sample.Action() |
|||
} |
|||
select {} |
|||
} |
|||
@ -0,0 +1,28 @@ |
|||
# All targets.
|
|||
default: bench |
|||
|
|||
bench: /usr/local/bin/go /etc/redis/redis.conf /usr/local/bin/docker-compose main.go |
|||
rm -f ../conf.sample.yml |
|||
go build -o bench |
|||
|
|||
go: /usr/local/bin/go |
|||
|
|||
redis: /etc/redis/redis.conf |
|||
|
|||
mysql: /usr/local/bin/docker-compose |
|||
|
|||
/usr/local/bin/go: |
|||
wget https://golang.org/dl/go1.17.1.linux-amd64.tar.gz |
|||
rm -rf /usr/local/go && tar -C /usr/local -xzf go1.17.1.linux-amd64.tar.gz && cp -f /usr/local/go/bin/go /usr/local/bin/go && rm go1.* |
|||
|
|||
/etc/redis/redis.conf: |
|||
apt update |
|||
apt install -y redis redis-tools |
|||
|
|||
/usr/local/bin/docker-compose: |
|||
apt update |
|||
apt install -y sysbench apache2-utils mysql-client-core-8.0 |
|||
curl -fsSL https://get.docker.com | sh |
|||
curl -L "https://github.com/docker/compose/releases/download/1.29.2/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose |
|||
chmod +x /usr/local/bin/docker-compose |
|||
cd .. && docker-compose -f helper/compose.mysql.yml up -d && cd bench |
|||
@ -0,0 +1,10 @@ |
|||
# !/bin/bash |
|||
apt update |
|||
apt install -y git |
|||
git clone https://github.com/dtm-labs/dtm.git && cd dtm && git checkout alpha && cd bench && make |
|||
|
|||
|
|||
echo 'all prepared. you shoud run following commands to test in different terminal' |
|||
echo |
|||
echo 'cd dtm && go run bench/main.go redis|boltdb|db' |
|||
echo 'cd dtm && bench/run-redis|boltdb|mysql.sh' |
|||
@ -1,8 +0,0 @@ |
|||
# !/bin/bash |
|||
|
|||
cd /usr/share/sysbench/ |
|||
echo 'create database sbtest;' > mysql -h 127.0.0.1 -uroot |
|||
|
|||
sysbench oltp_write_only.lua --time=60 --mysql-host=127.0.0.1 --mysql-port=3306 --mysql-user=root --mysql-password= --mysql-db=sbtest --table-size=1000000 --tables=10 --threads=10 --events=999999999 --report-interval=10 prepare |
|||
|
|||
sysbench oltp_write_only.lua --time=60 --mysql-host=127.0.0.1 --mysql-port=3306 --mysql-user=root --mysql-password= --mysql-db=sbtest --table-size=1000000 --tables=10 --threads=10 --events=999999999 --report-interval=10 run |
|||
@ -1,5 +0,0 @@ |
|||
# !/bin/bash |
|||
|
|||
# start all services |
|||
docker-compose -f helper/compose.mysql.yml up -d |
|||
go run app/main.go bench > /dev/nul |
|||
@ -0,0 +1,6 @@ |
|||
# !/bin/bash |
|||
apt update |
|||
apt install -y software-properties-common |
|||
add-apt-repository -y ppa:redislabs/redis |
|||
apt install -y redis redis-tools |
|||
|
|||
@ -0,0 +1,5 @@ |
|||
# !/bin/bash |
|||
|
|||
set -x |
|||
|
|||
ab -n 50000 -c 10 "http://127.0.0.1:8083/api/busi_bench/benchEmptyUrl" |
|||
@ -0,0 +1,27 @@ |
|||
# !/bin/bash |
|||
|
|||
set -x |
|||
|
|||
export LOG_LEVEL=warn |
|||
export STORE_DRIVER=redis |
|||
export STORE_HOST=localhost |
|||
export STORE_PORT=6379 |
|||
cd .. && bench/bench redis & |
|||
echo 'sleeping 3s for dtm bench to run up.' && sleep 3 |
|||
ab -n 1000000 -c 10 "http://127.0.0.1:8083/api/busi_bench/benchEmptyUrl" |
|||
pkill bench |
|||
|
|||
redis-benchmark -n 300000 SET 'abcdefg' 'ddddddd' |
|||
|
|||
redis-benchmark -n 300000 EVAL "redis.call('SET', 'abcdedf', 'ddddddd')" 0 |
|||
|
|||
redis-benchmark -n 300000 EVAL "redis.call('SET', KEYS[1], ARGV[1])" 1 'aaaaaaaaa' 'bbbbbbbbbb' |
|||
|
|||
redis-benchmark -n 3000000 -P 50 SET 'abcdefg' 'ddddddd' |
|||
|
|||
redis-benchmark -n 300000 EVAL "for k=1, 10 do; redis.call('SET', KEYS[1], ARGV[1]);end" 1 'aaaaaaaaa' 'bbbbbbbbbb' |
|||
|
|||
redis-benchmark -n 300000 -P 50 EVAL "redis.call('SET', KEYS[1], ARGV[1])" 1 'aaaaaaaaa' 'bbbbbbbbbb' |
|||
|
|||
redis-benchmark -n 300000 EVAL "for k=1,10 do;local c = cjson.decode(ARGV[1]);end" 1 'aaaaaaaaa' '{"aaaaa":"bbbbb","b":1,"t":"2012-01-01 14:00:00"}' |
|||
|
|||
@ -1,120 +0,0 @@ |
|||
package common |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"errors" |
|||
"io/ioutil" |
|||
"path/filepath" |
|||
|
|||
"github.com/yedf/dtm/dtmcli" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
"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, |
|||
Passwrod: 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"` |
|||
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) |
|||
dtmimp.FatalIfError(err) |
|||
} |
|||
scont, err := json.MarshalIndent(&Config, "", " ") |
|||
dtmimp.FatalIfError(err) |
|||
dtmimp.Logf("config is: \n%s", scont) |
|||
err = checkConfig() |
|||
dtmimp.LogIfFatalf(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") |
|||
} |
|||
if Config.Store.Driver == BoltDb { |
|||
return nil |
|||
} |
|||
if Config.Store.Driver == 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 ") |
|||
} |
|||
} |
|||
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,30 +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 ( |
|||
"fmt" |
|||
"sync" |
|||
|
|||
"github.com/go-redis/redis/v8" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
) |
|||
|
|||
var rdb *redis.Client |
|||
var once sync.Once |
|||
|
|||
func RedisGet() *redis.Client { |
|||
once.Do(func() { |
|||
dtmimp.Logf("connecting to redis: %v", Config.Store) |
|||
rdb = redis.NewClient(&redis.Options{ |
|||
Addr: fmt.Sprintf("%s:%d", Config.Store.Host, Config.Store.Port), |
|||
Username: Config.Store.User, |
|||
Password: Config.Store.Password, |
|||
}) |
|||
}) |
|||
return rdb |
|||
} |
|||
@ -1,66 +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/stretchr/testify/assert" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
) |
|||
|
|||
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 |
|||
} |
|||
@ -1,121 +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 ( |
|||
"bytes" |
|||
"encoding/json" |
|||
"io/ioutil" |
|||
"os" |
|||
"path/filepath" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/go-resty/resty/v2" |
|||
|
|||
"github.com/yedf/dtm/dtmcli" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
) |
|||
|
|||
// GetGinApp init and return gin
|
|||
func GetGinApp() *gin.Engine { |
|||
gin.SetMode(gin.ReleaseMode) |
|||
app := gin.Default() |
|||
app.Use(func(c *gin.Context) { |
|||
body := "" |
|||
if c.Request.Body != nil { |
|||
rb, err := c.GetRawData() |
|||
dtmimp.E2P(err) |
|||
if len(rb) > 0 { |
|||
body = string(rb) |
|||
c.Request.Body = ioutil.NopCloser(bytes.NewBuffer(rb)) |
|||
} |
|||
} |
|||
began := time.Now() |
|||
dtmimp.Logf("begin %s %s query: %s body: %s", c.Request.Method, c.FullPath(), c.Request.URL.RawQuery, body) |
|||
c.Next() |
|||
dtmimp.Logf("used %d ms %s %s query: %s body: %s", time.Since(began).Milliseconds(), c.Request.Method, c.FullPath(), c.Request.URL.RawQuery, body) |
|||
|
|||
}) |
|||
app.Any("/api/ping", func(c *gin.Context) { c.JSON(200, map[string]interface{}{"msg": "pong"}) }) |
|||
return app |
|||
} |
|||
|
|||
// WrapHandler name is clear
|
|||
func WrapHandler(fn func(*gin.Context) (interface{}, error)) gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
r, err := func() (r interface{}, rerr error) { |
|||
defer dtmimp.P2E(&rerr) |
|||
return fn(c) |
|||
}() |
|||
var b = []byte{} |
|||
if resp, ok := r.(*resty.Response); ok { // 如果是response，则取出body直接处理
|
|||
b = resp.Body() |
|||
} else if err == nil { |
|||
b, err = json.Marshal(r) |
|||
} |
|||
if err != nil { |
|||
dtmimp.Logf("status: 500, code: 500 message: %s", err.Error()) |
|||
c.JSON(500, map[string]interface{}{"code": 500, "message": err.Error()}) |
|||
} else { |
|||
dtmimp.Logf("status: 200, content: %s", string(b)) |
|||
c.Status(200) |
|||
c.Writer.Header().Add("Content-Type", "application/json") |
|||
_, err = c.Writer.Write(b) |
|||
dtmimp.E2P(err) |
|||
} |
|||
} |
|||
} |
|||
|
|||
// MustGetwd must version of os.Getwd
|
|||
func MustGetwd() string { |
|||
wd, err := os.Getwd() |
|||
dtmimp.E2P(err) |
|||
return wd |
|||
} |
|||
|
|||
// GetSqlDir 获取调用该函数的caller源代码的目录，主要用于测试时，查找相关文件
|
|||
func GetSqlDir() string { |
|||
wd := MustGetwd() |
|||
if filepath.Base(wd) == "test" { |
|||
wd = filepath.Dir(wd) |
|||
} |
|||
return wd + "/sqls" |
|||
} |
|||
|
|||
func RecoverPanic(err *error) { |
|||
if x := recover(); x != nil { |
|||
e := dtmimp.AsError(x) |
|||
if err != nil { |
|||
*err = e |
|||
} |
|||
} |
|||
} |
|||
|
|||
func GetNextTime(second int64) *time.Time { |
|||
next := time.Now().Add(time.Duration(second) * time.Second) |
|||
return &next |
|||
} |
|||
|
|||
// RunSQLScript 1
|
|||
func RunSQLScript(conf dtmcli.DBConf, script string, skipDrop bool) { |
|||
con, err := dtmimp.StandaloneDB(conf) |
|||
dtmimp.FatalIfError(err) |
|||
defer func() { con.Close() }() |
|||
content, err := ioutil.ReadFile(script) |
|||
dtmimp.FatalIfError(err) |
|||
sqls := strings.Split(string(content), ";") |
|||
for _, sql := range sqls { |
|||
s := strings.TrimSpace(sql) |
|||
if s == "" || (skipDrop && strings.Contains(s, "drop")) { |
|||
continue |
|||
} |
|||
_, err = dtmimp.DBExec(con, s) |
|||
dtmimp.FatalIfError(err) |
|||
} |
|||
} |
|||
@ -0,0 +1,2 @@ |
|||
## 注意 |
|||
此包带imp后缀，主要被dtm内部使用，相关接口可能会发生变更，请勿使用这里的接口 |
|||
@ -0,0 +1,111 @@ |
|||
package logger |
|||
|
|||
import ( |
|||
"fmt" |
|||
"log" |
|||
"net/url" |
|||
"os" |
|||
|
|||
"github.com/natefinch/lumberjack" |
|||
"go.uber.org/zap" |
|||
"go.uber.org/zap/zapcore" |
|||
) |
|||
|
|||
//var logger *zap.SugaredLogger = nil
|
|||
|
|||
var logger Logger |
|||
|
|||
func init() { |
|||
InitLog(os.Getenv("LOG_LEVEL")) |
|||
} |
|||
|
|||
// Logger logger interface
|
|||
type Logger interface { |
|||
Debugf(format string, args ...interface{}) |
|||
Infof(format string, args ...interface{}) |
|||
Warnf(format string, args ...interface{}) |
|||
Errorf(format string, args ...interface{}) |
|||
} |
|||
|
|||
type lumberjackSink struct { |
|||
*lumberjack.Logger |
|||
} |
|||
|
|||
func (lumberjackSink) Sync() error { |
|||
return nil |
|||
} |
|||
|
|||
// WithLogger replaces default logger
|
|||
func WithLogger(log Logger) { |
|||
logger = log |
|||
} |
|||
|
|||
// InitLog is an initialization for a logger
|
|||
// level can be: debug info warn error
|
|||
func InitLog(level string) { |
|||
config := loadConfig(level) |
|||
p, err := config.Build(zap.AddCallerSkip(1)) |
|||
FatalIfError(err) |
|||
logger = p.Sugar() |
|||
} |
|||
|
|||
// InitRotateLog is an initialization for a rotated logger by lumberjack
|
|||
func InitRotateLog(logLevel string, ll *lumberjack.Logger) { |
|||
config := loadConfig(logLevel) |
|||
config.OutputPaths = []string{fmt.Sprintf("lumberjack:%s", ll.Filename), "stdout"} |
|||
err := zap.RegisterSink("lumberjack", func(*url.URL) (zap.Sink, error) { |
|||
return lumberjackSink{ |
|||
Logger: ll, |
|||
}, nil |
|||
}) |
|||
FatalIfError(err) |
|||
|
|||
p, err := config.Build(zap.AddCallerSkip(1)) |
|||
FatalIfError(err) |
|||
logger = p.Sugar() |
|||
} |
|||
|
|||
func loadConfig(logLevel string) zap.Config { |
|||
config := zap.NewProductionConfig() |
|||
err := config.Level.UnmarshalText([]byte(logLevel)) |
|||
FatalIfError(err) |
|||
config.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder |
|||
if os.Getenv("DTM_DEBUG") != "" { |
|||
config.Encoding = "console" |
|||
config.EncoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder |
|||
} |
|||
return config |
|||
} |
|||
|
|||
// Debugf log to level debug
|
|||
func Debugf(fmt string, args ...interface{}) { |
|||
logger.Debugf(fmt, args...) |
|||
} |
|||
|
|||
// Infof log to level info
|
|||
func Infof(fmt string, args ...interface{}) { |
|||
logger.Infof(fmt, args...) |
|||
} |
|||
|
|||
// Warnf log to level warn
|
|||
func Warnf(fmt string, args ...interface{}) { |
|||
logger.Warnf(fmt, args...) |
|||
} |
|||
|
|||
// Errorf log to level error
|
|||
func Errorf(fmt string, args ...interface{}) { |
|||
logger.Errorf(fmt, args...) |
|||
} |
|||
|
|||
// FatalfIf log to level error
|
|||
func FatalfIf(cond bool, fmt string, args ...interface{}) { |
|||
if !cond { |
|||
return |
|||
} |
|||
log.Fatalf(fmt, args...) |
|||
} |
|||
|
|||
// FatalIfError if err is not nil, then log to level fatal and call os.Exit
|
|||
func FatalIfError(err error) { |
|||
FatalfIf(err != nil, "fatal error: %v", err) |
|||
} |
|||
@ -0,0 +1,52 @@ |
|||
package logger |
|||
|
|||
import ( |
|||
"os" |
|||
"testing" |
|||
|
|||
"github.com/natefinch/lumberjack" |
|||
"go.uber.org/zap" |
|||
) |
|||
|
|||
func TestInitLog(t *testing.T) { |
|||
os.Setenv("DTM_DEBUG", "1") |
|||
InitLog("debug") |
|||
Debugf("a debug msg") |
|||
Infof("a info msg") |
|||
Warnf("a warn msg") |
|||
Errorf("a error msg") |
|||
FatalfIf(false, "nothing") |
|||
FatalIfError(nil) |
|||
} |
|||
|
|||
func TestWithLogger(t *testing.T) { |
|||
logger := zap.NewExample().Sugar() |
|||
WithLogger(logger) |
|||
Debugf("a debug msg") |
|||
Infof("a info msg") |
|||
Warnf("a warn msg") |
|||
Errorf("a error msg") |
|||
FatalfIf(false, "nothing") |
|||
FatalIfError(nil) |
|||
} |
|||
|
|||
func TestInitRotateLog(t *testing.T) { |
|||
os.Setenv("DTM_DEBUG", "1") |
|||
ll := lumberjack.Logger{ |
|||
Filename: "test.log", |
|||
MaxSize: 1, |
|||
MaxBackups: 1, |
|||
MaxAge: 1, |
|||
Compress: false, |
|||
} |
|||
InitRotateLog("debug", &ll) |
|||
Debugf("a debug msg") |
|||
Infof("a info msg") |
|||
Warnf("a warn msg") |
|||
Errorf("a error msg") |
|||
FatalfIf(false, "nothing") |
|||
FatalIfError(nil) |
|||
s := lumberjackSink{&ll} |
|||
_ = s.Sync() |
|||
_ = os.Remove("test.log") |
|||
} |
|||
@ -0,0 +1,2 @@ |
|||
## 注意 |
|||
此包带imp后缀，主要被dtm内部使用，相关接口可能会发生变更，请勿使用这里的接口 |
|||
@ -1,510 +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. |
|||
*/ |
|||
|
|||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|||
// versions:
|
|||
// protoc-gen-go v1.27.1
|
|||
// protoc v3.17.3
|
|||
// source: dtmgrpc/dtmgimp/dtmgimp.proto
|
|||
|
|||
package dtmgimp |
|||
|
|||
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) |
|||
) |
|||
|
|||
type DtmTransOptions struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
WaitResult bool `protobuf:"varint,1,opt,name=WaitResult,proto3" json:"WaitResult,omitempty"` |
|||
TimeoutToFail int64 `protobuf:"varint,2,opt,name=TimeoutToFail,proto3" json:"TimeoutToFail,omitempty"` |
|||
RetryInterval int64 `protobuf:"varint,3,opt,name=RetryInterval,proto3" json:"RetryInterval,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmTransOptions) Reset() { |
|||
*x = DtmTransOptions{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[0] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmTransOptions) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmTransOptions) ProtoMessage() {} |
|||
|
|||
func (x *DtmTransOptions) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_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 DtmTransOptions.ProtoReflect.Descriptor instead.
|
|||
func (*DtmTransOptions) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescGZIP(), []int{0} |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetWaitResult() bool { |
|||
if x != nil { |
|||
return x.WaitResult |
|||
} |
|||
return false |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetTimeoutToFail() int64 { |
|||
if x != nil { |
|||
return x.TimeoutToFail |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetRetryInterval() int64 { |
|||
if x != nil { |
|||
return x.RetryInterval |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
// DtmRequest request sent to dtm server
|
|||
type DtmRequest struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Gid string `protobuf:"bytes,1,opt,name=Gid,proto3" json:"Gid,omitempty"` |
|||
TransType string `protobuf:"bytes,2,opt,name=TransType,proto3" json:"TransType,omitempty"` |
|||
TransOptions *DtmTransOptions `protobuf:"bytes,3,opt,name=TransOptions,proto3" json:"TransOptions,omitempty"` |
|||
CustomedData string `protobuf:"bytes,4,opt,name=CustomedData,proto3" json:"CustomedData,omitempty"` |
|||
BinPayloads [][]byte `protobuf:"bytes,5,rep,name=BinPayloads,proto3" json:"BinPayloads,omitempty"` // for MSG/SAGA branch payloads
|
|||
QueryPrepared string `protobuf:"bytes,6,opt,name=QueryPrepared,proto3" json:"QueryPrepared,omitempty"` // for MSG
|
|||
Steps string `protobuf:"bytes,7,opt,name=Steps,proto3" json:"Steps,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmRequest) Reset() { |
|||
*x = DtmRequest{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[1] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmRequest) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmRequest) ProtoMessage() {} |
|||
|
|||
func (x *DtmRequest) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_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 DtmRequest.ProtoReflect.Descriptor instead.
|
|||
func (*DtmRequest) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescGZIP(), []int{1} |
|||
} |
|||
|
|||
func (x *DtmRequest) GetGid() string { |
|||
if x != nil { |
|||
return x.Gid |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetTransType() string { |
|||
if x != nil { |
|||
return x.TransType |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetTransOptions() *DtmTransOptions { |
|||
if x != nil { |
|||
return x.TransOptions |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmRequest) GetCustomedData() string { |
|||
if x != nil { |
|||
return x.CustomedData |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetBinPayloads() [][]byte { |
|||
if x != nil { |
|||
return x.BinPayloads |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmRequest) GetQueryPrepared() string { |
|||
if x != nil { |
|||
return x.QueryPrepared |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetSteps() string { |
|||
if x != nil { |
|||
return x.Steps |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
type DtmGidReply struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Gid string `protobuf:"bytes,1,opt,name=Gid,proto3" json:"Gid,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmGidReply) Reset() { |
|||
*x = DtmGidReply{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[2] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmGidReply) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmGidReply) ProtoMessage() {} |
|||
|
|||
func (x *DtmGidReply) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[2] |
|||
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 DtmGidReply.ProtoReflect.Descriptor instead.
|
|||
func (*DtmGidReply) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescGZIP(), []int{2} |
|||
} |
|||
|
|||
func (x *DtmGidReply) GetGid() string { |
|||
if x != nil { |
|||
return x.Gid |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
type DtmBranchRequest struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Gid string `protobuf:"bytes,1,opt,name=Gid,proto3" json:"Gid,omitempty"` |
|||
TransType string `protobuf:"bytes,2,opt,name=TransType,proto3" json:"TransType,omitempty"` |
|||
BranchID string `protobuf:"bytes,3,opt,name=BranchID,proto3" json:"BranchID,omitempty"` |
|||
Op string `protobuf:"bytes,4,opt,name=Op,proto3" json:"Op,omitempty"` |
|||
Data map[string]string `protobuf:"bytes,5,rep,name=Data,proto3" json:"Data,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` |
|||
BusiPayload []byte `protobuf:"bytes,6,opt,name=BusiPayload,proto3" json:"BusiPayload,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) Reset() { |
|||
*x = DtmBranchRequest{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[3] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmBranchRequest) ProtoMessage() {} |
|||
|
|||
func (x *DtmBranchRequest) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[3] |
|||
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 DtmBranchRequest.ProtoReflect.Descriptor instead.
|
|||
func (*DtmBranchRequest) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescGZIP(), []int{3} |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetGid() string { |
|||
if x != nil { |
|||
return x.Gid |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetTransType() string { |
|||
if x != nil { |
|||
return x.TransType |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetBranchID() string { |
|||
if x != nil { |
|||
return x.BranchID |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetOp() string { |
|||
if x != nil { |
|||
return x.Op |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetData() map[string]string { |
|||
if x != nil { |
|||
return x.Data |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetBusiPayload() []byte { |
|||
if x != nil { |
|||
return x.BusiPayload |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
var File_dtmgrpc_dtmgimp_dtmgimp_proto protoreflect.FileDescriptor |
|||
|
|||
var file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDesc = []byte{ |
|||
0x0a, 0x1d, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, |
|||
0x70, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, |
|||
0x07, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 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, 0x7d, 0x0a, 0x0f, 0x44, 0x74, 0x6d, 0x54, 0x72, 0x61, 0x6e, |
|||
0x73, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x12, 0x1e, 0x0a, 0x0a, 0x57, 0x61, 0x69, 0x74, |
|||
0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x0a, 0x57, 0x61, |
|||
0x69, 0x74, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x12, 0x24, 0x0a, 0x0d, 0x54, 0x69, 0x6d, 0x65, |
|||
0x6f, 0x75, 0x74, 0x54, 0x6f, 0x46, 0x61, 0x69, 0x6c, 0x18, 0x02, 0x20, 0x01, 0x28, 0x03, 0x52, |
|||
0x0d, 0x54, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x54, 0x6f, 0x46, 0x61, 0x69, 0x6c, 0x12, 0x24, |
|||
0x0a, 0x0d, 0x52, 0x65, 0x74, 0x72, 0x79, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c, 0x18, |
|||
0x03, 0x20, 0x01, 0x28, 0x03, 0x52, 0x0d, 0x52, 0x65, 0x74, 0x72, 0x79, 0x49, 0x6e, 0x74, 0x65, |
|||
0x72, 0x76, 0x61, 0x6c, 0x22, 0xfc, 0x01, 0x0a, 0x0a, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, |
|||
0x65, 0x73, 0x74, 0x12, 0x10, 0x0a, 0x03, 0x47, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, |
|||
0x52, 0x03, 0x47, 0x69, 0x64, 0x12, 0x1c, 0x0a, 0x09, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x54, 0x79, |
|||
0x70, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x54, |
|||
0x79, 0x70, 0x65, 0x12, 0x3c, 0x0a, 0x0c, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x70, 0x74, 0x69, |
|||
0x6f, 0x6e, 0x73, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x18, 0x2e, 0x64, 0x74, 0x6d, 0x67, |
|||
0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x70, 0x74, 0x69, |
|||
0x6f, 0x6e, 0x73, 0x52, 0x0c, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, |
|||
0x73, 0x12, 0x22, 0x0a, 0x0c, 0x43, 0x75, 0x73, 0x74, 0x6f, 0x6d, 0x65, 0x64, 0x44, 0x61, 0x74, |
|||
0x61, 0x18, 0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x43, 0x75, 0x73, 0x74, 0x6f, 0x6d, 0x65, |
|||
0x64, 0x44, 0x61, 0x74, 0x61, 0x12, 0x20, 0x0a, 0x0b, 0x42, 0x69, 0x6e, 0x50, 0x61, 0x79, 0x6c, |
|||
0x6f, 0x61, 0x64, 0x73, 0x18, 0x05, 0x20, 0x03, 0x28, 0x0c, 0x52, 0x0b, 0x42, 0x69, 0x6e, 0x50, |
|||
0x61, 0x79, 0x6c, 0x6f, 0x61, 0x64, 0x73, 0x12, 0x24, 0x0a, 0x0d, 0x51, 0x75, 0x65, 0x72, 0x79, |
|||
0x50, 0x72, 0x65, 0x70, 0x61, 0x72, 0x65, 0x64, 0x18, 0x06, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, |
|||
0x51, 0x75, 0x65, 0x72, 0x79, 0x50, 0x72, 0x65, 0x70, 0x61, 0x72, 0x65, 0x64, 0x12, 0x14, 0x0a, |
|||
0x05, 0x53, 0x74, 0x65, 0x70, 0x73, 0x18, 0x07, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x53, 0x74, |
|||
0x65, 0x70, 0x73, 0x22, 0x1f, 0x0a, 0x0b, 0x44, 0x74, 0x6d, 0x47, 0x69, 0x64, 0x52, 0x65, 0x70, |
|||
0x6c, 0x79, 0x12, 0x10, 0x0a, 0x03, 0x47, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, |
|||
0x03, 0x47, 0x69, 0x64, 0x22, 0x82, 0x02, 0x0a, 0x10, 0x44, 0x74, 0x6d, 0x42, 0x72, 0x61, 0x6e, |
|||
0x63, 0x68, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x10, 0x0a, 0x03, 0x47, 0x69, 0x64, |
|||
0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x47, 0x69, 0x64, 0x12, 0x1c, 0x0a, 0x09, 0x54, |
|||
0x72, 0x61, 0x6e, 0x73, 0x54, 0x79, 0x70, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, |
|||
0x54, 0x72, 0x61, 0x6e, 0x73, 0x54, 0x79, 0x70, 0x65, 0x12, 0x1a, 0x0a, 0x08, 0x42, 0x72, 0x61, |
|||
0x6e, 0x63, 0x68, 0x49, 0x44, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x42, 0x72, 0x61, |
|||
0x6e, 0x63, 0x68, 0x49, 0x44, 0x12, 0x0e, 0x0a, 0x02, 0x4f, 0x70, 0x18, 0x04, 0x20, 0x01, 0x28, |
|||
0x09, 0x52, 0x02, 0x4f, 0x70, 0x12, 0x37, 0x0a, 0x04, 0x44, 0x61, 0x74, 0x61, 0x18, 0x05, 0x20, |
|||
0x03, 0x28, 0x0b, 0x32, 0x23, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, |
|||
0x6d, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x2e, 0x44, |
|||
0x61, 0x74, 0x61, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x52, 0x04, 0x44, 0x61, 0x74, 0x61, 0x12, 0x20, |
|||
0x0a, 0x0b, 0x42, 0x75, 0x73, 0x69, 0x50, 0x61, 0x79, 0x6c, 0x6f, 0x61, 0x64, 0x18, 0x06, 0x20, |
|||
0x01, 0x28, 0x0c, 0x52, 0x0b, 0x42, 0x75, 0x73, 0x69, 0x50, 0x61, 0x79, 0x6c, 0x6f, 0x61, 0x64, |
|||
0x1a, 0x37, 0x0a, 0x09, 0x44, 0x61, 0x74, 0x61, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x12, 0x10, 0x0a, |
|||
0x03, 0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x6b, 0x65, 0x79, 0x12, |
|||
0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, |
|||
0x76, 0x61, 0x6c, 0x75, 0x65, 0x3a, 0x02, 0x38, 0x01, 0x32, 0xb1, 0x02, 0x0a, 0x03, 0x44, 0x74, |
|||
0x6d, 0x12, 0x38, 0x0a, 0x06, 0x4e, 0x65, 0x77, 0x47, 0x69, 0x64, 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, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, |
|||
0x6d, 0x47, 0x69, 0x64, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x22, 0x00, 0x12, 0x37, 0x0a, 0x06, 0x53, |
|||
0x75, 0x62, 0x6d, 0x69, 0x74, 0x12, 0x13, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, |
|||
0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 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, 0x07, 0x50, 0x72, 0x65, 0x70, 0x61, 0x72, 0x65, 0x12, |
|||
0x13, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, |
|||
0x75, 0x65, 0x73, 0x74, 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, 0x05, 0x41, 0x62, 0x6f, 0x72, 0x74, 0x12, 0x13, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, |
|||
0x70, 0x2e, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 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, 0x45, 0x0a, 0x0e, 0x52, 0x65, 0x67, 0x69, 0x73, 0x74, |
|||
0x65, 0x72, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x12, 0x19, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, |
|||
0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x52, 0x65, 0x71, 0x75, |
|||
0x65, 0x73, 0x74, 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, 0x0b, 0x5a, |
|||
0x09, 0x2e, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, |
|||
0x6f, 0x33, |
|||
} |
|||
|
|||
var ( |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescOnce sync.Once |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescData = file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDesc |
|||
) |
|||
|
|||
func file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescGZIP() []byte { |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescOnce.Do(func() { |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescData = protoimpl.X.CompressGZIP(file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescData) |
|||
}) |
|||
return file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDescData |
|||
} |
|||
|
|||
var file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes = make([]protoimpl.MessageInfo, 5) |
|||
var file_dtmgrpc_dtmgimp_dtmgimp_proto_goTypes = []interface{}{ |
|||
(*DtmTransOptions)(nil), // 0: dtmgimp.DtmTransOptions
|
|||
(*DtmRequest)(nil), // 1: dtmgimp.DtmRequest
|
|||
(*DtmGidReply)(nil), // 2: dtmgimp.DtmGidReply
|
|||
(*DtmBranchRequest)(nil), // 3: dtmgimp.DtmBranchRequest
|
|||
nil, // 4: dtmgimp.DtmBranchRequest.DataEntry
|
|||
(*emptypb.Empty)(nil), // 5: google.protobuf.Empty
|
|||
} |
|||
var file_dtmgrpc_dtmgimp_dtmgimp_proto_depIdxs = []int32{ |
|||
0, // 0: dtmgimp.DtmRequest.TransOptions:type_name -> dtmgimp.DtmTransOptions
|
|||
4, // 1: dtmgimp.DtmBranchRequest.Data:type_name -> dtmgimp.DtmBranchRequest.DataEntry
|
|||
5, // 2: dtmgimp.Dtm.NewGid:input_type -> google.protobuf.Empty
|
|||
1, // 3: dtmgimp.Dtm.Submit:input_type -> dtmgimp.DtmRequest
|
|||
1, // 4: dtmgimp.Dtm.Prepare:input_type -> dtmgimp.DtmRequest
|
|||
1, // 5: dtmgimp.Dtm.Abort:input_type -> dtmgimp.DtmRequest
|
|||
3, // 6: dtmgimp.Dtm.RegisterBranch:input_type -> dtmgimp.DtmBranchRequest
|
|||
2, // 7: dtmgimp.Dtm.NewGid:output_type -> dtmgimp.DtmGidReply
|
|||
5, // 8: dtmgimp.Dtm.Submit:output_type -> google.protobuf.Empty
|
|||
5, // 9: dtmgimp.Dtm.Prepare:output_type -> google.protobuf.Empty
|
|||
5, // 10: dtmgimp.Dtm.Abort:output_type -> google.protobuf.Empty
|
|||
5, // 11: dtmgimp.Dtm.RegisterBranch:output_type -> google.protobuf.Empty
|
|||
7, // [7:12] is the sub-list for method output_type
|
|||
2, // [2:7] is the sub-list for method input_type
|
|||
2, // [2:2] is the sub-list for extension type_name
|
|||
2, // [2:2] is the sub-list for extension extendee
|
|||
0, // [0:2] is the sub-list for field type_name
|
|||
} |
|||
|
|||
func init() { file_dtmgrpc_dtmgimp_dtmgimp_proto_init() } |
|||
func file_dtmgrpc_dtmgimp_dtmgimp_proto_init() { |
|||
if File_dtmgrpc_dtmgimp_dtmgimp_proto != nil { |
|||
return |
|||
} |
|||
if !protoimpl.UnsafeEnabled { |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmTransOptions); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmRequest); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmGidReply); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmBranchRequest); 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_dtmgrpc_dtmgimp_dtmgimp_proto_rawDesc, |
|||
NumEnums: 0, |
|||
NumMessages: 5, |
|||
NumExtensions: 0, |
|||
NumServices: 1, |
|||
}, |
|||
GoTypes: file_dtmgrpc_dtmgimp_dtmgimp_proto_goTypes, |
|||
DependencyIndexes: file_dtmgrpc_dtmgimp_dtmgimp_proto_depIdxs, |
|||
MessageInfos: file_dtmgrpc_dtmgimp_dtmgimp_proto_msgTypes, |
|||
}.Build() |
|||
File_dtmgrpc_dtmgimp_dtmgimp_proto = out.File |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_rawDesc = nil |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_goTypes = nil |
|||
file_dtmgrpc_dtmgimp_dtmgimp_proto_depIdxs = nil |
|||
} |
|||
@ -0,0 +1,534 @@ |
|||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|||
// versions:
|
|||
// protoc-gen-go v1.27.1
|
|||
// protoc v3.17.3
|
|||
// source: dtmgrpc/dtmgpb/dtmgimp.proto
|
|||
|
|||
package dtmgpb |
|||
|
|||
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) |
|||
) |
|||
|
|||
type DtmTransOptions struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
WaitResult bool `protobuf:"varint,1,opt,name=WaitResult,proto3" json:"WaitResult,omitempty"` |
|||
TimeoutToFail int64 `protobuf:"varint,2,opt,name=TimeoutToFail,proto3" json:"TimeoutToFail,omitempty"` |
|||
RetryInterval int64 `protobuf:"varint,3,opt,name=RetryInterval,proto3" json:"RetryInterval,omitempty"` |
|||
PassthroughHeaders []string `protobuf:"bytes,4,rep,name=PassthroughHeaders,proto3" json:"PassthroughHeaders,omitempty"` |
|||
BranchHeaders map[string]string `protobuf:"bytes,5,rep,name=BranchHeaders,proto3" json:"BranchHeaders,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` |
|||
} |
|||
|
|||
func (x *DtmTransOptions) Reset() { |
|||
*x = DtmTransOptions{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[0] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmTransOptions) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmTransOptions) ProtoMessage() {} |
|||
|
|||
func (x *DtmTransOptions) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_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 DtmTransOptions.ProtoReflect.Descriptor instead.
|
|||
func (*DtmTransOptions) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescGZIP(), []int{0} |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetWaitResult() bool { |
|||
if x != nil { |
|||
return x.WaitResult |
|||
} |
|||
return false |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetTimeoutToFail() int64 { |
|||
if x != nil { |
|||
return x.TimeoutToFail |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetRetryInterval() int64 { |
|||
if x != nil { |
|||
return x.RetryInterval |
|||
} |
|||
return 0 |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetPassthroughHeaders() []string { |
|||
if x != nil { |
|||
return x.PassthroughHeaders |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmTransOptions) GetBranchHeaders() map[string]string { |
|||
if x != nil { |
|||
return x.BranchHeaders |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
// DtmRequest request sent to dtm server
|
|||
type DtmRequest struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Gid string `protobuf:"bytes,1,opt,name=Gid,proto3" json:"Gid,omitempty"` |
|||
TransType string `protobuf:"bytes,2,opt,name=TransType,proto3" json:"TransType,omitempty"` |
|||
TransOptions *DtmTransOptions `protobuf:"bytes,3,opt,name=TransOptions,proto3" json:"TransOptions,omitempty"` |
|||
CustomedData string `protobuf:"bytes,4,opt,name=CustomedData,proto3" json:"CustomedData,omitempty"` |
|||
BinPayloads [][]byte `protobuf:"bytes,5,rep,name=BinPayloads,proto3" json:"BinPayloads,omitempty"` // for MSG/SAGA branch payloads
|
|||
QueryPrepared string `protobuf:"bytes,6,opt,name=QueryPrepared,proto3" json:"QueryPrepared,omitempty"` // for MSG
|
|||
Steps string `protobuf:"bytes,7,opt,name=Steps,proto3" json:"Steps,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmRequest) Reset() { |
|||
*x = DtmRequest{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[1] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmRequest) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmRequest) ProtoMessage() {} |
|||
|
|||
func (x *DtmRequest) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_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 DtmRequest.ProtoReflect.Descriptor instead.
|
|||
func (*DtmRequest) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescGZIP(), []int{1} |
|||
} |
|||
|
|||
func (x *DtmRequest) GetGid() string { |
|||
if x != nil { |
|||
return x.Gid |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetTransType() string { |
|||
if x != nil { |
|||
return x.TransType |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetTransOptions() *DtmTransOptions { |
|||
if x != nil { |
|||
return x.TransOptions |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmRequest) GetCustomedData() string { |
|||
if x != nil { |
|||
return x.CustomedData |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetBinPayloads() [][]byte { |
|||
if x != nil { |
|||
return x.BinPayloads |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmRequest) GetQueryPrepared() string { |
|||
if x != nil { |
|||
return x.QueryPrepared |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmRequest) GetSteps() string { |
|||
if x != nil { |
|||
return x.Steps |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
type DtmGidReply struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Gid string `protobuf:"bytes,1,opt,name=Gid,proto3" json:"Gid,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmGidReply) Reset() { |
|||
*x = DtmGidReply{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[2] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmGidReply) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmGidReply) ProtoMessage() {} |
|||
|
|||
func (x *DtmGidReply) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[2] |
|||
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 DtmGidReply.ProtoReflect.Descriptor instead.
|
|||
func (*DtmGidReply) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescGZIP(), []int{2} |
|||
} |
|||
|
|||
func (x *DtmGidReply) GetGid() string { |
|||
if x != nil { |
|||
return x.Gid |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
type DtmBranchRequest struct { |
|||
state protoimpl.MessageState |
|||
sizeCache protoimpl.SizeCache |
|||
unknownFields protoimpl.UnknownFields |
|||
|
|||
Gid string `protobuf:"bytes,1,opt,name=Gid,proto3" json:"Gid,omitempty"` |
|||
TransType string `protobuf:"bytes,2,opt,name=TransType,proto3" json:"TransType,omitempty"` |
|||
BranchID string `protobuf:"bytes,3,opt,name=BranchID,proto3" json:"BranchID,omitempty"` |
|||
Op string `protobuf:"bytes,4,opt,name=Op,proto3" json:"Op,omitempty"` |
|||
Data map[string]string `protobuf:"bytes,5,rep,name=Data,proto3" json:"Data,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` |
|||
BusiPayload []byte `protobuf:"bytes,6,opt,name=BusiPayload,proto3" json:"BusiPayload,omitempty"` |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) Reset() { |
|||
*x = DtmBranchRequest{} |
|||
if protoimpl.UnsafeEnabled { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[3] |
|||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
|||
ms.StoreMessageInfo(mi) |
|||
} |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) String() string { |
|||
return protoimpl.X.MessageStringOf(x) |
|||
} |
|||
|
|||
func (*DtmBranchRequest) ProtoMessage() {} |
|||
|
|||
func (x *DtmBranchRequest) ProtoReflect() protoreflect.Message { |
|||
mi := &file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[3] |
|||
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 DtmBranchRequest.ProtoReflect.Descriptor instead.
|
|||
func (*DtmBranchRequest) Descriptor() ([]byte, []int) { |
|||
return file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescGZIP(), []int{3} |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetGid() string { |
|||
if x != nil { |
|||
return x.Gid |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetTransType() string { |
|||
if x != nil { |
|||
return x.TransType |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetBranchID() string { |
|||
if x != nil { |
|||
return x.BranchID |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetOp() string { |
|||
if x != nil { |
|||
return x.Op |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetData() map[string]string { |
|||
if x != nil { |
|||
return x.Data |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (x *DtmBranchRequest) GetBusiPayload() []byte { |
|||
if x != nil { |
|||
return x.BusiPayload |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
var File_dtmgrpc_dtmgpb_dtmgimp_proto protoreflect.FileDescriptor |
|||
|
|||
var file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDesc = []byte{ |
|||
0x0a, 0x1c, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x70, 0x62, |
|||
0x2f, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x07, |
|||
0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 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, 0xc2, 0x02, 0x0a, 0x0f, 0x44, 0x74, 0x6d, 0x54, 0x72, 0x61, 0x6e, |
|||
0x73, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x12, 0x1e, 0x0a, 0x0a, 0x57, 0x61, 0x69, 0x74, |
|||
0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x0a, 0x57, 0x61, |
|||
0x69, 0x74, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x12, 0x24, 0x0a, 0x0d, 0x54, 0x69, 0x6d, 0x65, |
|||
0x6f, 0x75, 0x74, 0x54, 0x6f, 0x46, 0x61, 0x69, 0x6c, 0x18, 0x02, 0x20, 0x01, 0x28, 0x03, 0x52, |
|||
0x0d, 0x54, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x54, 0x6f, 0x46, 0x61, 0x69, 0x6c, 0x12, 0x24, |
|||
0x0a, 0x0d, 0x52, 0x65, 0x74, 0x72, 0x79, 0x49, 0x6e, 0x74, 0x65, 0x72, 0x76, 0x61, 0x6c, 0x18, |
|||
0x03, 0x20, 0x01, 0x28, 0x03, 0x52, 0x0d, 0x52, 0x65, 0x74, 0x72, 0x79, 0x49, 0x6e, 0x74, 0x65, |
|||
0x72, 0x76, 0x61, 0x6c, 0x12, 0x2e, 0x0a, 0x12, 0x50, 0x61, 0x73, 0x73, 0x74, 0x68, 0x72, 0x6f, |
|||
0x75, 0x67, 0x68, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x73, 0x18, 0x04, 0x20, 0x03, 0x28, 0x09, |
|||
0x52, 0x12, 0x50, 0x61, 0x73, 0x73, 0x74, 0x68, 0x72, 0x6f, 0x75, 0x67, 0x68, 0x48, 0x65, 0x61, |
|||
0x64, 0x65, 0x72, 0x73, 0x12, 0x51, 0x0a, 0x0d, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x48, 0x65, |
|||
0x61, 0x64, 0x65, 0x72, 0x73, 0x18, 0x05, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x2b, 0x2e, 0x64, 0x74, |
|||
0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x70, |
|||
0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x48, 0x65, 0x61, 0x64, |
|||
0x65, 0x72, 0x73, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x52, 0x0d, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, |
|||
0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x73, 0x1a, 0x40, 0x0a, 0x12, 0x42, 0x72, 0x61, 0x6e, 0x63, |
|||
0x68, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x73, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x12, 0x10, 0x0a, |
|||
0x03, 0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x6b, 0x65, 0x79, 0x12, |
|||
0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, |
|||
0x76, 0x61, 0x6c, 0x75, 0x65, 0x3a, 0x02, 0x38, 0x01, 0x22, 0xfc, 0x01, 0x0a, 0x0a, 0x44, 0x74, |
|||
0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x10, 0x0a, 0x03, 0x47, 0x69, 0x64, 0x18, |
|||
0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x47, 0x69, 0x64, 0x12, 0x1c, 0x0a, 0x09, 0x54, 0x72, |
|||
0x61, 0x6e, 0x73, 0x54, 0x79, 0x70, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x54, |
|||
0x72, 0x61, 0x6e, 0x73, 0x54, 0x79, 0x70, 0x65, 0x12, 0x3c, 0x0a, 0x0c, 0x54, 0x72, 0x61, 0x6e, |
|||
0x73, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x18, |
|||
0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x54, 0x72, 0x61, 0x6e, |
|||
0x73, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x52, 0x0c, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, |
|||
0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x12, 0x22, 0x0a, 0x0c, 0x43, 0x75, 0x73, 0x74, 0x6f, 0x6d, |
|||
0x65, 0x64, 0x44, 0x61, 0x74, 0x61, 0x18, 0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x43, 0x75, |
|||
0x73, 0x74, 0x6f, 0x6d, 0x65, 0x64, 0x44, 0x61, 0x74, 0x61, 0x12, 0x20, 0x0a, 0x0b, 0x42, 0x69, |
|||
0x6e, 0x50, 0x61, 0x79, 0x6c, 0x6f, 0x61, 0x64, 0x73, 0x18, 0x05, 0x20, 0x03, 0x28, 0x0c, 0x52, |
|||
0x0b, 0x42, 0x69, 0x6e, 0x50, 0x61, 0x79, 0x6c, 0x6f, 0x61, 0x64, 0x73, 0x12, 0x24, 0x0a, 0x0d, |
|||
0x51, 0x75, 0x65, 0x72, 0x79, 0x50, 0x72, 0x65, 0x70, 0x61, 0x72, 0x65, 0x64, 0x18, 0x06, 0x20, |
|||
0x01, 0x28, 0x09, 0x52, 0x0d, 0x51, 0x75, 0x65, 0x72, 0x79, 0x50, 0x72, 0x65, 0x70, 0x61, 0x72, |
|||
0x65, 0x64, 0x12, 0x14, 0x0a, 0x05, 0x53, 0x74, 0x65, 0x70, 0x73, 0x18, 0x07, 0x20, 0x01, 0x28, |
|||
0x09, 0x52, 0x05, 0x53, 0x74, 0x65, 0x70, 0x73, 0x22, 0x1f, 0x0a, 0x0b, 0x44, 0x74, 0x6d, 0x47, |
|||
0x69, 0x64, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x12, 0x10, 0x0a, 0x03, 0x47, 0x69, 0x64, 0x18, 0x01, |
|||
0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x47, 0x69, 0x64, 0x22, 0x82, 0x02, 0x0a, 0x10, 0x44, 0x74, |
|||
0x6d, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x10, |
|||
0x0a, 0x03, 0x47, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x47, 0x69, 0x64, |
|||
0x12, 0x1c, 0x0a, 0x09, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x54, 0x79, 0x70, 0x65, 0x18, 0x02, 0x20, |
|||
0x01, 0x28, 0x09, 0x52, 0x09, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x54, 0x79, 0x70, 0x65, 0x12, 0x1a, |
|||
0x0a, 0x08, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x49, 0x44, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, |
|||
0x52, 0x08, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x49, 0x44, 0x12, 0x0e, 0x0a, 0x02, 0x4f, 0x70, |
|||
0x18, 0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, 0x4f, 0x70, 0x12, 0x37, 0x0a, 0x04, 0x44, 0x61, |
|||
0x74, 0x61, 0x18, 0x05, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x23, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, |
|||
0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x52, 0x65, 0x71, 0x75, |
|||
0x65, 0x73, 0x74, 0x2e, 0x44, 0x61, 0x74, 0x61, 0x45, 0x6e, 0x74, 0x72, 0x79, 0x52, 0x04, 0x44, |
|||
0x61, 0x74, 0x61, 0x12, 0x20, 0x0a, 0x0b, 0x42, 0x75, 0x73, 0x69, 0x50, 0x61, 0x79, 0x6c, 0x6f, |
|||
0x61, 0x64, 0x18, 0x06, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x0b, 0x42, 0x75, 0x73, 0x69, 0x50, 0x61, |
|||
0x79, 0x6c, 0x6f, 0x61, 0x64, 0x1a, 0x37, 0x0a, 0x09, 0x44, 0x61, 0x74, 0x61, 0x45, 0x6e, 0x74, |
|||
0x72, 0x79, 0x12, 0x10, 0x0a, 0x03, 0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, |
|||
0x03, 0x6b, 0x65, 0x79, 0x12, 0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, |
|||
0x01, 0x28, 0x09, 0x52, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x3a, 0x02, 0x38, 0x01, 0x32, 0xb1, |
|||
0x02, 0x0a, 0x03, 0x44, 0x74, 0x6d, 0x12, 0x38, 0x0a, 0x06, 0x4e, 0x65, 0x77, 0x47, 0x69, 0x64, |
|||
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, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, |
|||
0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x47, 0x69, 0x64, 0x52, 0x65, 0x70, 0x6c, 0x79, 0x22, 0x00, |
|||
0x12, 0x37, 0x0a, 0x06, 0x53, 0x75, 0x62, 0x6d, 0x69, 0x74, 0x12, 0x13, 0x2e, 0x64, 0x74, 0x6d, |
|||
0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 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, 0x07, 0x50, 0x72, 0x65, |
|||
0x70, 0x61, 0x72, 0x65, 0x12, 0x13, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, |
|||
0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 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, 0x05, 0x41, 0x62, 0x6f, 0x72, 0x74, 0x12, 0x13, 0x2e, 0x64, |
|||
0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, |
|||
0x74, 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, 0x45, 0x0a, 0x0e, 0x52, |
|||
0x65, 0x67, 0x69, 0x73, 0x74, 0x65, 0x72, 0x42, 0x72, 0x61, 0x6e, 0x63, 0x68, 0x12, 0x19, 0x2e, |
|||
0x64, 0x74, 0x6d, 0x67, 0x69, 0x6d, 0x70, 0x2e, 0x44, 0x74, 0x6d, 0x42, 0x72, 0x61, 0x6e, 0x63, |
|||
0x68, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 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, 0x0a, 0x5a, 0x08, 0x2e, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x70, 0x62, 0x62, 0x06, |
|||
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, |
|||
} |
|||
|
|||
var ( |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescOnce sync.Once |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescData = file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDesc |
|||
) |
|||
|
|||
func file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescGZIP() []byte { |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescOnce.Do(func() { |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescData = protoimpl.X.CompressGZIP(file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescData) |
|||
}) |
|||
return file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDescData |
|||
} |
|||
|
|||
var file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes = make([]protoimpl.MessageInfo, 6) |
|||
var file_dtmgrpc_dtmgpb_dtmgimp_proto_goTypes = []interface{}{ |
|||
(*DtmTransOptions)(nil), // 0: dtmgimp.DtmTransOptions
|
|||
(*DtmRequest)(nil), // 1: dtmgimp.DtmRequest
|
|||
(*DtmGidReply)(nil), // 2: dtmgimp.DtmGidReply
|
|||
(*DtmBranchRequest)(nil), // 3: dtmgimp.DtmBranchRequest
|
|||
nil, // 4: dtmgimp.DtmTransOptions.BranchHeadersEntry
|
|||
nil, // 5: dtmgimp.DtmBranchRequest.DataEntry
|
|||
(*emptypb.Empty)(nil), // 6: google.protobuf.Empty
|
|||
} |
|||
var file_dtmgrpc_dtmgpb_dtmgimp_proto_depIdxs = []int32{ |
|||
4, // 0: dtmgimp.DtmTransOptions.BranchHeaders:type_name -> dtmgimp.DtmTransOptions.BranchHeadersEntry
|
|||
0, // 1: dtmgimp.DtmRequest.TransOptions:type_name -> dtmgimp.DtmTransOptions
|
|||
5, // 2: dtmgimp.DtmBranchRequest.Data:type_name -> dtmgimp.DtmBranchRequest.DataEntry
|
|||
6, // 3: dtmgimp.Dtm.NewGid:input_type -> google.protobuf.Empty
|
|||
1, // 4: dtmgimp.Dtm.Submit:input_type -> dtmgimp.DtmRequest
|
|||
1, // 5: dtmgimp.Dtm.Prepare:input_type -> dtmgimp.DtmRequest
|
|||
1, // 6: dtmgimp.Dtm.Abort:input_type -> dtmgimp.DtmRequest
|
|||
3, // 7: dtmgimp.Dtm.RegisterBranch:input_type -> dtmgimp.DtmBranchRequest
|
|||
2, // 8: dtmgimp.Dtm.NewGid:output_type -> dtmgimp.DtmGidReply
|
|||
6, // 9: dtmgimp.Dtm.Submit:output_type -> google.protobuf.Empty
|
|||
6, // 10: dtmgimp.Dtm.Prepare:output_type -> google.protobuf.Empty
|
|||
6, // 11: dtmgimp.Dtm.Abort:output_type -> google.protobuf.Empty
|
|||
6, // 12: dtmgimp.Dtm.RegisterBranch:output_type -> google.protobuf.Empty
|
|||
8, // [8:13] is the sub-list for method output_type
|
|||
3, // [3:8] is the sub-list for method input_type
|
|||
3, // [3:3] is the sub-list for extension type_name
|
|||
3, // [3:3] is the sub-list for extension extendee
|
|||
0, // [0:3] is the sub-list for field type_name
|
|||
} |
|||
|
|||
func init() { file_dtmgrpc_dtmgpb_dtmgimp_proto_init() } |
|||
func file_dtmgrpc_dtmgpb_dtmgimp_proto_init() { |
|||
if File_dtmgrpc_dtmgpb_dtmgimp_proto != nil { |
|||
return |
|||
} |
|||
if !protoimpl.UnsafeEnabled { |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmTransOptions); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmRequest); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmGidReply); i { |
|||
case 0: |
|||
return &v.state |
|||
case 1: |
|||
return &v.sizeCache |
|||
case 2: |
|||
return &v.unknownFields |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} { |
|||
switch v := v.(*DtmBranchRequest); 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_dtmgrpc_dtmgpb_dtmgimp_proto_rawDesc, |
|||
NumEnums: 0, |
|||
NumMessages: 6, |
|||
NumExtensions: 0, |
|||
NumServices: 1, |
|||
}, |
|||
GoTypes: file_dtmgrpc_dtmgpb_dtmgimp_proto_goTypes, |
|||
DependencyIndexes: file_dtmgrpc_dtmgpb_dtmgimp_proto_depIdxs, |
|||
MessageInfos: file_dtmgrpc_dtmgpb_dtmgimp_proto_msgTypes, |
|||
}.Build() |
|||
File_dtmgrpc_dtmgpb_dtmgimp_proto = out.File |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_rawDesc = nil |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_goTypes = nil |
|||
file_dtmgrpc_dtmgpb_dtmgimp_proto_depIdxs = nil |
|||
} |
|||
@ -0,0 +1,107 @@ |
|||
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 is metric port
|
|||
DtmMetricsPort = 8889 |
|||
// Mysql is mysql driver
|
|||
Mysql = "mysql" |
|||
// Redis is redis driver
|
|||
Redis = "redis" |
|||
// BoltDb is boltdb driver
|
|||
BoltDb = "boltdb" |
|||
// Postgres is postgres driver
|
|||
Postgres = "postgres" |
|||
) |
|||
|
|||
// MicroService config type for micro service
|
|||
type MicroService struct { |
|||
Driver string `yaml:"Driver" default:"default"` |
|||
Target string `yaml:"Target"` |
|||
EndPoint string `yaml:"EndPoint"` |
|||
} |
|||
|
|||
// Log config customize log
|
|||
type Log struct { |
|||
Level string `yaml:"Level" default:"info"` |
|||
Output string `yaml:"Output" default:"console"` |
|||
FileName string `yaml:"FileName" default:"/tmp/dtm.log"` |
|||
FileMaxSize int64 `yaml:"FileMaxSize" default:"10"` |
|||
FileMaxBackups int64 `yaml:"FileMaxBackups" default:"5"` |
|||
FileMaxAge int64 `yaml:"FileMaxAge" default:"30"` |
|||
FileCompress int64 `yaml:"FileCompress" default:"0"` |
|||
} |
|||
|
|||
// Store defines storage relevant info
|
|||
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:"TransBranchOpTable" default:"dtm.trans_branch_op"` |
|||
} |
|||
|
|||
// IsDB checks config driver is mysql or postgres
|
|||
func (s *Store) IsDB() bool { |
|||
return s.Driver == dtmcli.DBTypeMysql || s.Driver == dtmcli.DBTypePostgres |
|||
} |
|||
|
|||
// GetDBConf returns db conf info
|
|||
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"` |
|||
RequestTimeout int64 `yaml:"RequestTimeout" default:"3"` |
|||
HTTPPort int64 `yaml:"HttpPort" default:"36789"` |
|||
GrpcPort int64 `yaml:"GrpcPort" default:"36790"` |
|||
MicroService MicroService `yaml:"MicroService"` |
|||
UpdateBranchSync int64 `yaml:"UpdateBranchSync"` |
|||
UpdateBranchAsyncGoroutineNum int64 `yaml:"UpdateBranchAsyncGoroutineNum" default:"1"` |
|||
Log Log `yaml:"Log"` |
|||
} |
|||
|
|||
// 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,257 +0,0 @@ |
|||
package storage |
|||
|
|||
import ( |
|||
"context" |
|||
"fmt" |
|||
"time" |
|||
|
|||
"github.com/go-redis/redis/v8" |
|||
"github.com/yedf/dtm/common" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
"gorm.io/gorm" |
|||
) |
|||
|
|||
var ctx context.Context = context.Background() |
|||
|
|||
type RedisStore struct { |
|||
} |
|||
|
|||
func (s *RedisStore) Ping() error { |
|||
_, err := redisGet().Ping(ctx).Result() |
|||
return err |
|||
} |
|||
|
|||
func (s *RedisStore) PopulateData(skipDrop bool) { |
|||
_, err := redisGet().FlushAll(ctx).Result() |
|||
dtmimp.PanicIf(err != nil, err) |
|||
} |
|||
|
|||
func (s *RedisStore) FindTransGlobalStore(gid string) *TransGlobalStore { |
|||
r, err := redisGet().Get(ctx, config.Store.RedisPrefix+"_g_"+gid).Result() |
|||
if err == redis.Nil { |
|||
return nil |
|||
} |
|||
dtmimp.E2P(err) |
|||
trans := &TransGlobalStore{} |
|||
dtmimp.MustUnmarshalString(r, trans) |
|||
return trans |
|||
} |
|||
|
|||
func (s *RedisStore) ScanTransGlobalStores(position *string, limit int64) []TransGlobalStore { |
|||
lid := uint64(0) |
|||
if *position != "" { |
|||
lid = uint64(dtmimp.MustAtoi(*position)) |
|||
} |
|||
keys, cursor, err := redisGet().Scan(ctx, lid, config.Store.RedisPrefix+"_g_*", limit).Result() |
|||
dtmimp.E2P(err) |
|||
globals := []TransGlobalStore{} |
|||
if len(keys) > 0 { |
|||
values, err := redisGet().MGet(ctx, keys...).Result() |
|||
dtmimp.E2P(err) |
|||
for _, v := range values { |
|||
global := TransGlobalStore{} |
|||
dtmimp.MustUnmarshalString(v.(string), &global) |
|||
globals = append(globals, global) |
|||
} |
|||
} |
|||
if cursor > 0 { |
|||
*position = fmt.Sprintf("%d", cursor) |
|||
} else { |
|||
*position = "" |
|||
} |
|||
return globals |
|||
} |
|||
|
|||
func (s *RedisStore) FindBranches(gid string) []TransBranchStore { |
|||
sa, err := redisGet().LRange(ctx, config.Store.RedisPrefix+"_b_"+gid, 0, -1).Result() |
|||
dtmimp.E2P(err) |
|||
branches := make([]TransBranchStore, len(sa)) |
|||
for k, v := range sa { |
|||
dtmimp.MustUnmarshalString(v, &branches[k]) |
|||
} |
|||
return branches |
|||
} |
|||
|
|||
func (s *RedisStore) UpdateBranchesSql(branches []TransBranchStore, updates []string) *gorm.DB { |
|||
return nil // not implemented
|
|||
} |
|||
|
|||
type argList struct { |
|||
Keys []string |
|||
List []interface{} |
|||
} |
|||
|
|||
func newArgList() *argList { |
|||
a := &argList{} |
|||
return a.AppendRaw(config.Store.RedisPrefix).AppendObject(config.Store.DataExpire) |
|||
} |
|||
|
|||
func (a *argList) AppendGid(gid string) *argList { |
|||
a.Keys = append(a.Keys, config.Store.RedisPrefix+"_g_"+gid) |
|||
a.Keys = append(a.Keys, config.Store.RedisPrefix+"_b_"+gid) |
|||
a.Keys = append(a.Keys, config.Store.RedisPrefix+"_u") |
|||
return a |
|||
} |
|||
|
|||
func (a *argList) AppendRaw(v interface{}) *argList { |
|||
a.List = append(a.List, v) |
|||
return a |
|||
} |
|||
|
|||
func (a *argList) AppendObject(v interface{}) *argList { |
|||
return a.AppendRaw(dtmimp.MustMarshalString(v)) |
|||
} |
|||
|
|||
func (a *argList) AppendBranches(branches []TransBranchStore) *argList { |
|||
for _, b := range branches { |
|||
a.AppendRaw(dtmimp.MustMarshalString(b)) |
|||
} |
|||
return a |
|||
} |
|||
|
|||
func handleRedisResult(ret interface{}, err error) (string, error) { |
|||
dtmimp.Logf("result is: '%v', err: '%v'", ret, err) |
|||
if err != nil && err != redis.Nil { |
|||
return "", err |
|||
} |
|||
s, _ := ret.(string) |
|||
err = map[string]error{ |
|||
"NOT_FOUND": ErrNotFound, |
|||
"UNIQUE_CONFLICT": ErrUniqueConflict, |
|||
}[s] |
|||
return s, err |
|||
} |
|||
|
|||
func callLua(a *argList, lua string) (string, error) { |
|||
dtmimp.Logf("calling lua. args: %v\nlua:%s", a, lua) |
|||
ret, err := redisGet().Eval(ctx, lua, a.Keys, a.List...).Result() |
|||
return handleRedisResult(ret, err) |
|||
} |
|||
|
|||
func (s *RedisStore) MaySaveNewTrans(global *TransGlobalStore, branches []TransBranchStore) error { |
|||
a := newArgList(). |
|||
AppendGid(global.Gid). |
|||
AppendObject(global). |
|||
AppendRaw(global.NextCronTime.Unix()). |
|||
AppendBranches(branches) |
|||
global.Steps = nil |
|||
global.Payloads = nil |
|||
_, err := callLua(a, `-- MaySaveNewTrans |
|||
local gs = cjson.decode(ARGV[3]) |
|||
local g = redis.call('GET', KEYS[1]) |
|||
if g ~= false then |
|||
return 'UNIQUE_CONFLICT' |
|||
end |
|||
|
|||
redis.call('SET', KEYS[1], ARGV[3], 'EX', ARGV[2]) |
|||
redis.call('ZADD', KEYS[3], ARGV[4], gs.gid) |
|||
for k = 5, table.getn(ARGV) do |
|||
redis.call('RPUSH', KEYS[2], ARGV[k]) |
|||
end |
|||
redis.call('EXPIRE', KEYS[2], ARGV[2]) |
|||
`) |
|||
return err |
|||
} |
|||
|
|||
func (s *RedisStore) LockGlobalSaveBranches(gid string, status string, branches []TransBranchStore, branchStart int) { |
|||
args := newArgList(). |
|||
AppendGid(gid). |
|||
AppendObject(&TransGlobalStore{Gid: gid, Status: status}). |
|||
AppendRaw(branchStart). |
|||
AppendBranches(branches) |
|||
_, err := callLua(args, ` |
|||
local gs = cjson.decode(ARGV[3]) |
|||
local g = redis.call('GET', KEYS[1]) |
|||
if (g == false) then |
|||
return 'NOT_FOUND' |
|||
end |
|||
local js = cjson.decode(g) |
|||
if js.status ~= gs.status then |
|||
return 'NOT_FOUND' |
|||
end |
|||
local start = ARGV[4] |
|||
for k = 5, table.getn(ARGV) do |
|||
if start == "-1" then |
|||
redis.call('RPUSH', KEYS[2], ARGV[k]) |
|||
else |
|||
redis.call('LSET', KEYS[2], start+k-5, ARGV[k]) |
|||
end |
|||
end |
|||
redis.call('EXPIRE', KEYS[2], ARGV[2]) |
|||
`) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
func (s *RedisStore) ChangeGlobalStatus(global *TransGlobalStore, newStatus string, updates []string, finished bool) { |
|||
old := global.Status |
|||
global.Status = newStatus |
|||
args := newArgList().AppendGid(global.Gid).AppendObject(global).AppendRaw(old).AppendRaw(finished) |
|||
_, err := callLua(args, `-- ChangeGlobalStatus |
|||
local gs = cjson.decode(ARGV[3]) |
|||
local old = redis.call('GET', KEYS[1]) |
|||
if old == false then |
|||
return 'NOT_FOUND' |
|||
end |
|||
local os = cjson.decode(old) |
|||
if os.status ~= ARGV[4] then |
|||
return 'NOT_FOUND' |
|||
end |
|||
redis.call('SET', KEYS[1], ARGV[3], 'EX', ARGV[2]) |
|||
redis.log(redis.LOG_WARNING, 'finished: ', ARGV[5]) |
|||
if ARGV[5] == '1' then |
|||
redis.call('ZREM', KEYS[3], gs.gid) |
|||
end |
|||
`) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
func (s *RedisStore) LockOneGlobalTrans(expireIn time.Duration) *TransGlobalStore { |
|||
expired := time.Now().Add(expireIn).Unix() |
|||
next := time.Now().Add(time.Duration(config.RetryInterval) * time.Second).Unix() |
|||
args := newArgList().AppendGid("").AppendRaw(expired).AppendRaw(next) |
|||
lua := `-- LocakOneGlobalTrans |
|||
local r = redis.call('ZRANGE', KEYS[3], 0, 0, 'WITHSCORES') |
|||
local gid = r[1] |
|||
if gid == nil then |
|||
return 'NOT_FOUND' |
|||
end |
|||
|
|||
if tonumber(r[2]) > tonumber(ARGV[3]) then |
|||
return 'NOT_FOUND' |
|||
end |
|||
redis.call('ZADD', KEYS[3], ARGV[4], gid) |
|||
return gid |
|||
` |
|||
for { |
|||
r, err := callLua(args, lua) |
|||
if err == ErrNotFound { |
|||
return nil |
|||
} |
|||
dtmimp.E2P(err) |
|||
global := s.FindTransGlobalStore(r) |
|||
if global != nil { |
|||
return global |
|||
} |
|||
} |
|||
} |
|||
|
|||
func (s *RedisStore) TouchCronTime(global *TransGlobalStore, nextCronInterval int64) { |
|||
global.NextCronTime = common.GetNextTime(nextCronInterval) |
|||
global.UpdateTime = common.GetNextTime(0) |
|||
global.NextCronInterval = nextCronInterval |
|||
args := newArgList().AppendGid(global.Gid).AppendObject(global).AppendRaw(global.NextCronTime.Unix()) |
|||
_, err := callLua(args, `-- TouchCronTime |
|||
local g = cjson.decode(ARGV[3]) |
|||
local old = redis.call('GET', KEYS[1]) |
|||
if old == false then |
|||
return 'NOT_FOUND' |
|||
end |
|||
local os = cjson.decode(old) |
|||
if os.status ~= g.status then |
|||
return 'NOT_FOUND' |
|||
end |
|||
redis.call('ZADD', KEYS[3], ARGV[4], g.gid) |
|||
redis.call('SET', KEYS[1], ARGV[3], 'EX', ARGV[2]) |
|||
`) |
|||
dtmimp.E2P(err) |
|||
} |
|||
@ -0,0 +1,300 @@ |
|||
package redis |
|||
|
|||
import ( |
|||
"context" |
|||
"errors" |
|||
"fmt" |
|||
"sync" |
|||
"time" |
|||
|
|||
"github.com/go-redis/redis/v8" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
"github.com/dtm-labs/dtm/dtmsvr/config" |
|||
"github.com/dtm-labs/dtm/dtmsvr/storage" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
) |
|||
|
|||
// TODO: optimize this, it's very strange to use pointer to dtmutil.Config
|
|||
var conf = &config.Config |
|||
|
|||
// TODO: optimize this, all function should have context as first parameter
|
|||
var ctx = context.Background() |
|||
|
|||
// Store is the storage with redis, all transaction information will bachend with redis
|
|||
type Store struct { |
|||
} |
|||
|
|||
// Ping execs ping cmd to redis
|
|||
func (s *Store) Ping() error { |
|||
_, err := redisGet().Ping(ctx).Result() |
|||
return err |
|||
} |
|||
|
|||
// PopulateData populates data to redis
|
|||
func (s *Store) PopulateData(skipDrop bool) { |
|||
if !skipDrop { |
|||
_, err := redisGet().FlushAll(ctx).Result() |
|||
logger.Infof("call redis flushall. result: %v", err) |
|||
dtmimp.PanicIf(err != nil, err) |
|||
} |
|||
} |
|||
|
|||
// FindTransGlobalStore finds GlobalTrans data by gid
|
|||
func (s *Store) FindTransGlobalStore(gid string) *storage.TransGlobalStore { |
|||
logger.Debugf("calling FindTransGlobalStore: %s", gid) |
|||
r, err := redisGet().Get(ctx, conf.Store.RedisPrefix+"_g_"+gid).Result() |
|||
if err == redis.Nil { |
|||
return nil |
|||
} |
|||
dtmimp.E2P(err) |
|||
trans := &storage.TransGlobalStore{} |
|||
dtmimp.MustUnmarshalString(r, trans) |
|||
return trans |
|||
} |
|||
|
|||
// ScanTransGlobalStores lists GlobalTrans data
|
|||
func (s *Store) ScanTransGlobalStores(position *string, limit int64) []storage.TransGlobalStore { |
|||
logger.Debugf("calling ScanTransGlobalStores: %s %d", *position, limit) |
|||
lid := uint64(0) |
|||
if *position != "" { |
|||
lid = uint64(dtmimp.MustAtoi(*position)) |
|||
} |
|||
keys, cursor, err := redisGet().Scan(ctx, lid, conf.Store.RedisPrefix+"_g_*", limit).Result() |
|||
dtmimp.E2P(err) |
|||
globals := []storage.TransGlobalStore{} |
|||
if len(keys) > 0 { |
|||
values, err := redisGet().MGet(ctx, keys...).Result() |
|||
dtmimp.E2P(err) |
|||
for _, v := range values { |
|||
global := storage.TransGlobalStore{} |
|||
dtmimp.MustUnmarshalString(v.(string), &global) |
|||
globals = append(globals, global) |
|||
} |
|||
} |
|||
if cursor > 0 { |
|||
*position = fmt.Sprintf("%d", cursor) |
|||
} else { |
|||
*position = "" |
|||
} |
|||
return globals |
|||
} |
|||
|
|||
// FindBranches finds Branch data by gid
|
|||
func (s *Store) FindBranches(gid string) []storage.TransBranchStore { |
|||
logger.Debugf("calling FindBranches: %s", gid) |
|||
sa, err := redisGet().LRange(ctx, conf.Store.RedisPrefix+"_b_"+gid, 0, -1).Result() |
|||
dtmimp.E2P(err) |
|||
branches := make([]storage.TransBranchStore, len(sa)) |
|||
for k, v := range sa { |
|||
dtmimp.MustUnmarshalString(v, &branches[k]) |
|||
} |
|||
return branches |
|||
} |
|||
|
|||
// UpdateBranches updates branches info
|
|||
func (s *Store) UpdateBranches(branches []storage.TransBranchStore, updates []string) (int, error) { |
|||
return 0, nil // not implemented
|
|||
} |
|||
|
|||
type argList struct { |
|||
Keys []string |
|||
List []interface{} |
|||
} |
|||
|
|||
func newArgList() *argList { |
|||
a := &argList{} |
|||
return a.AppendRaw(conf.Store.RedisPrefix).AppendObject(conf.Store.DataExpire) |
|||
} |
|||
|
|||
func (a *argList) AppendGid(gid string) *argList { |
|||
a.Keys = append(a.Keys, conf.Store.RedisPrefix+"_g_"+gid) |
|||
a.Keys = append(a.Keys, conf.Store.RedisPrefix+"_b_"+gid) |
|||
a.Keys = append(a.Keys, conf.Store.RedisPrefix+"_u") |
|||
a.Keys = append(a.Keys, conf.Store.RedisPrefix+"_s_"+gid) |
|||
return a |
|||
} |
|||
|
|||
func (a *argList) AppendRaw(v interface{}) *argList { |
|||
a.List = append(a.List, v) |
|||
return a |
|||
} |
|||
|
|||
func (a *argList) AppendObject(v interface{}) *argList { |
|||
return a.AppendRaw(dtmimp.MustMarshalString(v)) |
|||
} |
|||
|
|||
func (a *argList) AppendBranches(branches []storage.TransBranchStore) *argList { |
|||
for _, b := range branches { |
|||
a.AppendRaw(dtmimp.MustMarshalString(b)) |
|||
} |
|||
return a |
|||
} |
|||
|
|||
func handleRedisResult(ret interface{}, err error) (string, error) { |
|||
logger.Debugf("result is: '%v', err: '%v'", ret, err) |
|||
if err != nil && err != redis.Nil { |
|||
return "", err |
|||
} |
|||
s, _ := ret.(string) |
|||
err = map[string]error{ |
|||
"NOT_FOUND": storage.ErrNotFound, |
|||
"UNIQUE_CONFLICT": storage.ErrUniqueConflict, |
|||
}[s] |
|||
return s, err |
|||
} |
|||
|
|||
func callLua(a *argList, lua string) (string, error) { |
|||
logger.Debugf("calling lua. args: %v\nlua:%s", a, lua) |
|||
ret, err := redisGet().Eval(ctx, lua, a.Keys, a.List...).Result() |
|||
return handleRedisResult(ret, err) |
|||
} |
|||
|
|||
// MaySaveNewTrans creates a new trans
|
|||
func (s *Store) MaySaveNewTrans(global *storage.TransGlobalStore, branches []storage.TransBranchStore) error { |
|||
a := newArgList(). |
|||
AppendGid(global.Gid). |
|||
AppendObject(global). |
|||
AppendRaw(global.NextCronTime.Unix()). |
|||
AppendRaw(global.Gid). |
|||
AppendRaw(global.Status). |
|||
AppendBranches(branches) |
|||
global.Steps = nil |
|||
global.Payloads = nil |
|||
_, err := callLua(a, `-- MaySaveNewTrans |
|||
local g = redis.call('GET', KEYS[1]) |
|||
if g ~= false then |
|||
return 'UNIQUE_CONFLICT' |
|||
end |
|||
|
|||
redis.call('SET', KEYS[1], ARGV[3], 'EX', ARGV[2]) |
|||
redis.call('SET', KEYS[4], ARGV[6], 'EX', ARGV[2]) |
|||
redis.call('ZADD', KEYS[3], ARGV[4], ARGV[5]) |
|||
for k = 7, table.getn(ARGV) do |
|||
redis.call('RPUSH', KEYS[2], ARGV[k]) |
|||
end |
|||
redis.call('EXPIRE', KEYS[2], ARGV[2]) |
|||
`) |
|||
return err |
|||
} |
|||
|
|||
// LockGlobalSaveBranches creates branches
|
|||
func (s *Store) LockGlobalSaveBranches(gid string, status string, branches []storage.TransBranchStore, branchStart int) { |
|||
args := newArgList(). |
|||
AppendGid(gid). |
|||
AppendRaw(status). |
|||
AppendRaw(branchStart). |
|||
AppendBranches(branches) |
|||
_, err := callLua(args, `-- LockGlobalSaveBranches |
|||
local old = redis.call('GET', KEYS[4]) |
|||
if old ~= ARGV[3] then |
|||
return 'NOT_FOUND' |
|||
end |
|||
local start = ARGV[4] |
|||
for k = 5, table.getn(ARGV) do |
|||
if start == "-1" then |
|||
redis.call('RPUSH', KEYS[2], ARGV[k]) |
|||
else |
|||
redis.call('LSET', KEYS[2], start+k-5, ARGV[k]) |
|||
end |
|||
end |
|||
redis.call('EXPIRE', KEYS[2], ARGV[2]) |
|||
`) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
// ChangeGlobalStatus changes global trans status
|
|||
func (s *Store) ChangeGlobalStatus(global *storage.TransGlobalStore, newStatus string, updates []string, finished bool) { |
|||
old := global.Status |
|||
global.Status = newStatus |
|||
args := newArgList(). |
|||
AppendGid(global.Gid). |
|||
AppendObject(global). |
|||
AppendRaw(old). |
|||
AppendRaw(finished). |
|||
AppendRaw(global.Gid). |
|||
AppendRaw(newStatus) |
|||
_, err := callLua(args, `-- ChangeGlobalStatus |
|||
local old = redis.call('GET', KEYS[4]) |
|||
if old ~= ARGV[4] then |
|||
return 'NOT_FOUND' |
|||
end |
|||
redis.call('SET', KEYS[1], ARGV[3], 'EX', ARGV[2]) |
|||
redis.call('SET', KEYS[4], ARGV[7], 'EX', ARGV[2]) |
|||
if ARGV[5] == '1' then |
|||
redis.call('ZREM', KEYS[3], ARGV[6]) |
|||
end |
|||
`) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
// LockOneGlobalTrans finds GlobalTrans
|
|||
func (s *Store) LockOneGlobalTrans(expireIn time.Duration) *storage.TransGlobalStore { |
|||
expired := time.Now().Add(expireIn).Unix() |
|||
next := time.Now().Add(time.Duration(conf.RetryInterval) * time.Second).Unix() |
|||
args := newArgList().AppendGid("").AppendRaw(expired).AppendRaw(next) |
|||
lua := `-- LockOneGlobalTrans |
|||
local r = redis.call('ZRANGE', KEYS[3], 0, 0, 'WITHSCORES') |
|||
local gid = r[1] |
|||
if gid == nil then |
|||
return 'NOT_FOUND' |
|||
end |
|||
|
|||
if tonumber(r[2]) > tonumber(ARGV[3]) then |
|||
return 'NOT_FOUND' |
|||
end |
|||
redis.call('ZADD', KEYS[3], ARGV[4], gid) |
|||
return gid |
|||
` |
|||
for { |
|||
r, err := callLua(args, lua) |
|||
if errors.Is(err, storage.ErrNotFound) { |
|||
return nil |
|||
} |
|||
dtmimp.E2P(err) |
|||
global := s.FindTransGlobalStore(r) |
|||
if global != nil { |
|||
return global |
|||
} |
|||
} |
|||
} |
|||
|
|||
// TouchCronTime updates cronTime
|
|||
func (s *Store) TouchCronTime(global *storage.TransGlobalStore, nextCronInterval int64) { |
|||
global.NextCronTime = dtmutil.GetNextTime(nextCronInterval) |
|||
global.UpdateTime = dtmutil.GetNextTime(0) |
|||
global.NextCronInterval = nextCronInterval |
|||
args := newArgList(). |
|||
AppendGid(global.Gid). |
|||
AppendObject(global). |
|||
AppendRaw(global.NextCronTime.Unix()). |
|||
AppendRaw(global.Status). |
|||
AppendRaw(global.Gid) |
|||
_, err := callLua(args, `-- TouchCronTime |
|||
local old = redis.call('GET', KEYS[4]) |
|||
if old ~= ARGV[5] then |
|||
return 'NOT_FOUND' |
|||
end |
|||
redis.call('ZADD', KEYS[3], ARGV[4], ARGV[6]) |
|||
redis.call('SET', KEYS[1], ARGV[3], 'EX', ARGV[2]) |
|||
`) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
var ( |
|||
rdb *redis.Client |
|||
once sync.Once |
|||
) |
|||
|
|||
func redisGet() *redis.Client { |
|||
once.Do(func() { |
|||
logger.Debugf("connecting to redis: %v", conf.Store) |
|||
rdb = redis.NewClient(&redis.Options{ |
|||
Addr: fmt.Sprintf("%s:%d", conf.Store.Host, conf.Store.Port), |
|||
Username: conf.Store.User, |
|||
Password: conf.Store.Password, |
|||
}) |
|||
}) |
|||
return rdb |
|||
} |
|||
@ -0,0 +1,25 @@ |
|||
package registry |
|||
|
|||
import ( |
|||
"sync" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmsvr/storage" |
|||
) |
|||
|
|||
// SingletonFactory is the factory to build store in SINGLETON pattern.
|
|||
type SingletonFactory struct { |
|||
once sync.Once |
|||
|
|||
store storage.Store |
|||
|
|||
creatorFunction func() storage.Store |
|||
} |
|||
|
|||
// GetStorage implement the StorageFactory.GetStorage
|
|||
func (f *SingletonFactory) GetStorage() storage.Store { |
|||
f.once.Do(func() { |
|||
f.store = f.creatorFunction() |
|||
}) |
|||
|
|||
return f.store |
|||
} |
|||
@ -1,140 +0,0 @@ |
|||
package storage |
|||
|
|||
import ( |
|||
"fmt" |
|||
"math" |
|||
"time" |
|||
|
|||
"github.com/google/uuid" |
|||
"github.com/yedf/dtm/common" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
"gorm.io/gorm" |
|||
"gorm.io/gorm/clause" |
|||
) |
|||
|
|||
type SqlStore struct { |
|||
} |
|||
|
|||
func (s *SqlStore) Ping() error { |
|||
db, err := dtmimp.StandaloneDB(config.Store.GetDBConf()) |
|||
dtmimp.E2P(err) |
|||
_, err = db.Exec("select 1") |
|||
return err |
|||
} |
|||
|
|||
func (s *SqlStore) PopulateData(skipDrop bool) { |
|||
file := fmt.Sprintf("%s/dtmsvr.storage.%s.sql", common.GetSqlDir(), config.Store.Driver) |
|||
common.RunSQLScript(config.Store.GetDBConf(), file, skipDrop) |
|||
} |
|||
|
|||
func (s *SqlStore) FindTransGlobalStore(gid string) *TransGlobalStore { |
|||
trans := &TransGlobalStore{} |
|||
dbr := dbGet().Model(trans).Where("gid=?", gid).First(trans) |
|||
if dbr.Error == gorm.ErrRecordNotFound { |
|||
return nil |
|||
} |
|||
dtmimp.E2P(dbr.Error) |
|||
return trans |
|||
} |
|||
|
|||
func (s *SqlStore) ScanTransGlobalStores(position *string, limit int64) []TransGlobalStore { |
|||
globals := []TransGlobalStore{} |
|||
lid := math.MaxInt64 |
|||
if *position != "" { |
|||
lid = dtmimp.MustAtoi(*position) |
|||
} |
|||
dbr := dbGet().Must().Where("id < ?", lid).Order("id desc").Limit(int(limit)).Find(&globals) |
|||
if dbr.RowsAffected < limit { |
|||
*position = "" |
|||
} else { |
|||
*position = fmt.Sprintf("%d", globals[len(globals)-1].ID) |
|||
} |
|||
return globals |
|||
} |
|||
|
|||
func (s *SqlStore) FindBranches(gid string) []TransBranchStore { |
|||
branches := []TransBranchStore{} |
|||
dbGet().Must().Where("gid=?", gid).Order("id asc").Find(&branches) |
|||
return branches |
|||
} |
|||
|
|||
func (s *SqlStore) UpdateBranchesSql(branches []TransBranchStore, updates []string) *gorm.DB { |
|||
return dbGet().Clauses(clause.OnConflict{ |
|||
OnConstraint: "trans_branch_op_pkey", |
|||
DoUpdates: clause.AssignmentColumns(updates), |
|||
}).Create(branches) |
|||
} |
|||
|
|||
func (s *SqlStore) LockGlobalSaveBranches(gid string, status string, branches []TransBranchStore, branchStart int) { |
|||
err := dbGet().Transaction(func(tx *gorm.DB) error { |
|||
g := &TransGlobalStore{} |
|||
dbr := tx.Clauses(clause.Locking{Strength: "UPDATE"}).Model(g).Where("gid=? and status=?", gid, status).First(g) |
|||
if dbr.Error == nil { |
|||
dbr = tx.Save(branches) |
|||
} |
|||
return wrapError(dbr.Error) |
|||
}) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
func (s *SqlStore) MaySaveNewTrans(global *TransGlobalStore, branches []TransBranchStore) error { |
|||
return dbGet().Transaction(func(db1 *gorm.DB) error { |
|||
db := &common.DB{DB: db1} |
|||
dbr := db.Must().Clauses(clause.OnConflict{ |
|||
DoNothing: true, |
|||
}).Create(global) |
|||
if dbr.RowsAffected <= 0 { // 如果这个不是新事务，返回错误
|
|||
return ErrUniqueConflict |
|||
} |
|||
if len(branches) > 0 { |
|||
db.Must().Clauses(clause.OnConflict{ |
|||
DoNothing: true, |
|||
}).Create(&branches) |
|||
} |
|||
return nil |
|||
}) |
|||
} |
|||
|
|||
func (s *SqlStore) ChangeGlobalStatus(global *TransGlobalStore, newStatus string, updates []string, finished bool) { |
|||
old := global.Status |
|||
global.Status = newStatus |
|||
dbr := dbGet().Must().Model(global).Where("status=? and gid=?", old, global.Gid).Select(updates).Updates(global) |
|||
if dbr.RowsAffected == 0 { |
|||
dtmimp.E2P(ErrNotFound) |
|||
} |
|||
} |
|||
|
|||
func (s *SqlStore) TouchCronTime(global *TransGlobalStore, nextCronInterval int64) { |
|||
global.NextCronTime = common.GetNextTime(nextCronInterval) |
|||
global.UpdateTime = common.GetNextTime(0) |
|||
global.NextCronInterval = nextCronInterval |
|||
dbGet().Must().Model(global).Where("status=? and gid=?", global.Status, global.Gid). |
|||
Select([]string{"next_cron_time", "update_time", "next_cron_interval"}).Updates(global) |
|||
} |
|||
|
|||
func (s *SqlStore) LockOneGlobalTrans(expireIn time.Duration) *TransGlobalStore { |
|||
db := dbGet() |
|||
getTime := func(second int) string { |
|||
return map[string]string{ |
|||
"mysql": fmt.Sprintf("date_add(now(), interval %d second)", second), |
|||
"postgres": fmt.Sprintf("current_timestamp + interval '%d second'", second), |
|||
}[config.Store.Driver] |
|||
} |
|||
expire := int(expireIn / time.Second) |
|||
whereTime := fmt.Sprintf("next_cron_time < %s", getTime(expire)) |
|||
owner := uuid.NewString() |
|||
global := &TransGlobalStore{} |
|||
dbr := db.Must().Model(global). |
|||
Where(whereTime + "and status in ('prepared', 'aborting', 'submitted')"). |
|||
Limit(1). |
|||
Select([]string{"owner", "next_cron_time"}). |
|||
Updates(&TransGlobalStore{ |
|||
Owner: owner, |
|||
NextCronTime: common.GetNextTime(common.Config.RetryInterval), |
|||
}) |
|||
if dbr.RowsAffected == 0 { |
|||
return nil |
|||
} |
|||
dbr = db.Must().Where("owner=?", owner).First(global) |
|||
return global |
|||
} |
|||
@ -0,0 +1,178 @@ |
|||
package sql |
|||
|
|||
import ( |
|||
"fmt" |
|||
"math" |
|||
"time" |
|||
|
|||
"github.com/lithammer/shortuuid/v3" |
|||
"gorm.io/gorm" |
|||
"gorm.io/gorm/clause" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmsvr/config" |
|||
"github.com/dtm-labs/dtm/dtmsvr/storage" |
|||
"github.com/dtm-labs/dtm/dtmutil" |
|||
) |
|||
|
|||
var conf = &config.Config |
|||
|
|||
// Store implements storage.Store, and storage with db
|
|||
type Store struct { |
|||
} |
|||
|
|||
// Ping execs ping cmd to db
|
|||
func (s *Store) Ping() error { |
|||
db, err := dtmimp.StandaloneDB(conf.Store.GetDBConf()) |
|||
dtmimp.E2P(err) |
|||
_, err = db.Exec("select 1") |
|||
return err |
|||
} |
|||
|
|||
// PopulateData populates data to db
|
|||
func (s *Store) PopulateData(skipDrop bool) { |
|||
file := fmt.Sprintf("%s/dtmsvr.storage.%s.sql", dtmutil.GetSQLDir(), conf.Store.Driver) |
|||
dtmutil.RunSQLScript(conf.Store.GetDBConf(), file, skipDrop) |
|||
} |
|||
|
|||
// FindTransGlobalStore finds GlobalTrans data by gid
|
|||
func (s *Store) FindTransGlobalStore(gid string) *storage.TransGlobalStore { |
|||
trans := &storage.TransGlobalStore{} |
|||
dbr := dbGet().Model(trans).Where("gid=?", gid).First(trans) |
|||
if dbr.Error == gorm.ErrRecordNotFound { |
|||
return nil |
|||
} |
|||
dtmimp.E2P(dbr.Error) |
|||
return trans |
|||
} |
|||
|
|||
// ScanTransGlobalStores lists GlobalTrans data
|
|||
func (s *Store) ScanTransGlobalStores(position *string, limit int64) []storage.TransGlobalStore { |
|||
globals := []storage.TransGlobalStore{} |
|||
lid := math.MaxInt64 |
|||
if *position != "" { |
|||
lid = dtmimp.MustAtoi(*position) |
|||
} |
|||
dbr := dbGet().Must().Where("id < ?", lid).Order("id desc").Limit(int(limit)).Find(&globals) |
|||
if dbr.RowsAffected < limit { |
|||
*position = "" |
|||
} else { |
|||
*position = fmt.Sprintf("%d", globals[len(globals)-1].ID) |
|||
} |
|||
return globals |
|||
} |
|||
|
|||
// FindBranches finds Branch data by gid
|
|||
func (s *Store) FindBranches(gid string) []storage.TransBranchStore { |
|||
branches := []storage.TransBranchStore{} |
|||
dbGet().Must().Where("gid=?", gid).Order("id asc").Find(&branches) |
|||
return branches |
|||
} |
|||
|
|||
// UpdateBranches update branches info
|
|||
func (s *Store) UpdateBranches(branches []storage.TransBranchStore, updates []string) (int, error) { |
|||
db := dbGet().Clauses(clause.OnConflict{ |
|||
OnConstraint: "trans_branch_op_pkey", |
|||
DoUpdates: clause.AssignmentColumns(updates), |
|||
}).Create(branches) |
|||
return int(db.RowsAffected), db.Error |
|||
} |
|||
|
|||
// LockGlobalSaveBranches creates branches
|
|||
func (s *Store) LockGlobalSaveBranches(gid string, status string, branches []storage.TransBranchStore, branchStart int) { |
|||
err := dbGet().Transaction(func(tx *gorm.DB) error { |
|||
g := &storage.TransGlobalStore{} |
|||
dbr := tx.Clauses(clause.Locking{Strength: "UPDATE"}).Model(g).Where("gid=? and status=?", gid, status).First(g) |
|||
if dbr.Error == nil { |
|||
dbr = tx.Save(branches) |
|||
} |
|||
return wrapError(dbr.Error) |
|||
}) |
|||
dtmimp.E2P(err) |
|||
} |
|||
|
|||
// MaySaveNewTrans creates a new trans
|
|||
func (s *Store) MaySaveNewTrans(global *storage.TransGlobalStore, branches []storage.TransBranchStore) error { |
|||
return dbGet().Transaction(func(db1 *gorm.DB) error { |
|||
db := &dtmutil.DB{DB: db1} |
|||
dbr := db.Must().Clauses(clause.OnConflict{ |
|||
DoNothing: true, |
|||
}).Create(global) |
|||
if dbr.RowsAffected <= 0 { // 如果这个不是新事务，返回错误
|
|||
return storage.ErrUniqueConflict |
|||
} |
|||
if len(branches) > 0 { |
|||
db.Must().Clauses(clause.OnConflict{ |
|||
DoNothing: true, |
|||
}).Create(&branches) |
|||
} |
|||
return nil |
|||
}) |
|||
} |
|||
|
|||
// ChangeGlobalStatus changes global trans status
|
|||
func (s *Store) ChangeGlobalStatus(global *storage.TransGlobalStore, newStatus string, updates []string, finished bool) { |
|||
old := global.Status |
|||
global.Status = newStatus |
|||
dbr := dbGet().Must().Model(global).Where("status=? and gid=?", old, global.Gid).Select(updates).Updates(global) |
|||
if dbr.RowsAffected == 0 { |
|||
dtmimp.E2P(storage.ErrNotFound) |
|||
} |
|||
} |
|||
|
|||
// TouchCronTime updates cronTime
|
|||
func (s *Store) TouchCronTime(global *storage.TransGlobalStore, nextCronInterval int64) { |
|||
global.NextCronTime = dtmutil.GetNextTime(nextCronInterval) |
|||
global.UpdateTime = dtmutil.GetNextTime(0) |
|||
global.NextCronInterval = nextCronInterval |
|||
dbGet().Must().Model(global).Where("status=? and gid=?", global.Status, global.Gid). |
|||
Select([]string{"next_cron_time", "update_time", "next_cron_interval"}).Updates(global) |
|||
} |
|||
|
|||
// LockOneGlobalTrans finds GlobalTrans
|
|||
func (s *Store) LockOneGlobalTrans(expireIn time.Duration) *storage.TransGlobalStore { |
|||
db := dbGet() |
|||
getTime := func(second int) string { |
|||
return map[string]string{ |
|||
"mysql": fmt.Sprintf("date_add(now(), interval %d second)", second), |
|||
"postgres": fmt.Sprintf("current_timestamp + interval '%d second'", second), |
|||
}[conf.Store.Driver] |
|||
} |
|||
expire := int(expireIn / time.Second) |
|||
whereTime := fmt.Sprintf("next_cron_time < %s", getTime(expire)) |
|||
owner := shortuuid.New() |
|||
global := &storage.TransGlobalStore{} |
|||
dbr := db.Must().Model(global). |
|||
Where(whereTime + "and status in ('prepared', 'aborting', 'submitted')"). |
|||
Limit(1). |
|||
Select([]string{"owner", "next_cron_time"}). |
|||
Updates(&storage.TransGlobalStore{ |
|||
Owner: owner, |
|||
NextCronTime: dtmutil.GetNextTime(conf.RetryInterval), |
|||
}) |
|||
if dbr.RowsAffected == 0 { |
|||
return nil |
|||
} |
|||
db.Must().Where("owner=?", owner).First(global) |
|||
return global |
|||
} |
|||
|
|||
// SetDBConn sets db conn pool
|
|||
func SetDBConn(db *gorm.DB) { |
|||
sqldb, _ := db.DB() |
|||
sqldb.SetMaxOpenConns(int(conf.Store.MaxOpenConns)) |
|||
sqldb.SetMaxIdleConns(int(conf.Store.MaxIdleConns)) |
|||
sqldb.SetConnMaxLifetime(time.Duration(conf.Store.ConnMaxLifeTime) * time.Minute) |
|||
} |
|||
|
|||
func dbGet() *dtmutil.DB { |
|||
return dtmutil.DbGet(conf.Store.GetDBConf(), SetDBConn) |
|||
} |
|||
|
|||
func wrapError(err error) error { |
|||
if err == gorm.ErrRecordNotFound { |
|||
return storage.ErrNotFound |
|||
} |
|||
dtmimp.E2P(err) |
|||
return err |
|||
} |
|||
@ -1,16 +0,0 @@ |
|||
package storage |
|||
|
|||
import ( |
|||
"github.com/go-redis/redis/v8" |
|||
"github.com/yedf/dtm/common" |
|||
) |
|||
|
|||
var config = &common.Config |
|||
|
|||
func dbGet() *common.DB { |
|||
return common.DbGet(config.Store.GetDBConf()) |
|||
} |
|||
|
|||
func redisGet() *redis.Client { |
|||
return common.RedisGet() |
|||
} |
|||
@ -1,7 +0,0 @@ |
|||
package dtmsvr |
|||
|
|||
import ( |
|||
_ "github.com/ychensha/dtmdriver-polaris" |
|||
_ "github.com/yedf/dtmdriver-gozero" |
|||
_ "github.com/yedf/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" |
|||
) |
|||
@ -0,0 +1,157 @@ |
|||
/* |
|||
* 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 |
|||
|
|||
import ( |
|||
"bytes" |
|||
"encoding/json" |
|||
"errors" |
|||
"io/ioutil" |
|||
"net/http" |
|||
"os" |
|||
"path/filepath" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/go-resty/resty/v2" |
|||
|
|||
"github.com/dtm-labs/dtm/dtmcli" |
|||
"github.com/dtm-labs/dtm/dtmcli/dtmimp" |
|||
"github.com/dtm-labs/dtm/dtmcli/logger" |
|||
) |
|||
|
|||
// GetGinApp init and return gin
|
|||
func GetGinApp() *gin.Engine { |
|||
gin.SetMode(gin.ReleaseMode) |
|||
app := gin.New() |
|||
app.Use(gin.Recovery()) |
|||
app.Use(func(c *gin.Context) { |
|||
body := "" |
|||
if c.Request.Body != nil { |
|||
rb, err := c.GetRawData() |
|||
dtmimp.E2P(err) |
|||
if len(rb) > 0 { |
|||
body = string(rb) |
|||
c.Request.Body = ioutil.NopCloser(bytes.NewBuffer(rb)) |
|||
} |
|||
} |
|||
logger.Debugf("begin %s %s body: %s", c.Request.Method, c.Request.URL, body) |
|||
c.Next() |
|||
}) |
|||
app.Any("/api/ping", func(c *gin.Context) { c.JSON(200, map[string]interface{}{"msg": "pong"}) }) |
|||
return app |
|||
} |
|||
|
|||
// WrapHandler2 wrap a function te bo the handler of gin request
|
|||
func WrapHandler2(fn func(*gin.Context) interface{}) gin.HandlerFunc { |
|||
return func(c *gin.Context) { |
|||
began := time.Now() |
|||
var err error |
|||
r := func() interface{} { |
|||
defer dtmimp.P2E(&err) |
|||
return fn(c) |
|||
}() |
|||
|
|||
status := http.StatusOK |
|||
|
|||
// in dtm test/busi, there are some functions, which will return a resty response
|
|||
// pass resty response as gin's response
|
|||
if resp, ok := r.(*resty.Response); ok { |
|||
b := resp.Body() |
|||
status = resp.StatusCode() |
|||
r = nil |
|||
err = json.Unmarshal(b, &r) |
|||
} |
|||
|
|||
// error maybe returned in r, assign it to err
|
|||
if ne, ok := r.(error); ok && err == nil { |
|||
err = ne |
|||
} |
|||
|
|||
// if err != nil || r == nil. then set the status and dtm_result
|
|||
// dtm_result is for compatible with version lower than v1.10
|
|||
// when >= v1.10, result test should base on status, not dtm_result.
|
|||
result := map[string]interface{}{} |
|||
if err != nil { |
|||
if errors.Is(err, dtmcli.ErrFailure) { |
|||
status = http.StatusConflict |
|||
result["dtm_result"] = dtmcli.ResultFailure |
|||
} else if errors.Is(err, dtmcli.ErrOngoing) { |
|||
status = http.StatusTooEarly |
|||
result["dtm_result"] = dtmcli.ResultOngoing |
|||
} else if err != nil { |
|||
status = http.StatusInternalServerError |
|||
} |
|||
result["message"] = err.Error() |
|||
r = result |
|||
} else if r == nil { |
|||
result["dtm_result"] = dtmcli.ResultSuccess |
|||
r = result |
|||
} |
|||
|
|||
b, _ := json.Marshal(r) |
|||
cont := string(b) |
|||
if status == http.StatusOK || status == http.StatusTooEarly { |
|||
logger.Infof("%2dms %d %s %s %s", time.Since(began).Milliseconds(), status, c.Request.Method, c.Request.RequestURI, cont) |
|||
} else { |
|||
logger.Errorf("%2dms %d %s %s %s", time.Since(began).Milliseconds(), status, c.Request.Method, c.Request.RequestURI, cont) |
|||
} |
|||
c.JSON(status, r) |
|||
} |
|||
} |
|||
|
|||
// MustGetwd must version of os.Getwd
|
|||
func MustGetwd() string { |
|||
wd, err := os.Getwd() |
|||
dtmimp.E2P(err) |
|||
return wd |
|||
} |
|||
|
|||
// GetSQLDir 获取调用该函数的caller源代码的目录，主要用于测试时，查找相关文件
|
|||
func GetSQLDir() string { |
|||
wd := MustGetwd() |
|||
if filepath.Base(wd) == "test" { |
|||
wd = filepath.Dir(wd) |
|||
} |
|||
return wd + "/sqls" |
|||
} |
|||
|
|||
// RecoverPanic execs recovery operation
|
|||
func RecoverPanic(err *error) { |
|||
if x := recover(); x != nil { |
|||
e := dtmimp.AsError(x) |
|||
if err != nil { |
|||
*err = e |
|||
} |
|||
} |
|||
} |
|||
|
|||
// GetNextTime gets next time from second
|
|||
func GetNextTime(second int64) *time.Time { |
|||
next := time.Now().Add(time.Duration(second) * time.Second) |
|||
return &next |
|||
} |
|||
|
|||
// RunSQLScript 1
|
|||
func RunSQLScript(conf dtmcli.DBConf, script string, skipDrop bool) { |
|||
con, err := dtmimp.StandaloneDB(conf) |
|||
logger.FatalIfError(err) |
|||
defer func() { _ = con.Close() }() |
|||
content, err := ioutil.ReadFile(script) |
|||
logger.FatalIfError(err) |
|||
sqls := strings.Split(string(content), ";") |
|||
for _, sql := range sqls { |
|||
s := strings.TrimSpace(sql) |
|||
if s == "" || (skipDrop && strings.Contains(s, "drop")) { |
|||
continue |
|||
} |
|||
_, err = dtmimp.DBExec(con, s) |
|||
logger.FatalIfError(err) |
|||
logger.Infof("sql scripts finished: %s", s) |
|||
} |
|||
} |
|||
@ -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,192 +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" |
|||
"errors" |
|||
"fmt" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/yedf/dtm/common" |
|||
"github.com/yedf/dtm/dtmcli" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
"gorm.io/driver/mysql" |
|||
"gorm.io/driver/postgres" |
|||
"gorm.io/gorm" |
|||
) |
|||
|
|||
const ( |
|||
// BusiAPI busi api prefix
|
|||
BusiAPI = "/api/busi" |
|||
// BusiPort busi server port
|
|||
BusiPort = 8081 |
|||
// BusiGrpcPort busi server port
|
|||
BusiGrpcPort = 58081 |
|||
) |
|||
|
|||
type setupFunc func(*gin.Engine) |
|||
|
|||
var setupFuncs = map[string]setupFunc{} |
|||
|
|||
// Busi busi service url prefix
|
|||
var Busi string = fmt.Sprintf("http://localhost:%d%s", BusiPort, BusiAPI) |
|||
|
|||
// BaseAppStartup base app startup
|
|||
func BaseAppStartup() *gin.Engine { |
|||
dtmimp.Logf("examples starting") |
|||
app := common.GetGinApp() |
|||
app.Use(func(c *gin.Context) { |
|||
v := MainSwitch.NextResult.Fetch() |
|||
if v != "" { |
|||
c.JSON(200, gin.H{"dtm_result": v}) |
|||
c.Abort() |
|||
return |
|||
} |
|||
c.Next() |
|||
}) |
|||
|
|||
BaseAddRoute(app) |
|||
for k, v := range setupFuncs { |
|||
dtmimp.Logf("initing %s", k) |
|||
v(app) |
|||
} |
|||
dtmimp.Logf("Starting busi at: %d", BusiPort) |
|||
go app.Run(fmt.Sprintf(":%d", BusiPort)) |
|||
|
|||
time.Sleep(100 * time.Millisecond) |
|||
return app |
|||
} |
|||
|
|||
// AutoEmptyString auto reset to empty when used once
|
|||
type AutoEmptyString struct { |
|||
value string |
|||
} |
|||
|
|||
// SetOnce set a value once
|
|||
func (s *AutoEmptyString) SetOnce(v string) { |
|||
s.value = v |
|||
} |
|||
|
|||
// Fetch fetch the stored value, then reset the value to empty
|
|||
func (s *AutoEmptyString) Fetch() string { |
|||
v := s.value |
|||
s.value = "" |
|||
return v |
|||
} |
|||
|
|||
type mainSwitchType struct { |
|||
TransInResult AutoEmptyString |
|||
TransOutResult AutoEmptyString |
|||
TransInConfirmResult AutoEmptyString |
|||
TransOutConfirmResult AutoEmptyString |
|||
TransInRevertResult AutoEmptyString |
|||
TransOutRevertResult AutoEmptyString |
|||
CanSubmitResult AutoEmptyString |
|||
NextResult AutoEmptyString |
|||
} |
|||
|
|||
// MainSwitch controls busi success or fail
|
|||
var MainSwitch mainSwitchType |
|||
|
|||
func handleGeneralBusiness(c *gin.Context, result1 string, result2 string, busi string) (interface{}, error) { |
|||
info := infoFromContext(c) |
|||
res := dtmimp.OrString(result1, result2, dtmcli.ResultSuccess) |
|||
dtmimp.Logf("%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 |
|||
} |
|||
|
|||
// BaseAddRoute add base route handler
|
|||
func BaseAddRoute(app *gin.Engine) { |
|||
app.POST(BusiAPI+"/TransIn", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return handleGeneralBusiness(c, MainSwitch.TransInResult.Fetch(), reqFrom(c).TransInResult, "transIn") |
|||
})) |
|||
app.POST(BusiAPI+"/TransOut", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return handleGeneralBusiness(c, MainSwitch.TransOutResult.Fetch(), reqFrom(c).TransOutResult, "TransOut") |
|||
})) |
|||
app.POST(BusiAPI+"/TransInConfirm", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return handleGeneralBusiness(c, MainSwitch.TransInConfirmResult.Fetch(), "", "TransInConfirm") |
|||
})) |
|||
app.POST(BusiAPI+"/TransOutConfirm", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return handleGeneralBusiness(c, MainSwitch.TransOutConfirmResult.Fetch(), "", "TransOutConfirm") |
|||
})) |
|||
app.POST(BusiAPI+"/TransInRevert", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return handleGeneralBusiness(c, MainSwitch.TransInRevertResult.Fetch(), "", "TransInRevert") |
|||
})) |
|||
app.POST(BusiAPI+"/TransOutRevert", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
return handleGeneralBusiness(c, MainSwitch.TransOutRevertResult.Fetch(), "", "TransOutRevert") |
|||
})) |
|||
app.GET(BusiAPI+"/CanSubmit", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
dtmimp.Logf("%s CanSubmit", c.Query("gid")) |
|||
return dtmimp.OrString(MainSwitch.CanSubmitResult.Fetch(), dtmcli.ResultSuccess), nil |
|||
})) |
|||
app.POST(BusiAPI+"/TransInXa", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
err := XaClient.XaLocalTransaction(c.Request.URL.Query(), func(db *sql.DB, xa *dtmcli.Xa) error { |
|||
if reqFrom(c).TransInResult == dtmcli.ResultFailure { |
|||
return dtmcli.ErrFailure |
|||
} |
|||
_, err := dtmimp.DBExec(db, "update dtm_busi.user_account set balance=balance+? where user_id=?", reqFrom(c).Amount, 2) |
|||
return err |
|||
}) |
|||
return error2Resp(err) |
|||
})) |
|||
app.POST(BusiAPI+"/TransOutXa", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
err := XaClient.XaLocalTransaction(c.Request.URL.Query(), func(db *sql.DB, xa *dtmcli.Xa) error { |
|||
if reqFrom(c).TransOutResult == dtmcli.ResultFailure { |
|||
return dtmcli.ErrFailure |
|||
} |
|||
_, err := dtmimp.DBExec(db, "update dtm_busi.user_account set balance=balance-? where user_id=?", reqFrom(c).Amount, 1) |
|||
return err |
|||
}) |
|||
return error2Resp(err) |
|||
})) |
|||
|
|||
app.POST(BusiAPI+"/TransOutXaGorm", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
err := XaClient.XaLocalTransaction(c.Request.URL.Query(), func(db *sql.DB, xa *dtmcli.Xa) error { |
|||
if reqFrom(c).TransOutResult == dtmcli.ResultFailure { |
|||
return dtmcli.ErrFailure |
|||
} |
|||
var dia gorm.Dialector = nil |
|||
if dtmcli.GetCurrentDBType() == dtmcli.DBTypeMysql { |
|||
dia = mysql.New(mysql.Config{Conn: db}) |
|||
} else if dtmcli.GetCurrentDBType() == dtmcli.DBTypePostgres { |
|||
dia = postgres.New(postgres.Config{Conn: db}) |
|||
} |
|||
gdb, err := gorm.Open(dia, &gorm.Config{}) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
dbr := gdb.Exec("update dtm_busi.user_account set balance=balance-? where user_id=?", reqFrom(c).Amount, 1) |
|||
return dbr.Error |
|||
}) |
|||
return error2Resp(err) |
|||
})) |
|||
|
|||
app.POST(BusiAPI+"/TestPanic", common.WrapHandler(func(c *gin.Context) (interface{}, error) { |
|||
if c.Query("panic_error") != "" { |
|||
panic(errors.New("panic_error")) |
|||
} else if c.Query("panic_string") != "" { |
|||
panic("panic_string") |
|||
} |
|||
return "SUCCESS", nil |
|||
})) |
|||
} |
|||
@ -1,107 +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" |
|||
"fmt" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/yedf/dtm/common" |
|||
"github.com/yedf/dtm/dtmcli" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
"github.com/yedf/dtm/dtmgrpc" |
|||
) |
|||
|
|||
// DtmHttpServer dtm service address
|
|||
var DtmHttpServer = fmt.Sprintf("http://localhost:%d/api/dtmsvr", 36789) |
|||
|
|||
// DtmGrpcServer dtm grpc service address
|
|||
var DtmGrpcServer = fmt.Sprintf("localhost:%d", 36790) |
|||
|
|||
// TransReq transaction request payload
|
|||
type TransReq struct { |
|||
Amount int `json:"amount"` |
|||
TransInResult string `json:"transInResult"` |
|||
TransOutResult string `json:"transOutResult"` |
|||
} |
|||
|
|||
func (t *TransReq) String() string { |
|||
return fmt.Sprintf("amount: %d transIn: %s transOut: %s", t.Amount, t.TransInResult, t.TransOutResult) |
|||
} |
|||
|
|||
// GenTransReq 1
|
|||
func GenTransReq(amount int, outFailed bool, inFailed bool) *TransReq { |
|||
return &TransReq{ |
|||
Amount: amount, |
|||
TransOutResult: dtmimp.If(outFailed, dtmcli.ResultFailure, "").(string), |
|||
TransInResult: dtmimp.If(inFailed, dtmcli.ResultFailure, "").(string), |
|||
} |
|||
} |
|||
|
|||
// GenBusiReq 1
|
|||
func GenBusiReq(amount int, outFailed bool, inFailed bool) *BusiReq { |
|||
return &BusiReq{ |
|||
Amount: int64(amount), |
|||
TransOutResult: dtmimp.If(outFailed, dtmcli.ResultFailure, "").(string), |
|||
TransInResult: dtmimp.If(inFailed, dtmcli.ResultFailure, "").(string), |
|||
} |
|||
} |
|||
|
|||
func reqFrom(c *gin.Context) *TransReq { |
|||
v, ok := c.Get("trans_req") |
|||
if !ok { |
|||
req := TransReq{} |
|||
err := c.BindJSON(&req) |
|||
dtmimp.FatalIfError(err) |
|||
c.Set("trans_req", &req) |
|||
v = &req |
|||
} |
|||
return v.(*TransReq) |
|||
} |
|||
|
|||
func infoFromContext(c *gin.Context) *dtmcli.BranchBarrier { |
|||
info := dtmcli.BranchBarrier{ |
|||
TransType: c.Query("trans_type"), |
|||
Gid: c.Query("gid"), |
|||
BranchID: c.Query("branch_id"), |
|||
Op: c.Query("op"), |
|||
} |
|||
return &info |
|||
} |
|||
|
|||
func dbGet() *common.DB { |
|||
return common.DbGet(config.ExamplesDB) |
|||
} |
|||
|
|||
func sdbGet() *sql.DB { |
|||
db, err := dtmimp.PooledDB(config.ExamplesDB) |
|||
dtmimp.FatalIfError(err) |
|||
return db |
|||
} |
|||
|
|||
func txGet() *sql.Tx { |
|||
db := sdbGet() |
|||
tx, err := db.Begin() |
|||
dtmimp.FatalIfError(err) |
|||
return tx |
|||
} |
|||
|
|||
// MustBarrierFromGin 1
|
|||
func MustBarrierFromGin(c *gin.Context) *dtmcli.BranchBarrier { |
|||
ti, err := dtmcli.BarrierFromQuery(c.Request.URL.Query()) |
|||
dtmimp.FatalIfError(err) |
|||
return ti |
|||
} |
|||
|
|||
// MustBarrierFromGrpc 1
|
|||
func MustBarrierFromGrpc(ctx context.Context) *dtmcli.BranchBarrier { |
|||
ti, err := dtmgrpc.BarrierFromGrpc(ctx) |
|||
dtmimp.FatalIfError(err) |
|||
return ti |
|||
} |
|||
@ -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/yedf/dtm/common" |
|||
"github.com/yedf/dtm/dtmcli/dtmimp" |
|||
) |
|||
|
|||
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) { |
|||
dtmimp.LogIfFatalf(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/yedf/dtm/dtmcli/dtmimp" |
|||
dtmgrpc "github.com/yedf/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() |
|||
dtmimp.FatalIfError(err) |
|||
return msg.Gid |
|||
}) |
|||
} |
|||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue