kfcoding-terminal-api/main.go at master · kfcoding/kfcoding-terminal-api · GitHub

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
package main

import (
	"log"
	"net/http"
	"github.com/kfcoding-terminal-controller/config"
	"github.com/kfcoding-terminal-controller/service/common"
	"github.com/kfcoding-terminal-controller/service"
	"path"
	"github.com/kfcoding-terminal-controller/handler"
)

func main() {

	config.InitEnv()
	var k8sClient *common.K8sClient
	if config.InCluster != "" {
		k8sClient = common.InitInClusterK8sClient()
	} else {
		k8sClient = common.InitOutClusterK8sClient()
	}

	etcdClient := common.GetMyEtcdClient()
	etcdService := service.GetEtcdService(etcdClient)
	k8sService := service.GetK8sService(k8sClient)
	sessionService := service.GetSerssionService(k8sClient)

	terminalService := &service.TerminalService{
		EtcdService:    etcdService,
		K8sService:     k8sService,
		SessionService: sessionService,
	}

	etcdService.SetOnDeleteCallback(terminalService.Delete)
	sessionService.SetOnCloseCallback(terminalService.Delete)

	go etcdService.WatchSessionId(path.Join(config.KeeperPrefix, config.Version))

	apiHandler, err := handler.CreateHTTPAPIHandler(terminalService)
	if err != nil {
		log.Fatal(err)
	}

	http.Handle("/api/", apiHandler)
	http.Handle("/api/sockjs/", handler.CreateAttachHandler(sessionService, "/api/sockjs"))
	http.Handle("/", http.FileServer(http.Dir("/Users/wsl/Go/src/github.com/kfcoding-terminal-controller/ui/static/")))

	log.Println("Start terminal server")
	log.Fatal(http.ListenAndServe(config.ServerAddress, nil))
}