kubevela/pkg/server/api-server.go

package server

import (
	"context"
	"net/http"
	"time"

	"github.com/cloud-native-application/rudrx/pkg/server/util"

	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

type ApiServer struct {
	server *http.Server
}

func (s *ApiServer) Launch(kubeClient client.Client) {
	s.server = &http.Server{
		Addr:         util.Port,
		Handler:      setupRoute(kubeClient),
		ReadTimeout:  5 * time.Second,
		WriteTimeout: 10 * time.Second,
	}
	s.server.SetKeepAlivesEnabled(true)

	go (func() error {
		err := s.server.ListenAndServe()
		if err != nil && err != http.ErrServerClosed {
			ctrl.Log.Error(err, "failed to start the server")
		}
		return err
	})()
}

func (s *ApiServer) Shutdown(ctx context.Context) error {
	ctrl.Log.Info("sever shutting down")
	return s.server.Shutdown(ctx)
}