genofire
/
unified-push-xmpp
2
1
Fork 1
Code Issues 6 Pull Requests 3 Projects Releases Activity

add other part of conn.Settings check

This commit is contained in:
Karmanyaah Malhotra 2021-09-17 09:58:20 -05:00
parent 5e0ec90bf4
commit 656c3237dd
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
distributor/xmpp.go
View File

@ -6,6 +6,7 @@ import (
"encoding/xml"
"errors"
"io"
"strings"
"time"
"github.com/bdlm/log"
@ -112,7 +113,7 @@ func (s *XMPPService) message(msgHead stanza.Message, t xmlstream.TokenReadEncod
return nil
}
from := msgHead.From.String()
if from != conn.Settings {
if settings := strings.Split(conn.Settings, ":"); len(settings) > 1 && settings[0] == from {
log.WithField("from", from).Info("message not from gateway, that is no notification")
return nil
}
@ -178,7 +179,7 @@ func (s *XMPPService) selectGateway() {
if s.KeepGateway {
return
}
conns := s.store.GetUnequalSettings(s.gateway.String() + s.session.LocalAddr().Bare().String())
conns := s.store.GetUnequalSettings(s.gateway.String() + ":" + s.session.LocalAddr().Bare().String())
if len(conns) <= 0 {
return
}
@ -242,7 +243,7 @@ func (s *XMPPService) Register(appID, appToken string) (string, string, error) {
"appID": appID,
"appToken": appToken,
})
conn := s.store.NewConnection(appID, appToken, s.gateway.String()+s.session.LocalAddr().Bare().String())
conn := s.store.NewConnection(appID, appToken, s.gateway.String()+":"+s.session.LocalAddr().Bare().String())
if conn == nil {
errStr := "error to store public token"
err := errors.New(errStr)