/
gateway.go
58 lines (48 loc) · 1.28 KB
/
gateway.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
package main
import (
"flag"
"fmt"
kitlog "github.com/go-kit/kit/log"
"github.com/pp553933054/micro-go-book/common/discover"
"github.com/pp553933054/micro-go-book/common/loadbalance"
"log"
"net/http"
"os"
"os/signal"
"syscall"
)
func main() {
// 创建环境变量
var (
consulHost = flag.String("consul.host", "127.0.0.1", "consul server ip address")
consulPort = flag.Int("consul.port", 8500, "consul server port")
)
flag.Parse()
//创建日志组件
var logger kitlog.Logger
{
logger = kitlog.NewLogfmtLogger(os.Stderr)
logger = kitlog.With(logger, "ts", kitlog.DefaultTimestampUTC)
logger = kitlog.With(logger, "caller", kitlog.DefaultCaller)
}
consulClient, err := discover.NewKitDiscoverClient(*consulHost, *consulPort)
if err != nil {
logger.Log("err", err)
os.Exit(1)
}
//创建反向代理
proxy := NewHystrixHandler(consulClient, new(loadbalance.RandomLoadBalance), log.New(os.Stderr, "", log.LstdFlags))
errc := make(chan error)
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", proxy)
}()
// 开始运行,等待结束
logger.Log("exit", <-errc)
}