51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package main
|
||
|
||
import (
|
||
"os"
|
||
"time"
|
||
|
||
"github.com/yedf/dtm/dtmcli"
|
||
"github.com/yedf/dtm/dtmsvr"
|
||
"github.com/yedf/dtm/examples"
|
||
)
|
||
|
||
// M alias
|
||
type M = map[string]interface{}
|
||
|
||
func wait() {
|
||
for {
|
||
time.Sleep(10000 * time.Second)
|
||
}
|
||
}
|
||
|
||
func main() {
|
||
onlyServer := len(os.Args) > 1 && os.Args[1] == "dtmsvr"
|
||
if !onlyServer { // 实际线上运行，只启动dtmsvr，不准备table相关的数据
|
||
dtmsvr.PopulateDB(true)
|
||
}
|
||
dtmsvr.StartSvr() // 启动dtmsvr的api服务
|
||
go dtmsvr.CronExpiredTrans(-1) // 启动dtmsvr的定时过期查询
|
||
|
||
if onlyServer || len(os.Args) == 1 { // 没有参数，或者参数为dtmsvr，则不运行例子
|
||
wait()
|
||
}
|
||
|
||
examples.PopulateDB(true)
|
||
|
||
// quick_start 比较独立，单独作为一个例子运行，方便新人上手
|
||
if len(os.Args) > 1 && (os.Args[1] == "quick_start" || os.Args[1] == "qs") {
|
||
examples.QsStartSvr()
|
||
examples.QsFireRequest()
|
||
wait()
|
||
}
|
||
|
||
// 下面是各类的例子
|
||
examples.BaseAppStartup()
|
||
examples.GrpcStartup()
|
||
|
||
fn := examples.Samples[os.Args[1]]
|
||
dtmcli.LogIfFatalf(fn == nil, "no sample name for %s", os.Args[1])
|
||
fn()
|
||
wait()
|
||
}
|