jyPoints/rpc/integral.go

package main

import (
	"app.yhyue.com/moapp/jyPoints/entity"
	"app.yhyue.com/moapp/jyPoints/rpc/integral"
	"app.yhyue.com/moapp/jyPoints/rpc/internal/config"
	"app.yhyue.com/moapp/jyPoints/rpc/internal/server"
	"app.yhyue.com/moapp/jyPoints/rpc/internal/svc"
	"app.yhyue.com/moapp/jyPoints/service"
	"flag"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/go-xorm/xorm"
	"github.com/tal-tech/go-zero/core/conf"
	"github.com/tal-tech/go-zero/zrpc"
	"google.golang.org/grpc"
	"log"
)

var configFile = flag.String("f", "etc/integral.yaml", "the config file")
var c config.Config
var (
	integralService = &service.IntegralService{}
)


func main() {
	conf.MustLoad(*configFile, &c)
	ctx := svc.NewServiceContext(c)
	srv := server.NewIntegralServer(ctx)
	s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
		integral.RegisterIntegralServer(grpcServer, srv)
	})
	defer s.Stop()
	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
	s.Start()
}
//创建orm引擎
func init() {
	var err error
	conf.MustLoad(*configFile, &c)
	fmt.Println(c.DataSource)
	entity.Engine, err = xorm.NewEngine("mysql", c.DataSource)
	entity.Engine.ShowSQL(true)
	if err != nil {
		log.Fatal("数据库连接失败:", err)
	}
	fmt.Println(c.DataSource+"链接成功")
}