Hopefully fix xmpp server communication error handling, but fix is kinda dirty

This commit is contained in:
eleksir
2024-12-07 20:12:28 +03:00
parent 073138bde4
commit 21b5f59c1c
4 changed files with 101 additions and 79 deletions

View File

@@ -60,3 +60,12 @@ Cуществует несколько видов пинга:
------------------------------------------------------------------------------------------------------------------------ ------------------------------------------------------------------------------------------------------------------------
Бот не читает историю, когда присоединяется к чатам, поэтому если его какое-то время не было в комнате и кто-то произнёс Бот не читает историю, когда присоединяется к чатам, поэтому если его какое-то время не было в комнате и кто-то произнёс
непотребную фразу, ретроспективного бана не будет. непотребную фразу, ретроспективного бана не будет.
Проблема реконнекта
------------------------------------------------------------------------------------------------------------------------
На текущий момент решена не совсем чисто. Скорее всего, при реконнекте утекают ресурсы, связанные с jabber-соединением.
Проблема в том, что приходится пересоздавать основной "объект" со всеми его полями.
Кроме того, sig handler фактичеки не завершается, потому что в него не приходит ни одного сигнала и формально остаётся
висеть на channel-е навечно.
А ещё есть пингеры каналов и пингеры сервера. Их бы тоже завершать явным образом.

View File

@@ -16,6 +16,7 @@ import (
// main - фактичеcки, начало и основное тело программы. // main - фактичеcки, начало и основное тело программы.
func main() { func main() {
for {
var j = jabber.Jabber{ var j = jabber.Jabber{
SigChan: make(chan os.Signal, 1), SigChan: make(chan os.Signal, 1),
GTomb: tomb.Tomb{}, GTomb: tomb.Tomb{},
@@ -90,14 +91,11 @@ func main() {
// github.com/mattn/go-xmpp пишет в stdio, нам этого не надо, ловим выхлоп его в logrus с уровнем trace. // github.com/mattn/go-xmpp пишет в stdio, нам этого не надо, ловим выхлоп его в logrus с уровнем trace.
xmpp.DebugWriter = log.WithFields(log.Fields{"logger": "stdlib"}).WriterLevel(log.TraceLevel) xmpp.DebugWriter = log.WithFields(log.Fields{"logger": "stdlib"}).WriterLevel(log.TraceLevel)
// Хэндлер сигналов не надо трогать, он нужен для завершения программы целиком. // Хэндлер сигналов.
go j.SigHandler() j.GTomb.Go(func() error { return j.SigHandler() })
signal.Notify(j.SigChan, os.Interrupt) signal.Notify(j.SigChan, os.Interrupt)
// Враппер основной программы. Фактически на каждой серьёзной ошибке, например, сетевой, mainLoop и некоторые
// вспомогательные корутины прекращает свою работу. Здесь мы это ловим и запускаем их снова. Задача в том, чтобы
// запустить все асинхронные процессы заново, с чистого листа, но уже с распарсенным конфигом.
for {
j.Options = &xmpp.Options{ //nolint:exhaustruct j.Options = &xmpp.Options{ //nolint:exhaustruct
Host: fmt.Sprintf("%s:%d", j.C.Jabber.Server, j.C.Jabber.Port), Host: fmt.Sprintf("%s:%d", j.C.Jabber.Server, j.C.Jabber.Port),
User: j.C.Jabber.User, User: j.C.Jabber.User,
@@ -117,17 +115,18 @@ func main() {
DialTimeout: time.Duration(j.C.Jabber.ConnectionTimeout) * time.Second, DialTimeout: time.Duration(j.C.Jabber.ConnectionTimeout) * time.Second,
} }
// Враппер основной программы. Фактически на каждой серьёзной ошибке, например, сетевой, mainLoop и некоторые
// вспомогательные корутины прекращает свою работу. Здесь мы это ловим и запускаем их снова. Задача в том, чтобы
// запустить все асинхронные процессы заново, с чистого листа, но уже с распарсенным конфигом.
// Устанавливаем соединение и гребём события, посылаемые сервером - основной и вспомогательные циклы программы. // Устанавливаем соединение и гребём события, посылаемые сервером - основной и вспомогательные циклы программы.
j.GTomb.Go(func() error { return j.MyLoop() }) j.GTomb.Go(func() error { return j.MyLoop() })
// Логгируем причину завершения mainLoop и вспомогательных циклов программы. // Ловим первый же kill и не дождаемся остальных, хотя формально надо бы.
log.Error(j.GTomb.Wait()) _ = <-j.GTomb.Dying()
// Если у нас wire error, то вызов .Close() повлечёт за собой ошибку, но мы вынуждены звать .Close(), чтоб // Если у нас wire error, то вызов .Close() повлечёт за собой ошибку. А если у нас не wire error, то по ходу мы
// закрыть tls контекст и почистить всё что связанно с прерванным соединением. // получим утечку сокетов.
log.Infoln("Closing connection to jabber server")
_ = j.Talk.Close()
time.Sleep(time.Duration(j.C.Jabber.ReconnectDelay) * time.Second) time.Sleep(time.Duration(j.C.Jabber.ReconnectDelay) * time.Second)
} }

View File

@@ -2,6 +2,8 @@ package jabber
import ( import (
"time" "time"
log "github.com/sirupsen/logrus"
) )
// MyLoop - основной цикл программы. // MyLoop - основной цикл программы.
@@ -23,13 +25,21 @@ func (j *Jabber) MyLoop() error {
// Установим коннект // Установим коннект
if err := j.EstablishConnection(); err != nil { if err := j.EstablishConnection(); err != nil {
log.Error(err)
return err return err
} }
j.ServerPingTimestampRx = time.Now().Unix() // Считаем, что если коннект запустился, то первый пинг успешен. j.ServerPingTimestampRx = time.Now().Unix() // Считаем, что если коннект запустился, то первый пинг успешен.
// Тыкаем сервер палочкой, проверяем, что коннект жив и вываливаемся из mainLoop, если он не жив. // Тыкаем сервер палочкой, проверяем, что коннект жив и вываливаемся из mainLoop, если он не жив.
j.GTomb.Go(func() error { return j.ProbeServerLiveness() }) j.GTomb.Go(
func() error {
err := j.ProbeServerLiveness()
log.Error(err)
return err
},
)
// Тыкаем muc-и палочкой, проверяем, что они живы и вываливаемся из mainLoop, если пинги пропали. // Тыкаем muc-и палочкой, проверяем, что они живы и вываливаемся из mainLoop, если пинги пропали.
// Если пинги до комнаты пропали, то это фактически значит, что либо сервер потерял связь с MUC-компонентом, // Если пинги до комнаты пропали, то это фактически значит, что либо сервер потерял связь с MUC-компонентом,
@@ -56,6 +66,8 @@ func (j *Jabber) MyLoop() error {
chat, err := j.Talk.Recv() chat, err := j.Talk.Recv()
if err != nil { if err != nil {
log.Errorf("Unable to get recieve data from socket: %s", err)
return err return err
} }

View File

@@ -8,7 +8,7 @@ import (
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
) )
func (j *Jabber) SigHandler() { func (j *Jabber) SigHandler() error {
log.Debug("Installing signal handler") log.Debug("Installing signal handler")
for s := range j.SigChan { for s := range j.SigChan {
@@ -55,4 +55,6 @@ func (j *Jabber) SigHandler() {
os.Exit(0) os.Exit(0)
} }
return nil
} }