121 lines
2.3 KiB
Go
121 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"filer/internal/handler"
|
|
"filer/pkg/auxtool/auxgin"
|
|
"filer/pkg/logger"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
const (
|
|
httpTimeout = 360
|
|
)
|
|
|
|
type ServiceConfig struct {
|
|
Handler *handler.Handler
|
|
PortNum int
|
|
Hostname string
|
|
X509Cert []byte
|
|
X509Key []byte
|
|
}
|
|
|
|
type Service struct {
|
|
hand *handler.Handler
|
|
hsrv *http.Server
|
|
log *logger.Logger
|
|
engine *gin.Engine
|
|
|
|
portnum int
|
|
hostname string
|
|
x509Cert []byte
|
|
x509Key []byte
|
|
}
|
|
|
|
func NewService(conf *ServiceConfig) (*Service, error) {
|
|
var err error
|
|
svc := &Service{
|
|
hand: conf.Handler,
|
|
portnum: conf.PortNum,
|
|
hostname: conf.Hostname,
|
|
x509Cert: conf.X509Cert,
|
|
x509Key: conf.X509Key,
|
|
}
|
|
svc.log = logger.NewLogger("service")
|
|
return svc, err
|
|
}
|
|
|
|
func (svc *Service) Build() error {
|
|
var err error
|
|
svc.log.Debugf("Build service")
|
|
|
|
gin.SetMode(gin.ReleaseMode)
|
|
gin.DisableConsoleColor()
|
|
|
|
svc.engine = gin.New()
|
|
svc.engine.Use(gin.Recovery())
|
|
svc.engine.Use(auxgin.CorsMiddleware())
|
|
svc.engine.Use(auxgin.LogMiddleware())
|
|
|
|
svc.engine.GET("/*any", svc.hand.GetFile)
|
|
|
|
noRouteFunc := func(gctx *gin.Context) {
|
|
gctx.Status(http.StatusNotFound)
|
|
return
|
|
}
|
|
svc.engine.NoRoute(noRouteFunc)
|
|
|
|
tlsCert, err := tls.X509KeyPair(svc.x509Cert, svc.x509Key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tlsConfig := tls.Config{
|
|
Certificates: []tls.Certificate{tlsCert},
|
|
ClientAuth: tls.NoClientCert,
|
|
InsecureSkipVerify: true,
|
|
}
|
|
listenAddr := fmt.Sprintf(":%d", svc.portnum)
|
|
svc.hsrv = &http.Server{
|
|
Addr: listenAddr,
|
|
Handler: svc.engine,
|
|
TLSConfig: &tlsConfig,
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (svc *Service) Router() *gin.Engine {
|
|
return svc.engine
|
|
}
|
|
|
|
func (svc *Service) Handler() *handler.Handler {
|
|
return svc.hand
|
|
}
|
|
|
|
func (svc *Service) Run() error {
|
|
var err error
|
|
for _, route := range svc.engine.Routes() {
|
|
svc.log.Debugf("The route is registered: %s %s", route.Method, route.Path)
|
|
}
|
|
svc.log.Infof("Service listening at %d port", svc.portnum)
|
|
err = svc.hsrv.ListenAndServeTLS("", "")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (svc *Service) Stop() {
|
|
svc.log.Infof("Stopping service")
|
|
if svc.hsrv != nil {
|
|
downWaiting := 5 * time.Second
|
|
ctx, _ := context.WithTimeout(context.Background(), downWaiting)
|
|
svc.hsrv.Shutdown(ctx)
|
|
}
|
|
}
|