github/k8sgpt
1
0
Fork 0
mirror of https://github.com/k8sgpt-ai/k8sgpt.git synced 2025-07-06 20:08:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: enable rest/http support

Browse Source 
Signed-off-by: Sahil Badla <sahil_badla@intuit.com>
This commit is contained in:
Sahil Badla 2024-01-16 10:01:48 -08:00
parent 5ea1b35cd5
commit c2e01ea9b6
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cmd/serve/serve.go
View File

@ -33,7 +33,7 @@ var (
port string port string
metricsPort string metricsPort string
backend string backend string
enableRest bool enableHttp bool
) )
var ServeCmd = &cobra.Command{ var ServeCmd = &cobra.Command{
@ -127,7 +127,7 @@ var ServeCmd = &cobra.Command{
Backend: aiProvider.Name, Backend: aiProvider.Name,
Port: port, Port: port,
MetricsPort: metricsPort, MetricsPort: metricsPort,
EnableRest: enableRest, EnableHttp: enableHttp,
Token: aiProvider.Password, Token: aiProvider.Password,
Logger: logger, Logger: logger,
} }
@ -155,5 +155,5 @@ func init() {
ServeCmd.Flags().StringVarP(&port, "port", "p", "8080", "Port to run the server on") ServeCmd.Flags().StringVarP(&port, "port", "p", "8080", "Port to run the server on")
ServeCmd.Flags().StringVarP(&metricsPort, "metrics-port", "", "8081", "Port to run the metrics-server on") ServeCmd.Flags().StringVarP(&metricsPort, "metrics-port", "", "8081", "Port to run the metrics-server on")
ServeCmd.Flags().StringVarP(&backend, "backend", "b", "openai", "Backend AI provider") ServeCmd.Flags().StringVarP(&backend, "backend", "b", "openai", "Backend AI provider")
ServeCmd.Flags().BoolVarP(&enableRest, "enable-REST", "r", false, "Enable REST api using gppc-gateway") ServeCmd.Flags().BoolVarP(&enableHttp, "enable-http", "http", false, "Enable REST/http using gppc-gateway")
} }

6
pkg/server/server.go
View File

@ -49,7 +49,7 @@ type Config struct {
Logger *zap.Logger Logger *zap.Logger
metricsServer *http.Server metricsServer *http.Server
listener net.Listener listener net.Listener
EnableRest bool EnableHttp bool
} }
type Health struct { type Health struct {
@ -97,8 +97,8 @@ func (s *Config) Serve() error {
reflection.Register(grpcServer) reflection.Register(grpcServer)
rpc.RegisterServerServiceServer(grpcServer, s.Handler) rpc.RegisterServerServiceServer(grpcServer, s.Handler)
if s.EnableRest { if s.EnableHttp {
s.Logger.Info("enabling rest api") s.Logger.Info("enabling rest/http api")
gwmux := runtime.NewServeMux() gwmux := runtime.NewServeMux()
err = rpc.RegisterServerServiceHandlerFromEndpoint(context.Background(), gwmux, fmt.Sprintf("localhost:%s", s.Port), []grpc.DialOption{grpc.WithTransportCredentials(insecure.NewCredentials())}) err = rpc.RegisterServerServiceHandlerFromEndpoint(context.Background(), gwmux, fmt.Sprintf("localhost:%s", s.Port), []grpc.DialOption{grpc.WithTransportCredentials(insecure.NewCredentials())})
if err != nil { if err != nil {