43 lines
944 B
Go
43 lines
944 B
Go
package dtmsvr
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/yedf/dtm/common"
|
|
"github.com/yedf/dtm/dtmcli"
|
|
"github.com/yedf/dtm/examples"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
var dtmsvrPort = 8080
|
|
var dtmsvrGrpcPort = 50081
|
|
|
|
// StartSvr StartSvr
|
|
func StartSvr() {
|
|
dtmcli.Logf("start dtmsvr")
|
|
app := common.GetGinApp()
|
|
addRoute(app)
|
|
dtmcli.Logf("dtmsvr listen at: %d", dtmsvrPort)
|
|
go app.Run(fmt.Sprintf(":%d", dtmsvrPort))
|
|
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", dtmsvrGrpcPort))
|
|
dtmcli.FatalIfError(err)
|
|
s := grpc.NewServer()
|
|
dtmcli.RegisterDtmServer(s, &dtmServer{})
|
|
dtmcli.Logf("grpc listening at %v", lis.Addr())
|
|
go func() {
|
|
err := s.Serve(lis)
|
|
dtmcli.FatalIfError(err)
|
|
}()
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
|
|
// PopulateDB setup mysql data
|
|
func PopulateDB(skipDrop bool) {
|
|
file := fmt.Sprintf("%s/dtmsvr.%s.sql", common.GetCurrentCodeDir(), config.DB["driver"])
|
|
examples.RunSQLScript(config.DB, file, skipDrop)
|
|
}
|