Files
cgrates/ees/libactions.go
2024-03-13 18:25:06 +01:00

105 lines
3.3 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 ees
import (
"encoding/gob"
"encoding/json"
"net/http"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/utils"
)
func init() {
gob.Register(new(HTTPPosterRequest))
gob.Register(new(sqlPosterRequest))
engine.RegisterActionFunc(utils.MetaHTTPPost, callURL)
engine.RegisterActionFunc(utils.HttpPostAsync, callURLAsync)
engine.RegisterActionFunc(utils.MetaPostEvent, postEvent)
}
func getOneData(ub *engine.Account, extraData any) ([]byte, error) {
switch {
case ub != nil:
return json.Marshal(ub)
case extraData != nil:
return json.Marshal(extraData)
}
return nil, nil
}
func callURL(ub *engine.Account, a *engine.Action, _ engine.Actions, _ *engine.FilterS, extraData any,
_ time.Time, _ engine.ActionConnCfg) error {
body, err := getOneData(ub, extraData)
if err != nil {
return err
}
eeCfg := config.NewEventExporterCfg(a.Id, "", a.ExtraParameters, config.CgrConfig().GeneralCfg().FailedPostsDir,
config.CgrConfig().GeneralCfg().PosterAttempts, nil)
pstr, err := NewHTTPjsonMapEE(eeCfg, config.CgrConfig(), nil, nil)
if err != nil {
return err
}
err = ExportWithAttempts(pstr, &HTTPPosterRequest{Body: body, Header: make(http.Header)}, "")
if config.CgrConfig().GeneralCfg().FailedPostsDir != utils.MetaNone {
err = nil
}
return err
}
// Does not block for posts, no error reports
func callURLAsync(ub *engine.Account, a *engine.Action, _ engine.Actions, _ *engine.FilterS, extraData any,
_ time.Time, _ engine.ActionConnCfg) error {
body, err := getOneData(ub, extraData)
if err != nil {
return err
}
eeCfg := config.NewEventExporterCfg(a.Id, "", a.ExtraParameters, config.CgrConfig().GeneralCfg().FailedPostsDir,
config.CgrConfig().GeneralCfg().PosterAttempts, nil)
pstr, err := NewHTTPjsonMapEE(eeCfg, config.CgrConfig(), nil, nil)
if err != nil {
return err
}
go ExportWithAttempts(pstr, &HTTPPosterRequest{Body: body, Header: make(http.Header)}, "")
return nil
}
func postEvent(_ *engine.Account, a *engine.Action, _ engine.Actions, _ *engine.FilterS, extraData any,
_ time.Time, _ engine.ActionConnCfg) error {
body, err := json.Marshal(extraData)
if err != nil {
return err
}
eeCfg := config.NewEventExporterCfg(a.Id, "", a.ExtraParameters, config.CgrConfig().GeneralCfg().FailedPostsDir,
config.CgrConfig().GeneralCfg().PosterAttempts, nil)
pstr, err := NewHTTPjsonMapEE(eeCfg, config.CgrConfig(), nil, nil)
if err != nil {
return err
}
err = ExportWithAttempts(pstr, &HTTPPosterRequest{Body: body, Header: make(http.Header)}, "")
if config.CgrConfig().GeneralCfg().FailedPostsDir != utils.MetaNone {
err = nil
}
return err
}