dtm/app/main.go
2021-07-10 22:18:12 +08:00

68 lines
1.6 KiB
Go

package main
import (
"os"
"time"
"github.com/sirupsen/logrus"
"github.com/yedf/dtm/dtmsvr"
"github.com/yedf/dtm/examples"
)
type M = map[string]interface{}
func wait() {
time.Sleep(10000 * time.Second)
}
func main() {
if len(os.Args) > 1 && (os.Args[1] == "quick_start" || os.Args[1] == "qs") {
dtmsvr.PopulateMysql()
dtmsvr.MainStart()
examples.StartMain()
wait()
}
app := examples.BaseAppNew()
examples.BaseAppSetup(app)
if len(os.Args) == 1 || os.Args[1] == "dtmsvr" { // 只启动dtmsvr
go dtmsvr.MainStart()
} else if os.Args[1] == "xa" { // 启动xa示例
dtmsvr.PopulateMysql()
dtmsvr.MainStart()
examples.PopulateMysql()
examples.XaSetup(app)
examples.BaseAppStart(app)
examples.XaFireRequest()
} else if os.Args[1] == "saga" { // 启动saga示例
dtmsvr.PopulateMysql()
dtmsvr.MainStart()
examples.SagaSetup(app)
examples.BaseAppStart(app)
examples.SagaFireRequest()
} else if os.Args[1] == "all" { // 运行所有示例
dtmsvr.PopulateMysql()
examples.PopulateMysql()
dtmsvr.MainStart()
examples.SagaSetup(app)
examples.TccSetup(app)
examples.XaSetup(app)
examples.BaseAppStart(app)
examples.SagaFireRequest()
examples.TccFireRequest()
examples.XaFireRequest()
} else if os.Args[1] == "saga_barrier" {
dtmsvr.PopulateMysql()
dtmsvr.MainStart()
examples.PopulateMysql()
examples.SagaBarrierMainStart()
} else if os.Args[1] == "tcc_barrier" {
dtmsvr.PopulateMysql()
dtmsvr.MainStart()
examples.PopulateMysql()
examples.TccBarrierMainStart()
} else {
logrus.Fatalf("unknown arg: %s", os.Args[1])
}
wait()
}