add deploy scripts

Signed-off-by: cwen0 <[email protected]>

Author: cwen0

Dockerfile

@@ -1,8 +1,9 @@
-FROM alpine:3.10
-
-RUN apk add tzdata --no-cache
+FROM centos:7
 
 ADD ./bin/web-show /usr/local/bin/web-show
-ADD  ./web/build /build
+ADD  ./web/build /web
+
+RUN setcap cap_net_raw=+ep /usr/local/bin/web-show
+# RUN sysctl -w net.ipv4.ping_group_range='0 2147483647'
 
 CMD ["/usr/local/bin/web-show"]

README.md

@@ -0,0 +1 @@
+# web-show

deploy.sh

@@ -0,0 +1,12 @@
+#!/usr/bin/env bash
+
+set -e
+
+TARGET_IP=$(kubectl get pod -n kube-system -o wide| grep kube-controller | head -n 1 | awk '{print $6}')
+
+sed "s/TARGETIP/$TARGET_IP/g" deploy/deployment.yaml > deploy/deployment.yamlg
+
+mv deploy/deployment.yamlg deploy/deployment.yaml
+
+kubectl apply -f deploy/
+

deploy/deployment.yaml

@@ -0,0 +1,27 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: web-show
+  labels:
+    app: web-show
+spec:
+  replicas: 1
+  selector:
+    matchLabels:
+      app: web-show
+  template:
+    metadata:
+      labels:
+        app: web-show
+    spec:
+      containers:
+        - name: web-show
+          image: pingcap/web-show
+          imagePullPolicy: Always
+          command:
+            - /usr/local/bin/web-show
+            - --target-ip=TARGETIP
+          ports:
+            - name: web-port
+              containerPort: 8081
+              hostPort: 8081

deploy/service.yaml

@@ -0,0 +1,13 @@
+apiVersion: v1
+kind: Service
+metadata:
+  name: web-show
+  labels:
+    app: web-show
+spec:
+  selector:
+    app: web-show
+  ports:
+    - protocol: TCP
+      port: 8081
+      targetPort: 8081

go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/sparrc/go-ping v0.0.0-20190613174326-4e5b6552494c
 	github.com/stretchr/testify v1.4.0 // indirect
 	github.com/unrolled/render v1.0.2
+	go.uber.org/zap v1.9.1
 	golang.org/x/net v0.0.0-20200202094626-16171245cfb2 // indirect
 	golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a // indirect
 	gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22 // indirect

main.go

@@ -1,12 +1,21 @@
 package main
 
 import (
+	"flag"
+
 	"github.com/chaos-mesh/web-show/server"
 	"github.com/pingcap/log"
 )
 
 func main() {
+	conf := &server.Config{}
+
+	flag.StringVar(&conf.Port, "port", ":8081", "the port of address")
+	flag.StringVar(&conf.StaticPath, "static-path", "/web", "the path of static file")
+	flag.StringVar(&conf.TargetIP, "target-ip", "", "the ip of target pod")
+	flag.Parse()
+
 	log.Info("Starting server")
-	s := server.SetupServer()
+	s := server.SetupServer(conf)
 	s.Run()
 }

server/nework.go

@@ -9,10 +9,6 @@ import (
 	"github.com/unrolled/render"
 )
 
-const (
-	GoogleAddress = "www.google.com"
-)
-
 type PingD struct {
 	Time  int64   `json:"time"`
 	Delay float64 `json:"delay"`
@@ -30,12 +26,13 @@ func (s *Server) startPing() {
 	for {
 		select {
 		case <-ticker.C:
-			pinger, err := ping.NewPinger(GoogleAddress)
+			pinger, err := ping.NewPinger(s.conf.TargetIP)
 			if err != nil {
 				log.Error(err.Error())
 				s.push(&PingD{Time: time.Now().UnixNano() / 1e6})
 				continue
 			}
+			pinger.SetPrivileged(true)
 			pinger.Timeout = 5 * time.Second
 			pinger.Count = 1
 			pinger.Run() // blocks until finished

server/server.go

@@ -1,38 +1,48 @@
 package server
 
 import (
+	"go.uber.org/zap"
 	"net/http"
 
 	"github.com/chaos-mesh/web-show/pkg/queue"
 	"github.com/gorilla/mux"
 	"github.com/pingcap/log"
 )
 
-func SetupServer() Server {
+type Config struct {
+	Port       string
+	StaticPath string
+	TargetIP   string
+}
+
+func SetupServer(conf *Config) Server {
 	r := mux.NewRouter()
 
 	server := Server{
 		router:   r,
 		pingData: queue.NewListQueue(200),
+		conf:     conf,
 	}
 
 	r.PathPrefix("/api/network").HandlerFunc(server.network)
 
-	r.PathPrefix("/").Handler(server.web("/", "./web/build"))
+	r.PathPrefix("/").Handler(server.web("/", server.conf.StaticPath))
 
 	return server
 }
 
 func (s *Server) Run() {
 	go s.startPing()
-	log.Info("Starting Server on 0.0.0.0:8081")
-	err := http.ListenAndServe("0.0.0.0:8081", s.router)
+	log.Info("Starting Server", zap.Reflect("config", s.conf))
+	err := http.ListenAndServe(s.conf.Port, s.router)
 	if err != nil {
-		log.Error("Error while listening 0.0.0.0:8081")
+		log.Error("Error while listening 0.0.0.0:8081", zap.Reflect("config", s.conf))
 	}
 }
 
 type Server struct {
 	router   *mux.Router
 	pingData queue.Queue
+
+	conf *Config
 }

web/src/App.tsx

@@ -52,7 +52,7 @@ const App: React.FC = () => {
                 <Banner />
                 <BodyStyles className="body">
                     <Row>
-                        <TrafficChartContainer title="Ping www.google.com" />
+                        <TrafficChartContainer title="Ping kube-system Pod" />
                     </Row>
                 </BodyStyles>
             </AppStyles>