mirror of
https://github.com/k8sgpt-ai/k8sgpt.git
synced 2025-07-06 03:57:05 +00:00
feat: enable rest/http support
Signed-off-by: Sahil Badla <sahil_badla@intuit.com>
This commit is contained in:
parent
5ea1b35cd5
commit
c2e01ea9b6
@ -33,7 +33,7 @@ var (
|
||||
port string
|
||||
metricsPort string
|
||||
backend string
|
||||
enableRest bool
|
||||
enableHttp bool
|
||||
)
|
||||
|
||||
var ServeCmd = &cobra.Command{
|
||||
@ -127,7 +127,7 @@ var ServeCmd = &cobra.Command{
|
||||
Backend: aiProvider.Name,
|
||||
Port: port,
|
||||
MetricsPort: metricsPort,
|
||||
EnableRest: enableRest,
|
||||
EnableHttp: enableHttp,
|
||||
Token: aiProvider.Password,
|
||||
Logger: logger,
|
||||
}
|
||||
@ -155,5 +155,5 @@ func init() {
|
||||
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(&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")
|
||||
}
|
||||
|
@ -49,7 +49,7 @@ type Config struct {
|
||||
Logger *zap.Logger
|
||||
metricsServer *http.Server
|
||||
listener net.Listener
|
||||
EnableRest bool
|
||||
EnableHttp bool
|
||||
}
|
||||
|
||||
type Health struct {
|
||||
@ -97,8 +97,8 @@ func (s *Config) Serve() error {
|
||||
reflection.Register(grpcServer)
|
||||
rpc.RegisterServerServiceServer(grpcServer, s.Handler)
|
||||
|
||||
if s.EnableRest {
|
||||
s.Logger.Info("enabling rest api")
|
||||
if s.EnableHttp {
|
||||
s.Logger.Info("enabling rest/http api")
|
||||
gwmux := runtime.NewServeMux()
|
||||
err = rpc.RegisterServerServiceHandlerFromEndpoint(context.Background(), gwmux, fmt.Sprintf("localhost:%s", s.Port), []grpc.DialOption{grpc.WithTransportCredentials(insecure.NewCredentials())})
|
||||
if err != nil {
|
||||
|
Loading…
Reference in New Issue
Block a user