From 0607c4d6062baad9b290489aaf7d6ee7376b4796 Mon Sep 17 00:00:00 2001
From: DanB <danb@cgrates.org>
Date: Tue, 19 Jan 2016 18:57:17 +0100
Subject: [PATCH] Diameter Granted-Service-AVP not in terminate requests

---
 agents/dmtagent.go | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/agents/dmtagent.go b/agents/dmtagent.go
index 8477dc4eb..db26b69b7 100644
--- a/agents/dmtagent.go
+++ b/agents/dmtagent.go
@@ -140,10 +140,12 @@ func (self DiameterAgent) processCCR(ccr *CCR, reqProcessor *config.DARequestPro
 				return nil
 			}
 		}
-		if err := messageSetAVPsWithPath(cca.diamMessage, []interface{}{"Granted-Service-Unit", "CC-Time"}, strconv.FormatFloat(maxUsage, 'f', 0, 64),
-			false, self.cgrCfg.DiameterAgentCfg().Timezone); err != nil {
-			utils.Logger.Err(fmt.Sprintf("<DiameterAgent> Processing message: %+v set CCA Granted-Service-Unit, error: %s", ccr.diamMessage, err))
-			return nil
+		if ccr.CCRequestType != 3 { // For terminate, we don't add granted-service-unit AVP
+			if err := messageSetAVPsWithPath(cca.diamMessage, []interface{}{"Granted-Service-Unit", "CC-Time"}, strconv.FormatFloat(maxUsage, 'f', 0, 64),
+				false, self.cgrCfg.DiameterAgentCfg().Timezone); err != nil {
+				utils.Logger.Err(fmt.Sprintf("<DiameterAgent> Processing message: %+v set CCA Granted-Service-Unit, error: %s", ccr.diamMessage, err))
+				return nil
+			}
 		}
 	}
 	if err := cca.SetProcessorAVPs(reqProcessor, maxUsage); err != nil {