cc66ceb484
- use clearer method names, comments; - define dtm usage info; - determine action by switch-case instead of if-else, because only os.Args[1] needs to be checked; - use time layout '2006-01-02 15:04:05.999' to format milliseconds; - change filenames dtmsvr/main.go to dtmsvr/dtmsvr.go; - simplify code; - use `()` to think less about operator precedence; xxx
46 lines
1021 B
Go
46 lines
1021 B
Go
package dtmsvr
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/yedf/dtm/common"
|
|
"github.com/yedf/dtm/dtmcli"
|
|
"github.com/yedf/dtm/dtmgrpc"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/yedf/dtm/examples"
|
|
)
|
|
|
|
var dtmsvrPort = 8080
|
|
var dtmsvrGrpcPort = 58080
|
|
|
|
// 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(grpc.UnaryInterceptor(dtmgrpc.GrpcServerLog))
|
|
dtmgrpc.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.GetCallerCodeDir(), config.DB["driver"])
|
|
examples.RunSQLScript(config.DB, file, skipDrop)
|
|
}
|