121 lines
2.3 KiB
Go
121 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"hamlogger/internal/handler"
|
|
"hamlogger/internal/router"
|
|
"hamlogger/pkg/logger"
|
|
)
|
|
|
|
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
|
|
rout *router.Router
|
|
listen net.Listener
|
|
|
|
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("Building service")
|
|
|
|
svc.rout = router.NewRouter()
|
|
svc.rout.Use(router.NewRecoveryMiddleware(svc.log.Errorf))
|
|
svc.rout.Use(router.NewLoggingMiddleware(svc.log.Infof))
|
|
svc.rout.Use(router.NewCorsMiddleware())
|
|
|
|
svc.rout.Get(`/api/v1/service/hello`, svc.hand.GetHello)
|
|
svc.rout.Get(`/`, svc.hand.GetIndex)
|
|
|
|
svc.rout.NotFound(svc.hand.NotFound)
|
|
|
|
routes := svc.rout.ListRoutes()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, route := range routes {
|
|
svc.log.Infof("Registed route: %s %s", route.Method, route.Path)
|
|
}
|
|
const useTLS = true
|
|
if useTLS {
|
|
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,
|
|
}
|
|
|
|
listenAddress := fmt.Sprintf(":%d", svc.portnum)
|
|
svc.listen, err = tls.Listen("tcp", listenAddress, &tlsConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
} else {
|
|
listenAddress := fmt.Sprintf(":%d", svc.portnum)
|
|
svc.listen, err = net.Listen("tcp", listenAddress)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
svc.hsrv = &http.Server{
|
|
Handler: svc.rout,
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (svc *Service) Run() error {
|
|
var err error
|
|
err = svc.hsrv.Serve(svc.listen)
|
|
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)
|
|
}
|
|
}
|