use of org.wso2.carbon.apimgt.throttle.policy.deployer.dto.QuotaPolicy in project carbon-apimgt by wso2.
the class CommonThrottleMappingUtil method fromDTOToQuotaPolicy.
/**
* Converts a Throttle Limit DTO object into a Quota Policy object
*
* @param dto Throttle limit DTO object
* @return Derived Quota policy object from DTO
* @throws UnsupportedThrottleLimitTypeException
*/
public static QuotaPolicy fromDTOToQuotaPolicy(ThrottleLimitDTO dto) throws UnsupportedThrottleLimitTypeException {
String errorMessage;
QuotaPolicy quotaPolicy = new QuotaPolicy();
ThrottleLimitDTO.TypeEnum limitType = dto.getType();
if (limitType != null) {
switch(dto.getType()) {
case REQUESTCOUNTLIMIT:
{
if (dto.getRequestCount() != null) {
quotaPolicy.setLimit(fromDTOToRequestCountLimit(dto.getRequestCount()));
} else {
errorMessage = RestApiAdminUtils.constructMissingThrottleObjectErrorMessage(ThrottleLimitDTO.TypeEnum.REQUESTCOUNTLIMIT) + dto.toString();
throw new UnsupportedThrottleLimitTypeException(errorMessage);
}
break;
}
case BANDWIDTHLIMIT:
{
if (dto.getBandwidth() != null) {
quotaPolicy.setLimit(fromDTOToBandwidthLimit(dto.getBandwidth()));
} else {
errorMessage = RestApiAdminUtils.constructMissingThrottleObjectErrorMessage(ThrottleLimitDTO.TypeEnum.BANDWIDTHLIMIT) + dto.toString();
throw new UnsupportedThrottleLimitTypeException(errorMessage);
}
break;
}
case EVENTCOUNTLIMIT:
{
if (dto.getEventCount() != null) {
quotaPolicy.setLimit(fromDTOToEventCountLimit(dto.getEventCount()));
} else {
errorMessage = RestApiAdminUtils.constructMissingThrottleObjectErrorMessage(ThrottleLimitDTO.TypeEnum.EVENTCOUNTLIMIT) + dto.toString();
throw new UnsupportedThrottleLimitTypeException(errorMessage);
}
break;
}
}
quotaPolicy.setType(mapQuotaPolicyTypeFromDTOToModel(dto.getType()));
} else {
errorMessage = "defaultLimit 'type' property has not been specified\n" + dto.toString();
throw new UnsupportedThrottleLimitTypeException(errorMessage);
}
return quotaPolicy;
}
Aggregations