consul-demo/main.go

package main

import (
	"context"
	"log"
	"net"
	"os"
	"os/signal"
	"time"

	"google.golang.org/grpc"
	"myschools.me/suguo/consul-demo/consul"
	"myschools.me/suguo/consul-demo/health"
	pb "myschools.me/suguo/consul-demo/proto"
	"myschools.me/suguo/consul-demo/sample"
)

const (
	//APPNAME 每个服务默认有一个名称
	APPNAME string = "demo"
)

func main() {
	ctx := context.Background()
	ctx, cancel := context.WithCancel(ctx)
	defer cancel()

	// grpc服务定义
	s := grpc.NewServer()
	pb.RegisterHelloServer(s, &sample.Server{})
	pb.RegisterHealthServer(s, &health.Server{})

	// grpc服务启动
	go func() {
		log.Printf("starting grpc service on %s\n", ":9001")
		lis, err := net.Listen("tcp", ":9001")
		if err != nil {
			log.Fatal("fail to open port: ", err)
		}
		err = s.Serve(lis)
		if err != nil {
			log.Fatal("fail to open microservice: ", err)
		}
	}()

	//注册至consul
	consul.Register(APPNAME, APPNAME, 9001, "demo")
	defer consul.DeRegister(APPNAME)

	// 服务停止相应
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)
	<-c
	ctx, cancel = context.WithTimeout(context.Background(), 15*time.Second)
	defer cancel()
	log.Println("shutting down")
	os.Exit(0)
}