Examples with UsageRecord org.killbill.billing.usage.api.UsageRecord used on opensource projects

Example 1 with UsageRecord

use of org.killbill.billing.usage.api.UsageRecord in project killbill by killbill.

the class TestConsumableInArrear method setUsage.

private void setUsage(final UUID subscriptionId, final String unitType, final LocalDate startDate, final Long amount, final CallContext context) throws UsageApiException {
    final List<UsageRecord> usageRecords = new ArrayList<UsageRecord>();
    usageRecords.add(new UsageRecord(startDate, amount));
    final List<UnitUsageRecord> unitUsageRecords = new ArrayList<UnitUsageRecord>();
    unitUsageRecords.add(new UnitUsageRecord(unitType, usageRecords));
    final SubscriptionUsageRecord record = new SubscriptionUsageRecord(subscriptionId, UUID.randomUUID().toString(), unitUsageRecords);
    usageUserApi.recordRolledUpUsage(record, context);
}

Example 2 with UsageRecord

use of org.killbill.billing.usage.api.UsageRecord in project killbill by killbill.

the class DefaultUsageUserApi method recordRolledUpUsage.

@Override
public void recordRolledUpUsage(final SubscriptionUsageRecord record, final CallContext callContext) throws UsageApiException {
    final InternalCallContext internalCallContext = internalCallContextFactory.createInternalCallContext(record.getSubscriptionId(), ObjectType.SUBSCRIPTION, callContext);
    // check if we have (at least) one row with the supplied tracking id
    if (!Strings.isNullOrEmpty(record.getTrackingId()) && recordsWithTrackingIdExist(record, internalCallContext)) {
        throw new UsageApiException(ErrorCode.USAGE_RECORD_TRACKING_ID_ALREADY_EXISTS, record.getTrackingId());
    }
    final List<RolledUpUsageModelDao> usages = new ArrayList<RolledUpUsageModelDao>();
    for (final UnitUsageRecord unitUsageRecord : record.getUnitUsageRecord()) {
        for (final UsageRecord usageRecord : unitUsageRecord.getDailyAmount()) {
            usages.add(new RolledUpUsageModelDao(record.getSubscriptionId(), unitUsageRecord.getUnitType(), usageRecord.getDate(), usageRecord.getAmount(), record.getTrackingId()));
        }
    }
    rolledUpUsageDao.record(usages, internalCallContext);
}