com.mobius.software.telco.protocols.diameter.impl.primitives.DiameterGroupedAvpImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of diameter-impl Show documentation

There is a newer version: 10.0.0-18-java11

package com.mobius.software.telco.protocols.diameter.impl.primitives;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import com.mobius.software.telco.protocols.diameter.primitives.DiameterAvpKey;
import com.mobius.software.telco.protocols.diameter.primitives.DiameterGroupedAvp;
import com.mobius.software.telco.protocols.diameter.primitives.DiameterUnknownAvp;

import io.netty.buffer.ByteBuf;

/*
 * Mobius Software LTD, Open Source Cloud Communications
 * Copyright 2023, Mobius Software LTD and individual contributors
 * by the @authors tag.
 *
 * This program is free software: you can redistribute it and/or modify
 * under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see 
 */

/**
*
* @author yulian oifa
*
*/
public abstract class DiameterGroupedAvpImpl extends DiameterAvpImpl implements DiameterGroupedAvp
{
	public Map> optionalAvps=new ConcurrentHashMap>();
	
	protected DiameterGroupedAvpImpl() 
	{
	}
	
	@Override
	public Map> getOptionalAvps() 
	{
		return optionalAvps;
	}

	@Override
	public List getOptionalAvps(DiameterAvpKey avpKey) 
	{
		return optionalAvps.get(avpKey);
	}

	@Override
	public void addOptionalAvp(DiameterAvpKey avpKey, DiameterUnknownAvp avp) 
	{
		List currList = optionalAvps.get(avpKey);
		if(currList==null)
		{
			currList=new ArrayList();
			List oldAvps=optionalAvps.putIfAbsent(avpKey, currList);
			if(oldAvps!=null)
				currList = oldAvps;
		}
		
		currList.add(avp);
	}

	@Override
	public void addOptionalAvp(DiameterAvpKey avpKey, ByteBuf avpData, Boolean isProtected) 
	{
		List currList = optionalAvps.get(avpKey);
		if(currList==null)
		{
			currList=new ArrayList();
			List oldAvps=optionalAvps.putIfAbsent(avpKey, currList);
			if(oldAvps!=null)
				currList = oldAvps;
		}
		
		DiameterUnknownAvpImpl octetStringAvp=new DiameterUnknownAvpImpl(avpKey.getVendorID(),avpKey.getAvpID(),avpData);
		octetStringAvp.setProtected(isProtected);
		currList.add(octetStringAvp);
	}
	
	@Override
	public void setOptionalAvps(Map> avps)
	{
		this.optionalAvps = avps;
	}
}