mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
Improve concurrent request mechanish
This commit is contained in:
@@ -66,435 +66,249 @@ func (ssv1 *SessionSv1) Handlers() map[string]interface{} {
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1AuthorizeEvent(clnt *rpc2.Client, args *sessions.V1AuthorizeArgs,
|
||||
rply *sessions.V1AuthorizeReply) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1AuthorizeReply) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1AuthorizeEvent(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1AuthorizeEventWithDigest(clnt *rpc2.Client, args *sessions.V1AuthorizeArgs,
|
||||
rply *sessions.V1AuthorizeReplyWithDigest) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1AuthorizeReplyWithDigest) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1AuthorizeEventWithDigest(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1InitiateSession(clnt *rpc2.Client, args *sessions.V1InitSessionArgs,
|
||||
rply *sessions.V1InitSessionReply) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1InitSessionReply) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1InitiateSession(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1InitiateSessionWithDigest(clnt *rpc2.Client, args *sessions.V1InitSessionArgs,
|
||||
rply *sessions.V1InitReplyWithDigest) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1InitReplyWithDigest) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1InitiateSessionWithDigest(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1UpdateSession(clnt *rpc2.Client, args *sessions.V1UpdateSessionArgs,
|
||||
rply *sessions.V1UpdateSessionReply) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1UpdateSessionReply) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1UpdateSession(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1SyncSessions(clnt *rpc2.Client, args *utils.TenantWithArgDispatcher,
|
||||
rply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1SyncSessions(clnt, &utils.TenantWithArgDispatcher{}, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1TerminateSession(clnt *rpc2.Client, args *sessions.V1TerminateSessionArgs,
|
||||
rply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1TerminateSession(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1ProcessCDR(clnt *rpc2.Client, cgrEv *utils.CGREventWithArgDispatcher, rply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
func (ssv1 *SessionSv1) BiRPCv1ProcessCDR(clnt *rpc2.Client, cgrEv *utils.CGREventWithArgDispatcher,
|
||||
rply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1ProcessCDR(clnt, cgrEv, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1ProcessMessage(clnt *rpc2.Client, args *sessions.V1ProcessMessageArgs,
|
||||
rply *sessions.V1ProcessMessageReply) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1ProcessMessageReply) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1ProcessMessage(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1ProcessEvent(clnt *rpc2.Client, args *sessions.V1ProcessEventArgs,
|
||||
rply *sessions.V1ProcessEventReply) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1ProcessEventReply) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1ProcessEvent(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1GetCost(clnt *rpc2.Client, args *sessions.V1ProcessEventArgs,
|
||||
rply *sessions.V1GetCostReply) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *sessions.V1GetCostReply) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1GetCost(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1GetActiveSessions(clnt *rpc2.Client, args *utils.SessionFilter,
|
||||
rply *[]*sessions.ExternalSession) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *[]*sessions.ExternalSession) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1GetActiveSessions(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1GetActiveSessionsCount(clnt *rpc2.Client, args *utils.SessionFilter,
|
||||
rply *int) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *int) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1GetActiveSessionsCount(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1GetPassiveSessions(clnt *rpc2.Client, args *utils.SessionFilter,
|
||||
rply *[]*sessions.ExternalSession) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *[]*sessions.ExternalSession) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1GetPassiveSessions(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1GetPassiveSessionsCount(clnt *rpc2.Client, args *utils.SessionFilter,
|
||||
rply *int) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *int) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1GetPassiveSessionsCount(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1ForceDisconnect(clnt *rpc2.Client, args *utils.SessionFilter,
|
||||
rply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1ForceDisconnect(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1RegisterInternalBiJSONConn(clnt *rpc2.Client, args string,
|
||||
rply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
rply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1RegisterInternalBiJSONConn(clnt, args, rply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCPing(clnt *rpc2.Client, ign *utils.CGREventWithArgDispatcher, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
func (ssv1 *SessionSv1) BiRPCPing(clnt *rpc2.Client, ign *utils.CGREventWithArgDispatcher,
|
||||
reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ping(ign, reply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1ReplicateSessions(clnt *rpc2.Client,
|
||||
args sessions.ArgsReplicateSessions, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args sessions.ArgsReplicateSessions, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.BiRPCv1ReplicateSessions(clnt, args, reply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1SetPassiveSession(clnt *rpc2.Client,
|
||||
args *sessions.Session, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *sessions.Session, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1SetPassiveSession(clnt, args, reply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1ActivateSessions(clnt *rpc2.Client,
|
||||
args *utils.SessionIDsWithArgsDispatcher, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *utils.SessionIDsWithArgsDispatcher, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1ActivateSessions(clnt, args, reply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCv1DeactivateSessions(clnt *rpc2.Client,
|
||||
args *utils.SessionIDsWithArgsDispatcher, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *utils.SessionIDsWithArgsDispatcher, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1DeactivateSessions(clnt, args, reply)
|
||||
}
|
||||
|
||||
// BiRPCV1ReAuthorize sends the RAR for filterd sessions
|
||||
func (ssv1 *SessionSv1) BiRPCV1ReAuthorize(clnt *rpc2.Client,
|
||||
args *utils.SessionFilter, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *utils.SessionFilter, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1ReAuthorize(clnt, args, reply)
|
||||
}
|
||||
|
||||
// BiRPCV1DisconnectPeer sends the DPR for the OriginHost and OriginRealm
|
||||
func (ssv1 *SessionSv1) BiRPCV1DisconnectPeer(clnt *rpc2.Client,
|
||||
args *utils.DPRArgs, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *utils.DPRArgs, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1DisconnectPeer(clnt, args, reply)
|
||||
}
|
||||
|
||||
// BiRPCV1STIRAuthenticate checks the identity using STIR/SHAKEN
|
||||
func (ssv1 *SessionSv1) BiRPCV1STIRAuthenticate(clnt *rpc2.Client,
|
||||
args *sessions.V1STIRAuthenticateArgs, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *sessions.V1STIRAuthenticateArgs, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1STIRAuthenticate(clnt, args, reply)
|
||||
}
|
||||
|
||||
// BiRPCV1STIRIdentity creates the identity for STIR/SHAKEN
|
||||
func (ssv1 *SessionSv1) BiRPCV1STIRIdentity(clnt *rpc2.Client,
|
||||
args *sessions.V1STIRIdentityArgs, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
args *sessions.V1STIRIdentityArgs, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return ssv1.Ss.BiRPCv1STIRIdentity(nil, args, reply)
|
||||
}
|
||||
|
||||
func (ssv1 *SessionSv1) BiRPCV1Sleep(clnt *rpc2.Client, arg *DurationArgs, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
func (ssv1 *SessionSv1) BiRPCV1Sleep(clnt *rpc2.Client, arg *DurationArgs,
|
||||
reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
time.Sleep(arg.DurationTime)
|
||||
*reply = utils.OK
|
||||
return nil
|
||||
|
||||
@@ -36,85 +36,50 @@ func (smgv1 *SMGenericV1) Handlers() map[string]interface{} {
|
||||
|
||||
/// Returns MaxUsage (for calls in seconds), -1 for no limit
|
||||
func (smgv1 *SMGenericV1) BiRPCV1GetMaxUsage(clnt *rpc2.Client,
|
||||
ev map[string]interface{}, maxUsage *float64) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
ev map[string]interface{}, maxUsage *float64) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return smgv1.Ss.BiRPCV1GetMaxUsage(clnt, ev, maxUsage)
|
||||
}
|
||||
|
||||
// Called on session start, returns the maximum number of seconds the session can last
|
||||
func (smgv1 *SMGenericV1) BiRPCV1InitiateSession(clnt *rpc2.Client,
|
||||
ev map[string]interface{}, maxUsage *float64) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
ev map[string]interface{}, maxUsage *float64) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return smgv1.Ss.BiRPCV1InitiateSession(clnt, ev, maxUsage)
|
||||
}
|
||||
|
||||
// Interim updates, returns remaining duration from the rater
|
||||
func (smgv1 *SMGenericV1) BiRPCV1UpdateSession(clnt *rpc2.Client,
|
||||
ev map[string]interface{}, maxUsage *float64) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
ev map[string]interface{}, maxUsage *float64) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return smgv1.Ss.BiRPCV1UpdateSession(clnt, ev, maxUsage)
|
||||
}
|
||||
|
||||
// Called on session end, should stop debit loop
|
||||
func (smgv1 *SMGenericV1) BiRPCV1TerminateSession(clnt *rpc2.Client,
|
||||
ev map[string]interface{}, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
ev map[string]interface{}, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return smgv1.Ss.BiRPCV1TerminateSession(clnt, ev, reply)
|
||||
}
|
||||
|
||||
// Called on session end, should send the CDR to CDRS
|
||||
func (smgv1 *SMGenericV1) BiRPCV1ProcessCDR(clnt *rpc2.Client,
|
||||
ev map[string]interface{}, reply *string) error {
|
||||
err := utils.ConReqs.Allocate()
|
||||
if err != nil {
|
||||
return err
|
||||
ev map[string]interface{}, reply *string) (err error) {
|
||||
if err = utils.ConReqs.Allocate(); err != nil {
|
||||
return
|
||||
}
|
||||
defer func() {
|
||||
if err != nil {
|
||||
utils.ConReqs.Deallocate(err.Error())
|
||||
} else {
|
||||
utils.ConReqs.Deallocate(utils.EmptyString)
|
||||
}
|
||||
}()
|
||||
defer utils.ConReqs.Deallocate()
|
||||
return smgv1.Ss.BiRPCV1ProcessCDR(clnt, ev, reply)
|
||||
}
|
||||
|
||||
@@ -25,16 +25,16 @@ import (
|
||||
var ConReqs *ConcReqs
|
||||
|
||||
type ConcReqs struct {
|
||||
aReqs chan struct{}
|
||||
nAReqs int
|
||||
limit int
|
||||
strategy string
|
||||
aReqs chan struct{}
|
||||
}
|
||||
|
||||
func NewConReqs(reqs int, strategy string) *ConcReqs {
|
||||
cR := &ConcReqs{
|
||||
aReqs: make(chan struct{}, reqs),
|
||||
nAReqs: reqs,
|
||||
limit: reqs,
|
||||
strategy: strategy,
|
||||
aReqs: make(chan struct{}, reqs),
|
||||
}
|
||||
for i := 0; i < reqs; i++ {
|
||||
cR.aReqs <- struct{}{}
|
||||
@@ -45,8 +45,7 @@ func NewConReqs(reqs int, strategy string) *ConcReqs {
|
||||
var errDeny = fmt.Errorf("denying request due to maximum active requests reached")
|
||||
|
||||
func (cR *ConcReqs) Allocate() (err error) {
|
||||
fmt.Println("ENTER IN ALLOCATE ")
|
||||
if cR.nAReqs == 0 {
|
||||
if cR.limit == 0 {
|
||||
return
|
||||
}
|
||||
switch cR.strategy {
|
||||
@@ -61,8 +60,8 @@ func (cR *ConcReqs) Allocate() (err error) {
|
||||
return
|
||||
}
|
||||
|
||||
func (cR *ConcReqs) Deallocate(errStr string) {
|
||||
if cR.nAReqs == 0 || errStr == errDeny.Error() { // in case we receive denying request we don't need to put back the slot on channel because we returned error without getting it
|
||||
func (cR *ConcReqs) Deallocate() {
|
||||
if cR.limit == 0 {
|
||||
return
|
||||
}
|
||||
cR.aReqs <- struct{}{}
|
||||
|
||||
@@ -28,11 +28,12 @@ import (
|
||||
)
|
||||
|
||||
type concReqsGobServerCodec struct {
|
||||
rwc io.ReadWriteCloser
|
||||
dec *gob.Decoder
|
||||
enc *gob.Encoder
|
||||
encBuf *bufio.Writer
|
||||
closed bool
|
||||
rwc io.ReadWriteCloser
|
||||
dec *gob.Decoder
|
||||
enc *gob.Encoder
|
||||
encBuf *bufio.Writer
|
||||
closed bool
|
||||
allocated bool // populated if we have allocated a channel for concurrent requests
|
||||
}
|
||||
|
||||
func NewConcReqsGobServerCodec(conn io.ReadWriteCloser) rpc.ServerCodec {
|
||||
@@ -53,11 +54,17 @@ func (c *concReqsGobServerCodec) ReadRequestBody(body interface{}) error {
|
||||
if err := ConReqs.Allocate(); err != nil {
|
||||
return err
|
||||
}
|
||||
c.allocated = true
|
||||
return c.dec.Decode(body)
|
||||
}
|
||||
|
||||
func (c *concReqsGobServerCodec) WriteResponse(r *rpc.Response, body interface{}) (err error) {
|
||||
defer ConReqs.Deallocate(r.Error)
|
||||
if c.allocated {
|
||||
defer func() {
|
||||
ConReqs.Deallocate()
|
||||
c.allocated = false
|
||||
}()
|
||||
}
|
||||
if err = c.enc.Encode(r); err != nil {
|
||||
if c.encBuf.Flush() == nil {
|
||||
// Gob couldn't encode the header. Should not happen, so if it does,
|
||||
|
||||
@@ -44,6 +44,8 @@ type concReqsServerCodec struct {
|
||||
mutex sync.Mutex // protects seq, pending
|
||||
seq uint64
|
||||
pending map[uint64]*json.RawMessage
|
||||
|
||||
allocated bool // populated if we have allocated a channel for concurrent requests
|
||||
}
|
||||
|
||||
// NewConcReqsServerCodec returns a new rpc.ServerCodec using JSON-RPC on conn.
|
||||
@@ -80,6 +82,7 @@ func (c *concReqsServerCodec) ReadRequestBody(x interface{}) error {
|
||||
if err := ConReqs.Allocate(); err != nil {
|
||||
return err
|
||||
}
|
||||
c.allocated = true
|
||||
if x == nil {
|
||||
return nil
|
||||
}
|
||||
@@ -96,7 +99,13 @@ func (c *concReqsServerCodec) ReadRequestBody(x interface{}) error {
|
||||
}
|
||||
|
||||
func (c *concReqsServerCodec) WriteResponse(r *rpc.Response, x interface{}) error {
|
||||
defer ConReqs.Deallocate(r.Error)
|
||||
if c.allocated {
|
||||
defer func() {
|
||||
ConReqs.Deallocate()
|
||||
c.allocated = false
|
||||
}()
|
||||
}
|
||||
|
||||
c.mutex.Lock()
|
||||
b, ok := c.pending[r.Seq]
|
||||
if !ok {
|
||||
|
||||
@@ -51,6 +51,8 @@ type jsonServerCodec struct {
|
||||
mutex sync.Mutex // protects seq, pending
|
||||
seq uint64
|
||||
pending map[uint64]*json.RawMessage
|
||||
|
||||
allocated bool // populated if we have allocated a channel for concurrent requests
|
||||
}
|
||||
|
||||
// NewCustomJSONServerCodec is used only when DispatcherS is active to handle APIer methods generically
|
||||
@@ -112,6 +114,7 @@ func (c *jsonServerCodec) ReadRequestBody(x interface{}) error {
|
||||
if err := ConReqs.Allocate(); err != nil {
|
||||
return err
|
||||
}
|
||||
c.allocated = true
|
||||
if x == nil {
|
||||
return nil
|
||||
}
|
||||
@@ -140,7 +143,12 @@ func (c *jsonServerCodec) ReadRequestBody(x interface{}) error {
|
||||
var null = json.RawMessage([]byte("null"))
|
||||
|
||||
func (c *jsonServerCodec) WriteResponse(r *rpc.Response, x interface{}) error {
|
||||
defer ConReqs.Deallocate(r.Error)
|
||||
if c.allocated {
|
||||
defer func() {
|
||||
ConReqs.Deallocate()
|
||||
c.allocated = false
|
||||
}()
|
||||
}
|
||||
c.mutex.Lock()
|
||||
b, ok := c.pending[r.Seq]
|
||||
if !ok {
|
||||
|
||||
Reference in New Issue
Block a user