Examples with AccountItemTree org.killbill.billing.invoice.tree.AccountItemTree used on opensource projects

Example 1 with AccountItemTree

use of org.killbill.billing.invoice.tree.AccountItemTree in project killbill by killbill.

the class FixedAndRecurringInvoiceItemGenerator method generateItems.

public List<InvoiceItem> generateItems(final ImmutableAccountData account, final UUID invoiceId, final BillingEventSet eventSet, @Nullable final List<Invoice> existingInvoices, final LocalDate targetDate, final Currency targetCurrency, final Map<UUID, SubscriptionFutureNotificationDates> perSubscriptionFutureNotificationDate, final InternalCallContext internalCallContext) throws InvoiceApiException {
    final Multimap<UUID, LocalDate> createdItemsPerDayPerSubscription = LinkedListMultimap.<UUID, LocalDate>create();
    final AccountItemTree accountItemTree = new AccountItemTree(account.getId(), invoiceId);
    if (existingInvoices != null) {
        for (final Invoice invoice : existingInvoices) {
            for (final InvoiceItem item : invoice.getInvoiceItems()) {
                if (// Always include migration invoices, credits, external charges etc.
                item.getSubscriptionId() == null || !eventSet.getSubscriptionIdsWithAutoInvoiceOff().contains(item.getSubscriptionId())) {
                    //don't add items with auto_invoice_off tag
                    accountItemTree.addExistingItem(item);
                    trackInvoiceItemCreatedDay(item, createdItemsPerDayPerSubscription, internalCallContext);
                }
            }
        }
    }
    // Generate list of proposed invoice items based on billing events from junction-- proposed items are ALL items since beginning of time
    final List<InvoiceItem> proposedItems = new ArrayList<InvoiceItem>();
    processRecurringBillingEvents(invoiceId, account.getId(), eventSet, targetDate, targetCurrency, proposedItems, perSubscriptionFutureNotificationDate, existingInvoices, internalCallContext);
    processFixedBillingEvents(invoiceId, account.getId(), eventSet, targetDate, targetCurrency, proposedItems, internalCallContext);
    try {
        accountItemTree.mergeWithProposedItems(proposedItems);
    } catch (final IllegalStateException e) {
        // Proposed items have already been logged
        throw new InvoiceApiException(e, ErrorCode.UNEXPECTED_ERROR, String.format("ILLEGAL INVOICING STATE accountItemTree=%s", accountItemTree.toString()));
    }
    final List<InvoiceItem> resultingItems = accountItemTree.getResultingItemList();
    safetyBounds(resultingItems, createdItemsPerDayPerSubscription, internalCallContext);
    return resultingItems;
}