/
main.go
158 lines (142 loc) · 4.27 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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
package main
import (
"flag"
"fmt"
"github.com/afex/hystrix-go/hystrix"
"github.com/go-kit/kit/log"
"github.com/hashicorp/consul/api"
"github.com/openzipkin/zipkin-go"
zipkinhttpsvr "github.com/openzipkin/zipkin-go/middleware/http"
zipkinhttp "github.com/openzipkin/zipkin-go/reporter/http"
"net"
"net/http"
"os"
"os/signal"
"syscall"
)
func main() {
// 创建环境变量
var (
consulHost = flag.String("consul.host", "192.168.192.146", "consul server ip address")
consulPort = flag.String("consul.port", "8500", "consul server port")
zipkinURL = flag.String("zipkin.url", "http://192.168.192.146:9411/api/v2/spans", "Zipkin server url")
)
flag.Parse()
//创建日志组件
var logger log.Logger
{
logger = log.NewLogfmtLogger(os.Stderr)
logger = log.With(logger, "ts", log.DefaultTimestampUTC)
logger = log.With(logger, "caller", log.DefaultCaller)
}
var zipkinTracer *zipkin.Tracer
{
var (
err error
hostPort = "localhost:9090"
serviceName = "gateway-service"
useNoopTracer = (*zipkinURL == "")
reporter = zipkinhttp.NewReporter(*zipkinURL)
)
defer reporter.Close()
zEP, _ := zipkin.NewEndpoint(serviceName, hostPort)
zipkinTracer, err = zipkin.NewTracer(
reporter, zipkin.WithLocalEndpoint(zEP), zipkin.WithNoopTracer(useNoopTracer),
)
if err != nil {
logger.Log("err", err)
os.Exit(1)
}
if !useNoopTracer {
logger.Log("tracer", "Zipkin", "type", "Native", "URL", *zipkinURL)
}
}
// 创建consul api客户端
consulConfig := api.DefaultConfig()
consulConfig.Address = "http://" + *consulHost + ":" + *consulPort
consulClient, err := api.NewClient(consulConfig)
if err != nil {
logger.Log("err", err)
os.Exit(1)
}
//创建反向代理
//proxy := NewReverseProxy(consulClient, zipkinTracer, logger)
tags := map[string]string{
"component": "gateway_server",
}
hystrixRouter := Routes(consulClient, zipkinTracer, "Circuit Breaker:Service unavailable", logger)
handler := zipkinhttpsvr.NewServerMiddleware(
zipkinTracer,
zipkinhttpsvr.SpanName("gateway"),
zipkinhttpsvr.TagResponseSize(true),
zipkinhttpsvr.ServerTags(tags),
)(hystrixRouter)
errc := make(chan error)
//启用hystrix实时监控,监听端口为9010
hystrixStreamHandler := hystrix.NewStreamHandler()
hystrixStreamHandler.Start()
go func() {
errc <- http.ListenAndServe(net.JoinHostPort("", "9010"), hystrixStreamHandler)
}()
go func() {
c := make(chan os.Signal)
signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
errc <- fmt.Errorf("%s", <-c)
}()
//开始监听
go func() {
logger.Log("transport", "HTTP", "addr", "9090")
errc <- http.ListenAndServe(":9090", handler)
}()
// 开始运行,等待结束
logger.Log("exit", <-errc)
}
//// NewReverseProxy 创建反向代理处理方法
//func NewReverseProxy(client *api.Client, zikkinTracer *zipkin.Tracer, logger log.Logger) http.Handler {
//
// //创建Director
// director := func(req *http.Request) {
//
// //查询原始请求路径,如:/arithmetic/calculate/10/5
// reqPath := req.URL.Path
// if reqPath == "" {
// return
// }
// //按照分隔符'/'对路径进行分解,获取服务名称serviceName
// pathArray := strings.Split(reqPath, "/")
// serviceName := pathArray[1]
//
// //调用consul api查询serviceName的服务实例列表
// result, _, err := client.Catalog().Service(serviceName, "", nil)
// if err != nil {
// logger.Log("ReverseProxy failed", "query service instace error", err.Error())
// return
// }
//
// if len(result) == 0 {
// logger.Log("ReverseProxy failed", "no such service instance", serviceName)
// return
// }
//
// //重新组织请求路径,去掉服务名称部分
// destPath := strings.Join(pathArray[2:], "/")
//
// //随机选择一个服务实例
// tgt := result[rand.Int()%len(result)]
// logger.Log("service id", tgt.ServiceID)
//
// //设置代理服务地址信息
// req.URL.Scheme = "http"
// req.URL.Host = fmt.Sprintf("%s:%d", tgt.ServiceAddress, tgt.ServicePort)
// req.URL.Path = "/" + destPath
//
// }
//
// // 为反向代理增加追踪逻辑,使用如下RoundTrip代替默认Transport
// roundTrip, _ := zipkinhttpsvr.NewTransport(zikkinTracer, zipkinhttpsvr.TransportTrace(true))
//
// return &httputil.ReverseProxy{
// Director: director,
// Transport: roundTrip,
// }
//}