-
Notifications
You must be signed in to change notification settings - Fork 2
/
multi-server.go
81 lines (61 loc) · 1.98 KB
/
multi-server.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
package multiserver
import (
"context"
"net/http"
"github.com/americanas-go/ignite/gofiber/fiber.v2"
"github.com/americanas-go/log"
"github.com/americanas-go/multiserver"
f "github.com/gofiber/fiber/v2"
)
// Register registers a new multiserver plugin for fiber.
func Register(ctx context.Context, options *fiber.Options) (fiber.ConfigPlugin, fiber.AppPlugin) {
l := NewMultiServer()
return l.Register(ctx, options)
}
// MultiServer represents multiserver plugin for fiber.
type MultiServer struct {
options *Options
}
// NewMultiServerWithOptions returns a new multiserver plugin with options.
func NewMultiServerWithOptions(options *Options) *MultiServer {
return &MultiServer{options: options}
}
// NewMultiServerWithConfigPath returns a new multiserver plugin with options from config path.
func NewMultiServerWithConfigPath(path string) (*MultiServer, error) {
o, err := NewOptionsWithPath(path)
if err != nil {
return nil, err
}
return NewMultiServerWithOptions(o), nil
}
// NewMultiServer returns a new multiserver plugin with default options.
func NewMultiServer() *MultiServer {
o, err := NewOptions()
if err != nil {
log.Fatalf(err.Error())
}
return NewMultiServerWithOptions(o)
}
// Register registers this multiserver plugin for fiber.
func (i *MultiServer) Register(ctx context.Context, options *fiber.Options) (fiber.ConfigPlugin, fiber.AppPlugin) {
if !i.options.Enabled {
return nil, nil
}
logger := log.FromContext(ctx)
checkRoute := i.options.Route
logger.Tracef("configuring multi server check router on %s in fiber", checkRoute)
return nil, func(ctx context.Context, app *f.App) error {
app.Get(checkRoute, handler)
logger.Debugf("multi server check router configured on %s in fiber", checkRoute)
return nil
}
}
func handler(c *f.Ctx) error {
status := http.StatusOK
msg := "OK"
if err := multiserver.Check(c.Context()); err != nil {
status = http.StatusServiceUnavailable
msg = "Service Unavailable"
}
return c.Status(status).SendString(msg)
}