-
-
Notifications
You must be signed in to change notification settings - Fork 9
/
fx.go
57 lines (50 loc) · 1.49 KB
/
fx.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
// This file is part of the Smart Home
// Program complex distribution https://github.com/e154/smart-home
// Copyright (C) 2016-2021, Filippov Alex
//
// This library is free software: you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 3 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Library General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library. If not, see
// <https://www.gnu.org/licenses/>.
package common
import (
"context"
"fmt"
"os"
"os/signal"
"syscall"
"time"
"go.uber.org/fx"
)
// Start ...
func Start(app *fx.App) {
startCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
if err := app.Start(startCtx); err != nil {
fmt.Println("fx error:", err.Error())
return
}
}
// Work ...
func Work() {
var gracefulStop = make(chan os.Signal)
signal.Notify(gracefulStop, syscall.SIGINT, syscall.SIGTERM)
<-gracefulStop
}
// Stop ...
func Stop(app *fx.App) {
stopCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
if err := app.Stop(stopCtx); err != nil {
fmt.Println("fx error:", err.Error())
}
}