GitHub/matterbridge
3
0
Fork 0
mirror of https://github.com/42wim/matterbridge.git synced 2024-12-01 16:59:29 +01:00
Code Issues Projects Releases Wiki Activity

Pass peristent store errors up to the router level

Browse Source
This commit is contained in:
Yousef Mansy 2023-03-11 15:36:17 -08:00
parent eaa8907d17
commit 2845740e97
3 changed files with 27 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
gateway/gateway.go
View File

@ -47,7 +47,7 @@ const apiProtocol = "api"
// New creates a new Gateway object associated with the specified router and
// following the given configuration.
func New(rootLogger *logrus.Logger, cfg *config.Gateway, r *Router) *Gateway {
func New(rootLogger *logrus.Logger, cfg *config.Gateway, r *Router) (*Gateway, error) {
logger := rootLogger.WithFields(logrus.Fields{"prefix": "gateway"})
cache, _ := lru.New(5000)
@ -67,13 +67,25 @@ func New(rootLogger *logrus.Logger, cfg *config.Gateway, r *Router) *Gateway {
persistentMessageStorePath, usePersistent := gw.Config.GetString("PersistentMessageStorePath")
if usePersistent {
rootPath := fmt.Sprintf("%s/%s", persistentMessageStorePath, gw.Name)
os.MkdirAll(rootPath, os.ModePerm)
gw.MessageStore = gw.getMessageMapStore(fmt.Sprintf("%s/Messages", rootPath))
gw.CanonicalStore = gw.getMessageMapStore(fmt.Sprintf("%s/Canonical", rootPath))
err := os.MkdirAll(rootPath, os.ModePerm)
if err != nil {
return nil, err
}
return gw
MessageStore, err := gw.getMessageMapStore(fmt.Sprintf("%s/Messages", rootPath))
if err != nil {
return nil, err
}
gw.MessageStore = MessageStore
CanonicalStore, err := gw.getMessageMapStore(fmt.Sprintf("%s/Canonical", rootPath))
if err != nil {
return nil, err
}
gw.CanonicalStore = CanonicalStore
}
return gw, nil
}
func (gw *Gateway) SetMessageMap(canonicalMsgID string, msgIDs []*BrMsgID) {

7
gateway/persistent.go
View File

@ -8,7 +8,7 @@ import (
"github.com/philippgille/gokv/encoding"
)
func (gw *Gateway) getMessageMapStore(path string) gokv.Store {
func (gw *Gateway) getMessageMapStore(path string) (gokv.Store, error) {
options := badgerdb.Options{
Dir: path,
Codec: encoding.Gob,
@ -16,11 +16,12 @@ func (gw *Gateway) getMessageMapStore(path string) gokv.Store {
store, err := badgerdb.NewStore(options)
if err != nil {
gw.logger.Error(err)
gw.logger.Errorf("Could not connect to db: %s", path)
gw.logger.Error(err)
return nil, err
}
return store
return store, nil
}
func (gw *Gateway) getCanonicalMessageFromStore(messageID string) string {

7
gateway/router.go
View File

@ -50,7 +50,12 @@ func NewRouter(rootLogger *logrus.Logger, cfg config.Config, bridgeMap map[strin
if _, ok := r.Gateways[entry.Name]; ok {
return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name)
}
r.Gateways[entry.Name] = New(rootLogger, entry, r)
gw, err := New(rootLogger, entry, r)
if err != nil {
return nil, err
}
r.Gateways[entry.Name] = gw
}
return r, nil
}