mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-15 21:29:52 +05:00
0c91933065
Add ees_success_ids and ees_failed_ids fields in reader config. The former will be used to set EeIDs when the event processing returns no error, while the latter will be used otherwise. Add config sanity checks for the added options. Remove Processed opts and everything related to them since they should not be used anymore. Fixed test compilation errors caused by the change.
124 lines
3.2 KiB
Go
124 lines
3.2 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 services
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
"github.com/cgrates/cgrates/ers"
|
|
"github.com/cgrates/cgrates/servmanager"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
// NewEventReaderService returns the EventReader Service
|
|
func NewEventReaderService(cfg *config.CGRConfig, filterSChan chan *engine.FilterS,
|
|
shdChan *utils.SyncedChan, connMgr *engine.ConnManager,
|
|
srvDep map[string]*sync.WaitGroup) servmanager.Service {
|
|
return &EventReaderService{
|
|
rldChan: make(chan struct{}, 1),
|
|
cfg: cfg,
|
|
filterSChan: filterSChan,
|
|
shdChan: shdChan,
|
|
connMgr: connMgr,
|
|
srvDep: srvDep,
|
|
}
|
|
}
|
|
|
|
// EventReaderService implements Service interface
|
|
type EventReaderService struct {
|
|
sync.RWMutex
|
|
cfg *config.CGRConfig
|
|
filterSChan chan *engine.FilterS
|
|
shdChan *utils.SyncedChan
|
|
|
|
ers *ers.ERService
|
|
rldChan chan struct{}
|
|
stopChan chan struct{}
|
|
connMgr *engine.ConnManager
|
|
srvDep map[string]*sync.WaitGroup
|
|
}
|
|
|
|
// Start should handle the sercive start
|
|
func (erS *EventReaderService) Start() (err error) {
|
|
if erS.IsRunning() {
|
|
return utils.ErrServiceAlreadyRunning
|
|
}
|
|
|
|
erS.Lock()
|
|
defer erS.Unlock()
|
|
|
|
filterS := <-erS.filterSChan
|
|
erS.filterSChan <- filterS
|
|
|
|
// remake the stop chan
|
|
erS.stopChan = make(chan struct{})
|
|
|
|
utils.Logger.Info(fmt.Sprintf("<%s> starting <%s> subsystem", utils.CoreS, utils.ERs))
|
|
|
|
// build the service
|
|
erS.ers = ers.NewERService(erS.cfg, filterS, erS.connMgr)
|
|
go erS.listenAndServe(erS.ers, erS.stopChan, erS.rldChan)
|
|
return
|
|
}
|
|
|
|
func (erS *EventReaderService) listenAndServe(ers *ers.ERService, stopChan chan struct{}, rldChan chan struct{}) (err error) {
|
|
if err = ers.ListenAndServe(stopChan, rldChan); err != nil {
|
|
utils.Logger.Err(fmt.Sprintf("<%s> error: <%v>", utils.ERs, err))
|
|
erS.shdChan.CloseOnce()
|
|
}
|
|
return
|
|
}
|
|
|
|
// Reload handles the change of config
|
|
func (erS *EventReaderService) Reload() (err error) {
|
|
erS.RLock()
|
|
erS.rldChan <- struct{}{}
|
|
erS.RUnlock()
|
|
return
|
|
}
|
|
|
|
// Shutdown stops the service
|
|
func (erS *EventReaderService) Shutdown() (err error) {
|
|
erS.Lock()
|
|
close(erS.stopChan)
|
|
erS.ers = nil
|
|
erS.Unlock()
|
|
return
|
|
}
|
|
|
|
// IsRunning returns if the service is running
|
|
func (erS *EventReaderService) IsRunning() bool {
|
|
erS.RLock()
|
|
defer erS.RUnlock()
|
|
return erS.ers != nil
|
|
}
|
|
|
|
// ServiceName returns the service name
|
|
func (erS *EventReaderService) ServiceName() string {
|
|
return utils.ERs
|
|
}
|
|
|
|
// ShouldRun returns if the service should be running
|
|
func (erS *EventReaderService) ShouldRun() bool {
|
|
return erS.cfg.ERsCfg().Enabled
|
|
}
|