mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
119 lines
3.4 KiB
Go
119 lines
3.4 KiB
Go
// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
|
|
// Licensed under the GNU Affero General Public License (AGPL).
|
|
// See License-AGPL.txt in the project root for license information.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/collectors"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/spf13/cobra"
|
|
"google.golang.org/grpc"
|
|
|
|
common_grpc "github.com/gitpod-io/gitpod/common-go/grpc"
|
|
"github.com/gitpod-io/gitpod/common-go/log"
|
|
"github.com/gitpod-io/gitpod/common-go/pprof"
|
|
"github.com/gitpod-io/gitpod/ws-daemon/pkg/daemon"
|
|
)
|
|
|
|
// serveCmd represents the serve command
|
|
var runCmd = &cobra.Command{
|
|
Use: "run",
|
|
Short: "Connects to the messagebus and starts the workspace monitor",
|
|
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cfg := getConfig()
|
|
reg := prometheus.NewRegistry()
|
|
dmn, err := daemon.NewDaemon(cfg.Daemon, prometheus.WrapRegistererWithPrefix("gitpod_ws_daemon_", reg))
|
|
if err != nil {
|
|
log.WithError(err).Fatal("cannot create daemon")
|
|
}
|
|
|
|
common_grpc.SetupLogging()
|
|
|
|
grpcMetrics := grpc_prometheus.NewServerMetrics()
|
|
grpcMetrics.EnableHandlingTimeHistogram()
|
|
reg.MustRegister(grpcMetrics)
|
|
|
|
grpcOpts := common_grpc.ServerOptionsWithInterceptors(
|
|
[]grpc.StreamServerInterceptor{grpcMetrics.StreamServerInterceptor()},
|
|
[]grpc.UnaryServerInterceptor{grpcMetrics.UnaryServerInterceptor()},
|
|
)
|
|
tlsOpt, err := cfg.Service.TLS.ServerOption()
|
|
if err != nil {
|
|
log.WithError(err).Fatal("cannot use TLS config")
|
|
}
|
|
if tlsOpt != nil {
|
|
log.WithField("crt", cfg.Service.TLS.Certificate).WithField("key", cfg.Service.TLS.PrivateKey).Debug("securing gRPC server with TLS")
|
|
grpcOpts = append(grpcOpts, tlsOpt)
|
|
} else {
|
|
log.Warn("no TLS configured - gRPC server will be unsecured")
|
|
}
|
|
|
|
server := grpc.NewServer(grpcOpts...)
|
|
dmn.Register(server)
|
|
lis, err := net.Listen("tcp", cfg.Service.Addr)
|
|
if err != nil {
|
|
log.WithError(err).Fatalf("cannot listen on %s", cfg.Service.Addr)
|
|
}
|
|
go func() {
|
|
err := server.Serve(lis)
|
|
if err != nil {
|
|
log.WithError(err).Fatal("cannot start server")
|
|
}
|
|
}()
|
|
log.WithField("addr", cfg.Service.Addr).Info("started gRPC server")
|
|
|
|
if cfg.Prometheus.Addr != "" {
|
|
reg.MustRegister(
|
|
collectors.NewGoCollector(),
|
|
collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}),
|
|
)
|
|
|
|
handler := http.NewServeMux()
|
|
handler.Handle("/metrics", promhttp.HandlerFor(reg, promhttp.HandlerOpts{}))
|
|
|
|
go func() {
|
|
err := http.ListenAndServe(cfg.Prometheus.Addr, handler)
|
|
if err != nil {
|
|
log.WithError(err).Error("Prometheus metrics server failed")
|
|
}
|
|
}()
|
|
log.WithField("addr", cfg.Prometheus.Addr).Info("started Prometheus metrics server")
|
|
}
|
|
|
|
if cfg.PProf.Addr != "" {
|
|
go pprof.Serve(cfg.PProf.Addr)
|
|
}
|
|
|
|
err = dmn.Start()
|
|
if err != nil {
|
|
log.WithError(err).Fatal("cannot start daemon")
|
|
}
|
|
|
|
// run until we're told to stop
|
|
sigChan := make(chan os.Signal, 1)
|
|
signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
|
|
log.Info("🧫 ws-daemon is up and running. Stop with SIGINT or CTRL+C")
|
|
<-sigChan
|
|
server.Stop()
|
|
err = dmn.Stop()
|
|
if err != nil {
|
|
log.WithError(err).Error("cannot shut down gracefully")
|
|
}
|
|
log.Info("Received SIGINT - shutting down")
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(runCmd)
|
|
}
|