MessageCenter/rpc/message.go

// Code generated by goctl. DO NOT EDIT!
// Source: message.proto

package main

import (
	"app.yhyue.com/moapp/MessageCenter/entity"
	"flag"
	"fmt"
	"github.com/go-xorm/xorm"
	clientv3 "go.etcd.io/etcd/client/v3"
	"log"
	"time"

	"app.yhyue.com/moapp/MessageCenter/rpc/internal/config"
	"app.yhyue.com/moapp/MessageCenter/rpc/internal/server"
	"app.yhyue.com/moapp/MessageCenter/rpc/internal/svc"
	"app.yhyue.com/moapp/MessageCenter/rpc/message"
	_ "github.com/go-sql-driver/mysql"

	"github.com/tal-tech/go-zero/core/conf"
	"github.com/tal-tech/go-zero/zrpc"
	"google.golang.org/grpc"
)

var configFile = flag.String("f", "etc/message.yaml", "the config file")
var EtcdCli *clientv3.Client

func main() {
	flag.Parse()

	var c config.Config
	conf.MustLoad(*configFile, &c)
	ctx := svc.NewServiceContext(c)
	srv := server.NewMessageServer(ctx)

	s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
		message.RegisterMessageServer(grpcServer, srv)
	})
	defer s.Stop()

	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
	s.Start()
}

//创建orm引擎
func init() {
	conf.MustLoad(*configFile, &config.ConfigJson)
	var err error
	log.Println(config.ConfigJson.DataSource)
	entity.Engine, err = xorm.NewEngine("mysql", config.ConfigJson.DataSource)
	log.Println(err)
	entity.Engine.ShowSQL(true)
	if err != nil {
		log.Fatal("数据库连接失败:", err)
	}
	fmt.Println(config.ConfigJson.DataSource + "链接成功")

	//var err error
	EtcdCli, err = clientv3.New(clientv3.Config{
		Endpoints:   config.ConfigJson.Etcd.Hosts,
		DialTimeout: 5 * time.Second,
	})
	if err != nil {
		// handle error!
		log.Printf("connect to etcd failed, err:%v\n", err)
		return
	}
	log.Println("connect to etcd success")
}