/
main.go
88 lines (72 loc) · 2.67 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package main
import (
"context"
"flag"
"fmt"
"github.com/go-kit/kit/circuitbreaker"
"github.com/longjoy/micro-go-book/ch10-resiliency/use-string-service/config"
"github.com/longjoy/micro-go-book/ch10-resiliency/use-string-service/endpoint"
"github.com/longjoy/micro-go-book/ch10-resiliency/use-string-service/service"
"github.com/longjoy/micro-go-book/ch10-resiliency/use-string-service/transport"
"github.com/longjoy/micro-go-book/common/discover"
"github.com/longjoy/micro-go-book/common/loadbalance"
uuid "github.com/satori/go.uuid"
"net/http"
"os"
"os/signal"
"strconv"
"syscall"
)
func main() {
var (
servicePort = flag.Int("service.port", 10086, "service port")
serviceHost = flag.String("service.host", "127.0.0.1", "service host")
consulPort = flag.Int("consul.port", 8500, "consul port")
consulHost = flag.String("consul.host", "127.0.0.1", "consul host")
serviceName = flag.String("service.name", "use-string", "service name")
)
flag.Parse()
ctx := context.Background()
errChan := make(chan error)
var discoveryClient discover.DiscoveryClient
discoveryClient, err := discover.NewKitDiscoverClient(*consulHost, *consulPort)
if err != nil{
config.Logger.Println("Get Consul Client failed")
os.Exit(-1)
}
var svc service.Service
svc = service.NewUseStringService(discoveryClient, &loadbalance.RandomLoadBalance{} )
useStringEndpoint := endpoint.MakeUseStringEndpoint(svc)
useStringEndpoint = circuitbreaker.Hystrix(service.StringServiceCommandName)(useStringEndpoint)
//创建健康检查的Endpoint
healthEndpoint := endpoint.MakeHealthCheckEndpoint(svc)
//把算术运算Endpoint和健康检查Endpoint封装至StringEndpoints
endpts := endpoint.UseStringEndpoints{
UseStringEndpoint: useStringEndpoint,
HealthCheckEndpoint: healthEndpoint,
}
//创建http.Handler
r := transport.MakeHttpHandler(ctx, endpts, config.KitLogger)
instanceId := *serviceName + "-" + uuid.NewV4().String()
//http server
go func() {
config.Logger.Println("Http Server start at port:" + strconv.Itoa(*servicePort))
//启动前执行注册
if !discoveryClient.Register(*serviceName, instanceId, "/health", *serviceHost, *servicePort, nil, config.Logger){
config.Logger.Printf("use-string-service for service %s failed.", serviceName)
// 注册失败,服务启动失败
os.Exit(-1)
}
handler := r
errChan <- http.ListenAndServe(":" + strconv.Itoa(*servicePort), handler)
}()
go func() {
c := make(chan os.Signal, 1)
signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
errChan <- fmt.Errorf("%s", <-c)
}()
error := <-errChan
//服务退出取消注册
discoveryClient.DeRegister(instanceId, config.Logger)
config.Logger.Println(error)
}