-
Notifications
You must be signed in to change notification settings - Fork 9
/
worker.go
85 lines (74 loc) · 2.36 KB
/
worker.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// Copyright 2023 sigma
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package worker
import (
"context"
"net/http"
"os"
"os/signal"
"time"
"github.com/labstack/echo-contrib/echoprometheus"
"github.com/labstack/echo-contrib/pprof"
"github.com/labstack/echo/v4"
"github.com/rs/zerolog/log"
"github.com/go-sigma/sigma/pkg/builder"
"github.com/go-sigma/sigma/pkg/configs"
"github.com/go-sigma/sigma/pkg/consts"
"github.com/go-sigma/sigma/pkg/graceful"
"github.com/go-sigma/sigma/pkg/middlewares"
"github.com/go-sigma/sigma/pkg/modules/workq"
"github.com/go-sigma/sigma/pkg/types/enums"
"github.com/go-sigma/sigma/pkg/utils/ptr"
)
// Worker is the worker initialization
func Worker() error {
config := ptr.To(configs.GetConfiguration())
err := builder.Initialize(config)
if err != nil {
return err
}
err = workq.Initialize(config)
if err != nil {
return err
}
e := echo.New()
e.HideBanner = true
e.HidePort = true
e.Use(echoprometheus.NewMiddleware(consts.AppName))
e.GET("/metrics", echoprometheus.NewHandler())
e.Use(middlewares.Healthz())
if config.Log.Level == enums.LogLevelDebug || config.Log.Level == enums.LogLevelTrace {
pprof.Register(e, consts.PprofPath)
}
go func() {
log.Info().Str("addr", consts.WorkerPort).Msg("Server listening")
err = e.Start(consts.WorkerPort)
if err != http.ErrServerClosed {
log.Fatal().Err(err).Msg("Listening on interface failed")
}
}()
// Wait for interrupt signal to gracefully shutdown the server with a timeout of 10 seconds.
// Use a buffered channel to avoid missing signals as recommended for signal.Notify
quit := make(chan os.Signal, 1)
signal.Notify(quit, os.Interrupt)
<-quit
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
err = e.Shutdown(ctx)
if err != nil {
log.Error().Err(err).Msg("Server shutdown failed")
}
graceful.Shutdown()
return nil
}