2023-04-26 06:42:59 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
2023-05-04 06:20:55 +00:00
|
|
|
"flag"
|
2023-04-26 06:42:59 +00:00
|
|
|
"fmt"
|
|
|
|
|
"os"
|
|
|
|
|
"os/signal"
|
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
"myschools.me/campus/demo-2/consul"
|
|
|
|
|
"myschools.me/campus/demo-2/gin"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
|
|
2023-05-04 06:20:55 +00:00
|
|
|
appname := flag.String("appname", "demo2", "app name for current")
|
|
|
|
|
port := flag.Int("port", 8080, "port for gin")
|
|
|
|
|
addr := flag.String("addr", "localhost", "ip address for gin")
|
|
|
|
|
flag.Parse()
|
2023-04-26 06:42:59 +00:00
|
|
|
|
2023-05-04 06:20:55 +00:00
|
|
|
consul.Init(nil)
|
2023-04-26 06:42:59 +00:00
|
|
|
|
2023-05-04 06:20:55 +00:00
|
|
|
gin.Service(&gin.Config{
|
|
|
|
|
RootPath: `/demo2`,
|
|
|
|
|
Addr: *addr,
|
|
|
|
|
Port: *port,
|
2023-04-26 06:42:59 +00:00
|
|
|
})
|
|
|
|
|
|
2023-05-04 06:20:55 +00:00
|
|
|
traefikTag := fmt.Sprintf("traefik.http.routers.%s.rule=PathPrefix(`/demo2`)", *appname)
|
|
|
|
|
if err := consul.RegisterAPI(*appname, *addr, *port, traefikTag); err != nil {
|
2023-04-26 06:42:59 +00:00
|
|
|
fmt.Println("注册失败", err.Error())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 服务停止相应
|
|
|
|
|
c := make(chan os.Signal, 1)
|
|
|
|
|
signal.Notify(c, os.Interrupt)
|
|
|
|
|
<-c
|
|
|
|
|
_, cancel := context.WithTimeout(context.Background(), 15*time.Second)
|
|
|
|
|
defer cancel()
|
2023-05-04 06:20:55 +00:00
|
|
|
//取消注册
|
|
|
|
|
err := consul.DeRegister(*appname, *addr, *port)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Println("Error: ", err)
|
|
|
|
|
}
|
|
|
|
|
// exceptionless.SubmitLog("campus-demo2 service shutting down", "info")
|
2023-04-26 06:42:59 +00:00
|
|
|
os.Exit(0)
|
|
|
|
|
}
|