Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
faeeeea9f5 | ||
|
|
a976425a4f |
22
Makefile
Normal file
22
Makefile
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
#!/usr/bin/env gmake -f
|
||||||
|
|
||||||
|
GOOPTS=CGO_ENABLED=0
|
||||||
|
BUILDOPTS=-ldflags="-s -w" -a -gcflags=all=-l
|
||||||
|
|
||||||
|
all: clean build
|
||||||
|
|
||||||
|
build:
|
||||||
|
${GOOPTS} go build ${BUILDOPTS}
|
||||||
|
|
||||||
|
clean:
|
||||||
|
go clean
|
||||||
|
|
||||||
|
wipe:
|
||||||
|
go clean
|
||||||
|
rm -rf go.{mod,sum}
|
||||||
|
|
||||||
|
prep:
|
||||||
|
go mod init main
|
||||||
|
go mod tidy -compat=1.17
|
||||||
|
|
||||||
|
# vim: set ft=make noet ai ts=4 sw=4 sts=4:
|
||||||
69
main.go
69
main.go
@@ -6,8 +6,10 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
|
"os/signal"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-redis/redis/v8"
|
"github.com/go-redis/redis/v8"
|
||||||
@@ -76,8 +78,18 @@ var config myConfig
|
|||||||
// To break circular message forwarding we must set some sane default, it can be overridden via config
|
// To break circular message forwarding we must set some sane default, it can be overridden via config
|
||||||
var forwardMax int64 = 5
|
var forwardMax int64 = 5
|
||||||
|
|
||||||
// Дефолтный конфиг клиента-редиски
|
// Объектики клиента-редиски
|
||||||
var redisClient *redis.Client
|
var redisClient *redis.Client
|
||||||
|
var subscriber *redis.PubSub
|
||||||
|
|
||||||
|
// Main context
|
||||||
|
var ctx = context.Background()
|
||||||
|
|
||||||
|
// Ставится в true, если мы получили сигнал на выключение
|
||||||
|
var shutdown = false
|
||||||
|
|
||||||
|
// Канал, в который приходят уведомления для хэндлера сигналов от траппера сигналов
|
||||||
|
var sigChan = make(chan os.Signal, 1)
|
||||||
|
|
||||||
// Горутинка, которая парсит json-чики прилетевшие из REDIS-ки
|
// Горутинка, которая парсит json-чики прилетевшие из REDIS-ки
|
||||||
func msgParser(ctx context.Context, msg string) {
|
func msgParser(ctx context.Context, msg string) {
|
||||||
@@ -389,6 +401,41 @@ func readConfig() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Хэндлер сигналов закрывает все бд и сваливает из приложения
|
||||||
|
func sigHandler() {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
for {
|
||||||
|
var s = <-sigChan
|
||||||
|
switch s {
|
||||||
|
case syscall.SIGINT:
|
||||||
|
log.Infoln("Got SIGINT, quitting")
|
||||||
|
case syscall.SIGTERM:
|
||||||
|
log.Infoln("Got SIGTERM, quitting")
|
||||||
|
case syscall.SIGQUIT:
|
||||||
|
log.Infoln("Got SIGQUIT, quitting")
|
||||||
|
|
||||||
|
// Заходим на новую итерацию, если у нас "неинтересный" сигнал
|
||||||
|
default:
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Чтобы не срать в логи ошибками от редиски, проставим shutdown state приложения в true
|
||||||
|
shutdown = true
|
||||||
|
|
||||||
|
// Отпишемся от всех каналов и закроем коннект к редиске
|
||||||
|
if err = subscriber.Unsubscribe(ctx); err != nil {
|
||||||
|
log.Errorf("Unable to unsubscribe from redis channels cleanly: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = subscriber.Close(); err != nil {
|
||||||
|
log.Errorf("Unable to close redis connection cleanly: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Производит некоторую инициализацию перед запуском main()
|
// Производит некоторую инициализацию перед запуском main()
|
||||||
func init() {
|
func init() {
|
||||||
log.SetFormatter(&log.TextFormatter{
|
log.SetFormatter(&log.TextFormatter{
|
||||||
@@ -440,14 +487,30 @@ func main() {
|
|||||||
})
|
})
|
||||||
|
|
||||||
log.Debugf("Lazy connect() to redis at %s:%d", config.Server, config.Port)
|
log.Debugf("Lazy connect() to redis at %s:%d", config.Server, config.Port)
|
||||||
subscriber := redisClient.Subscribe(ctx, config.Channel)
|
subscriber = redisClient.Subscribe(ctx, config.Channel)
|
||||||
|
|
||||||
|
// Самое время поставить траппер сигналов
|
||||||
|
signal.Notify(sigChan,
|
||||||
|
syscall.SIGINT,
|
||||||
|
syscall.SIGTERM,
|
||||||
|
syscall.SIGQUIT)
|
||||||
|
|
||||||
|
go sigHandler()
|
||||||
|
|
||||||
// Обработчик событий от редиски
|
// Обработчик событий от редиски
|
||||||
for {
|
for {
|
||||||
|
if shutdown {
|
||||||
|
time.Sleep(1 * time.Second)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
msg, err := subscriber.ReceiveMessage(ctx)
|
msg, err := subscriber.ReceiveMessage(ctx)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Warnf("Unable to connect to redis at %s:%d: %s", config.Server, config.Port, err)
|
if !shutdown {
|
||||||
|
log.Warnf("Unable to connect to redis at %s:%d: %s", config.Server, config.Port, err)
|
||||||
|
}
|
||||||
|
|
||||||
time.Sleep(1 * time.Second)
|
time.Sleep(1 * time.Second)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user