mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-12 18:46:24 +05:00
125 lines
3.4 KiB
Go
125 lines
3.4 KiB
Go
/*
|
|
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
|
Copyright (C) ITsysCOM GmbH
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
|
|
package efs
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/cgrates/birpc/context"
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/guardian"
|
|
"github.com/cgrates/cgrates/utils"
|
|
"github.com/cgrates/ltcache"
|
|
)
|
|
|
|
var failedPostCache *ltcache.Cache
|
|
|
|
func init() {
|
|
failedPostCache = ltcache.NewCache(-1, 5*time.Second, true, writeFailedPosts)
|
|
}
|
|
|
|
// SetFailedPostCacheTTL recreates the failed cache
|
|
func SetFailedPostCacheTTL(ttl time.Duration) {
|
|
failedPostCache = ltcache.NewCache(-1, ttl, true, writeFailedPosts)
|
|
}
|
|
|
|
func writeFailedPosts(_ string, value interface{}) {
|
|
expEv, canConvert := value.(*FailedExportersLogg)
|
|
if !canConvert {
|
|
return
|
|
}
|
|
filePath := expEv.FilePath()
|
|
expEv.lk.RLock()
|
|
if err := WriteToFile(filePath, expEv); err != nil {
|
|
utils.Logger.Warning(fmt.Sprintf("Unable to write failed post to file <%s> because <%s>",
|
|
filePath, err))
|
|
expEv.lk.RUnlock()
|
|
return
|
|
}
|
|
expEv.lk.RUnlock()
|
|
}
|
|
|
|
// FilePath returns the file path it should use for saving the failed events
|
|
func (expEv *FailedExportersLogg) FilePath() string {
|
|
return path.Join(expEv.FailedPostsDir, expEv.Module+utils.PipeSep+utils.UUIDSha1Prefix()+utils.GOBSuffix)
|
|
}
|
|
|
|
type FailoverPoster interface {
|
|
ReplayFailedPosts(*context.Context, int, string) error
|
|
}
|
|
|
|
// WriteToFile writes the events to file
|
|
func WriteToFile(filePath string, expEv FailoverPoster) (err error) {
|
|
fileOut, err := os.Create(filePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
encd := gob.NewEncoder(fileOut)
|
|
gob.Register(new(utils.CGREvent))
|
|
err = encd.Encode(expEv)
|
|
fileOut.Close()
|
|
return
|
|
}
|
|
|
|
// NewFailoverPosterFromFile returns ExportEvents from the file
|
|
// used only on replay failed post
|
|
func NewFailoverPosterFromFile(filePath, providerType string, efs *EfS) (failPoster FailoverPoster, err error) {
|
|
var fileContent []byte
|
|
err = guardian.Guardian.Guard(context.TODO(), func(_ *context.Context) error {
|
|
if fileContent, err = os.ReadFile(filePath); err != nil {
|
|
return err
|
|
}
|
|
return os.Remove(filePath)
|
|
}, config.CgrConfig().GeneralCfg().LockingTimeout, utils.FileLockPrefix+filePath)
|
|
if err != nil {
|
|
return
|
|
}
|
|
dec := gob.NewDecoder(bytes.NewBuffer(fileContent))
|
|
// unmarshall it
|
|
expEv := new(FailedExportersLogg)
|
|
err = dec.Decode(&expEv)
|
|
switch providerType {
|
|
case utils.EEs:
|
|
opts, err := AsOptsEESConfig(expEv.Opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
failPoster = &FailedExportersEEs{
|
|
module: expEv.Module,
|
|
failedPostsDir: expEv.FailedPostsDir,
|
|
Path: expEv.Path,
|
|
Opts: opts,
|
|
Events: expEv.Events,
|
|
Format: expEv.Format,
|
|
|
|
connMngr: efs.connMgr,
|
|
}
|
|
case utils.Kafka:
|
|
expEv.cfg = efs.cfg
|
|
expEv.connMngr = efs.connMgr
|
|
failPoster = expEv
|
|
}
|
|
return
|
|
}
|