mirror of
https://github.com/argoproj/argo-cd.git
synced 2026-04-01 22:38:46 +02:00
133 lines
4.3 KiB
Go
133 lines
4.3 KiB
Go
package cmpserver
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"
|
|
|
|
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
|
grpc_logrus "github.com/grpc-ecosystem/go-grpc-middleware/logging/logrus"
|
|
grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
|
|
log "github.com/sirupsen/logrus"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/health"
|
|
"google.golang.org/grpc/health/grpc_health_v1"
|
|
"google.golang.org/grpc/reflection"
|
|
|
|
"google.golang.org/grpc/keepalive"
|
|
|
|
"github.com/argoproj/argo-cd/v2/cmpserver/apiclient"
|
|
"github.com/argoproj/argo-cd/v2/cmpserver/plugin"
|
|
"github.com/argoproj/argo-cd/v2/common"
|
|
versionpkg "github.com/argoproj/argo-cd/v2/pkg/apiclient/version"
|
|
"github.com/argoproj/argo-cd/v2/server/version"
|
|
"github.com/argoproj/argo-cd/v2/util/errors"
|
|
grpc_util "github.com/argoproj/argo-cd/v2/util/grpc"
|
|
)
|
|
|
|
// ArgoCDCMPServer is the config management plugin server implementation
|
|
type ArgoCDCMPServer struct {
|
|
log *log.Entry
|
|
opts []grpc.ServerOption
|
|
initConstants plugin.CMPServerInitConstants
|
|
stopCh chan os.Signal
|
|
doneCh chan interface{}
|
|
sig os.Signal
|
|
}
|
|
|
|
// NewServer returns a new instance of the Argo CD config management plugin server
|
|
func NewServer(initConstants plugin.CMPServerInitConstants) (*ArgoCDCMPServer, error) {
|
|
if os.Getenv(common.EnvEnableGRPCTimeHistogramEnv) == "true" {
|
|
grpc_prometheus.EnableHandlingTimeHistogram()
|
|
}
|
|
|
|
serverLog := log.NewEntry(log.StandardLogger())
|
|
streamInterceptors := []grpc.StreamServerInterceptor{
|
|
otelgrpc.StreamServerInterceptor(), //nolint:staticcheck // TODO: ignore SA1019 for depreciation: see https://github.com/argoproj/argo-cd/issues/18258
|
|
grpc_logrus.StreamServerInterceptor(serverLog),
|
|
grpc_prometheus.StreamServerInterceptor,
|
|
grpc_util.PanicLoggerStreamServerInterceptor(serverLog),
|
|
}
|
|
unaryInterceptors := []grpc.UnaryServerInterceptor{
|
|
otelgrpc.UnaryServerInterceptor(), //nolint:staticcheck // TODO: ignore SA1019 for depreciation: see https://github.com/argoproj/argo-cd/issues/18258
|
|
grpc_logrus.UnaryServerInterceptor(serverLog),
|
|
grpc_prometheus.UnaryServerInterceptor,
|
|
grpc_util.PanicLoggerUnaryServerInterceptor(serverLog),
|
|
}
|
|
|
|
serverOpts := []grpc.ServerOption{
|
|
grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(unaryInterceptors...)),
|
|
grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(streamInterceptors...)),
|
|
grpc.MaxRecvMsgSize(apiclient.MaxGRPCMessageSize),
|
|
grpc.MaxSendMsgSize(apiclient.MaxGRPCMessageSize),
|
|
grpc.KeepaliveEnforcementPolicy(
|
|
keepalive.EnforcementPolicy{
|
|
MinTime: common.GetGRPCKeepAliveEnforcementMinimum(),
|
|
},
|
|
),
|
|
}
|
|
|
|
return &ArgoCDCMPServer{
|
|
log: serverLog,
|
|
opts: serverOpts,
|
|
stopCh: make(chan os.Signal),
|
|
doneCh: make(chan interface{}),
|
|
initConstants: initConstants,
|
|
}, nil
|
|
}
|
|
|
|
func (a *ArgoCDCMPServer) Run() {
|
|
config := a.initConstants.PluginConfig
|
|
|
|
// Listen on the socket address
|
|
_ = os.Remove(config.Address())
|
|
listener, err := net.Listen("unix", config.Address())
|
|
errors.CheckError(err)
|
|
log.Infof("argocd-cmp-server %s serving on %s", common.GetVersion(), listener.Addr())
|
|
|
|
signal.Notify(a.stopCh, syscall.SIGINT, syscall.SIGTERM)
|
|
go a.Shutdown(config.Address())
|
|
|
|
grpcServer, err := a.CreateGRPC()
|
|
errors.CheckError(err)
|
|
err = grpcServer.Serve(listener)
|
|
errors.CheckError(err)
|
|
|
|
if a.sig != nil {
|
|
<-a.doneCh
|
|
}
|
|
}
|
|
|
|
// CreateGRPC creates new configured grpc server
|
|
func (a *ArgoCDCMPServer) CreateGRPC() (*grpc.Server, error) {
|
|
server := grpc.NewServer(a.opts...)
|
|
versionpkg.RegisterVersionServiceServer(server, version.NewServer(nil, func() (bool, error) {
|
|
return true, nil
|
|
}))
|
|
pluginService := plugin.NewService(a.initConstants)
|
|
err := pluginService.Init(common.GetCMPWorkDir())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error initializing plugin service: %w", err)
|
|
}
|
|
apiclient.RegisterConfigManagementPluginServiceServer(server, pluginService)
|
|
|
|
healthService := health.NewServer()
|
|
grpc_health_v1.RegisterHealthServer(server, healthService)
|
|
|
|
// Register reflection service on gRPC server.
|
|
reflection.Register(server)
|
|
|
|
return server, nil
|
|
}
|
|
|
|
func (a *ArgoCDCMPServer) Shutdown(address string) {
|
|
defer signal.Stop(a.stopCh)
|
|
a.sig = <-a.stopCh
|
|
_ = os.Remove(address)
|
|
close(a.doneCh)
|
|
}
|