Fix go-kit compile

Signed-off-by: Drasko Draskovic <[email protected]>

Author: drasko

cmd/main.go

@@ -1,64 +1,185 @@
+// Copyright (c) Mainflux
+// SPDX-License-Identifier: Apache-2.0
+
 package main
 
 import (
-	"html/template"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"log"
 	"net/http"
-
-	"github.com/go-chi/chi"
-	"github.com/go-chi/chi/middleware"
+	"os"
+	"os/signal"
+	"strconv"
+	"syscall"
+	"time"
+
+	"google.golang.org/grpc/credentials"
+
+	"github.com/drasko/gui/gui"
+	"github.com/drasko/gui/gui/api"
+	kitprometheus "github.com/go-kit/kit/metrics/prometheus"
+	"github.com/mainflux/mainflux"
+	"github.com/mainflux/mainflux/logger"
+	thingsapi "github.com/mainflux/mainflux/things/api/auth/grpc"
+	"github.com/opentracing/opentracing-go"
+	stdprometheus "github.com/prometheus/client_golang/prometheus"
+	jconfig "github.com/uber/jaeger-client-go/config"
+	"google.golang.org/grpc"
 )
 
 const (
-	staticDir = "gui/static"
+	defLogLevel          = "error"
+	defClientTLS         = "false"
+	defCACerts           = ""
+	defPort              = "8180"
+	defJaegerURL         = ""
+	defThingsAuthURL     = "localhost:8181"
+	defThingsAuthTimeout = "1s"
+
+	envLogLevel          = "MF_GUI_LOG_LEVEL"
+	envClientTLS         = "MF_GUI_CLIENT_TLS"
+	envCACerts           = "MF_GUI_CA_CERTS"
+	envPort              = "MF_GUI_PORT"
+	envNatsURL           = "MF_NATS_URL"
+	envJaegerURL         = "MF_JAEGER_URL"
+	envThingsAuthURL     = "MF_THINGS_AUTH_GRPC_URL"
+	envThingsAuthTimeout = "MF_THINGS_AUTH_GRPC_TIMEOUT"
 )
 
-var (
-	t *template.Template
-)
+type config struct {
+	natsURL           string
+	logLevel          string
+	port              string
+	clientTLS         bool
+	caCerts           string
+	jaegerURL         string
+	thingsAuthURL     string
+	thingsAuthTimeout time.Duration
+}
 
-func indexHandler(w http.ResponseWriter, r *http.Request) {
+func main() {
+	cfg := loadConfig()
 
-	data := struct {
-		Name string
-	}{"John Smith"}
+	logger, err := logger.New(os.Stdout, cfg.logLevel)
+	if err != nil {
+		log.Fatalf(err.Error())
+	}
 
-	t.ExecuteTemplate(w, "index", data)
+	conn := connectToThings(cfg, logger)
+	defer conn.Close()
+
+	tracer, closer := initJaeger("gui", cfg.jaegerURL, logger)
+	defer closer.Close()
+
+	thingsTracer, thingsCloser := initJaeger("things", cfg.jaegerURL, logger)
+	defer thingsCloser.Close()
+
+	tc := thingsapi.NewClient(conn, thingsTracer, cfg.thingsAuthTimeout)
+	svc := gui.New(tc)
+
+	svc = api.LoggingMiddleware(svc, logger)
+	svc = api.MetricsMiddleware(
+		svc,
+		kitprometheus.NewCounterFrom(stdprometheus.CounterOpts{
+			Namespace: "gui",
+			Subsystem: "api",
+			Name:      "request_count",
+			Help:      "Number of requests received.",
+		}, []string{"method"}),
+		kitprometheus.NewSummaryFrom(stdprometheus.SummaryOpts{
+			Namespace: "gui",
+			Subsystem: "api",
+			Name:      "request_latency_microseconds",
+			Help:      "Total duration of requests in microseconds.",
+		}, []string{"method"}),
+	)
+
+	errs := make(chan error, 2)
+
+	go func() {
+		p := fmt.Sprintf(":%s", cfg.port)
+		logger.Info(fmt.Sprintf("GUI service started on port %s", cfg.port))
+		errs <- http.ListenAndServe(p, api.MakeHandler(svc, tracer))
+	}()
+
+	go func() {
+		c := make(chan os.Signal)
+		signal.Notify(c, syscall.SIGINT)
+		errs <- fmt.Errorf("%s", <-c)
+	}()
+
+	err = <-errs
+	logger.Error(fmt.Sprintf("GUI service terminated: %s", err))
 }
 
-func thingsHandler(w http.ResponseWriter, r *http.Request) {
-	data := struct {
-		Name string
-	}{"John Smith"}
-
-	t.ExecuteTemplate(w, "things", data)
-}
+func loadConfig() config {
+	tls, err := strconv.ParseBool(mainflux.Env(envClientTLS, defClientTLS))
+	if err != nil {
+		log.Fatalf("Invalid value passed for %s\n", envClientTLS)
+	}
 
-func channelsHandler(w http.ResponseWriter, r *http.Request) {
-	data := struct {
-		Name string
-	}{"John Smith"}
+	authTimeout, err := time.ParseDuration(mainflux.Env(envThingsAuthTimeout, defThingsAuthTimeout))
+	if err != nil {
+		log.Fatalf("Invalid %s value: %s", envThingsAuthTimeout, err.Error())
+	}
 
-	t.ExecuteTemplate(w, "channels", data)
+	return config{
+		logLevel:          mainflux.Env(envLogLevel, defLogLevel),
+		port:              mainflux.Env(envPort, defPort),
+		clientTLS:         tls,
+		caCerts:           mainflux.Env(envCACerts, defCACerts),
+		jaegerURL:         mainflux.Env(envJaegerURL, defJaegerURL),
+		thingsAuthURL:     mainflux.Env(envThingsAuthURL, defThingsAuthURL),
+		thingsAuthTimeout: authTimeout,
+	}
 }
 
-func main() {
-	r := chi.NewRouter()
-	r.Use(middleware.Logger)
+func initJaeger(svcName, url string, logger logger.Logger) (opentracing.Tracer, io.Closer) {
+	if url == "" {
+		return opentracing.NoopTracer{}, ioutil.NopCloser(nil)
+	}
 
-	// Serve static files
-	fs := http.FileServer(http.Dir(staticDir))
-	r.Handle("/*", fs)
+	tracer, closer, err := jconfig.Configuration{
+		ServiceName: svcName,
+		Sampler: &jconfig.SamplerConfig{
+			Type:  "const",
+			Param: 1,
+		},
+		Reporter: &jconfig.ReporterConfig{
+			LocalAgentHostPort: url,
+			LogSpans:           true,
+		},
+	}.NewTracer()
+	if err != nil {
+		logger.Error(fmt.Sprintf("Failed to init Jaeger client: %s", err))
+		os.Exit(1)
+	}
 
-	// Routes
-	r.Get("/", indexHandler)
-	r.Get("/things", thingsHandler)
-	r.Get("/channels", channelsHandler)
+	return tracer, closer
+}
 
-	var err error
-	t, err = template.ParseGlob("gui/views/*")
-	if err != nil {
-		panic(err)
+func connectToThings(cfg config, logger logger.Logger) *grpc.ClientConn {
+	var opts []grpc.DialOption
+	if cfg.clientTLS {
+		if cfg.caCerts != "" {
+			tpc, err := credentials.NewClientTLSFromFile(cfg.caCerts, "")
+			if err != nil {
+				logger.Error(fmt.Sprintf("Failed to load certs: %s", err))
+				os.Exit(1)
+			}
+			opts = append(opts, grpc.WithTransportCredentials(tpc))
+		}
+	} else {
+		logger.Info("gRPC communication is not encrypted")
+		opts = append(opts, grpc.WithInsecure())
 	}
 
-	http.ListenAndServe(":3000", r)
+	conn, err := grpc.Dial(cfg.thingsAuthURL, opts...)
+	if err != nil {
+		logger.Error(fmt.Sprintf("Failed to connect to things service: %s", err))
+		os.Exit(1)
+	}
+	return conn
 }

go.mod

@@ -2,4 +2,15 @@ module github.com/drasko/gui
 
 go 1.15
 
-require github.com/go-chi/chi v1.5.1
+require (
+	github.com/go-chi/chi v1.5.1
+	github.com/go-kit/kit v0.10.0
+	github.com/go-zoo/bone v1.3.0
+	github.com/golang/protobuf v1.4.3
+	github.com/mainflux/mainflux v0.11.0
+	github.com/opentracing/opentracing-go v1.2.0
+	github.com/prometheus/client_golang v1.9.0
+	github.com/stretchr/testify v1.6.1
+	github.com/uber/jaeger-client-go v2.23.1+incompatible
+	google.golang.org/grpc v1.34.0
+)