Search in sources :

Example 11 with QuotaAccountVO

use of org.apache.cloudstack.quota.vo.QuotaAccountVO in project cloudstack by apache.

the class QuotaServiceImpl method saveQuotaAccount.

@Override
public boolean saveQuotaAccount(final AccountVO account, final BigDecimal aggrUsage, final Date endDate) {
    // update quota_accounts
    QuotaAccountVO quota_account = _quotaAcc.findByIdQuotaAccount(account.getAccountId());
    if (quota_account == null) {
        quota_account = new QuotaAccountVO(account.getAccountId());
        quota_account.setQuotaBalance(aggrUsage);
        quota_account.setQuotaBalanceDate(endDate);
        if (s_logger.isDebugEnabled()) {
            s_logger.debug(quota_account);
        }
        _quotaAcc.persistQuotaAccount(quota_account);
        return true;
    } else {
        quota_account.setQuotaBalance(aggrUsage);
        quota_account.setQuotaBalanceDate(endDate);
        if (s_logger.isDebugEnabled()) {
            s_logger.debug(quota_account);
        }
        return _quotaAcc.updateQuotaAccount(account.getAccountId(), quota_account);
    }
}
Also used : QuotaAccountVO(org.apache.cloudstack.quota.vo.QuotaAccountVO)

Example 12 with QuotaAccountVO

use of org.apache.cloudstack.quota.vo.QuotaAccountVO in project cloudstack by apache.

the class QuotaResponseBuilderImpl method createQuotaSummaryResponse.

@Override
public List<QuotaSummaryResponse> createQuotaSummaryResponse(Boolean listAll) {
    List<QuotaSummaryResponse> result = new ArrayList<QuotaSummaryResponse>();
    if (listAll) {
        for (final AccountVO account : _accountDao.listAll()) {
            QuotaSummaryResponse qr = getQuotaSummaryResponse(account);
            result.add(qr);
        }
    } else {
        for (final QuotaAccountVO quotaAccount : _quotaAccountDao.listAllQuotaAccount()) {
            AccountVO account = _accountDao.findById(quotaAccount.getId());
            if (account == null)
                continue;
            QuotaSummaryResponse qr = getQuotaSummaryResponse(account);
            result.add(qr);
        }
    }
    return result;
}
Also used : ArrayList(java.util.ArrayList) QuotaAccountVO(org.apache.cloudstack.quota.vo.QuotaAccountVO) QuotaAccountVO(org.apache.cloudstack.quota.vo.QuotaAccountVO) AccountVO(com.cloud.user.AccountVO)

Aggregations

QuotaAccountVO (org.apache.cloudstack.quota.vo.QuotaAccountVO)12 AccountVO (com.cloud.user.AccountVO)6 BigDecimal (java.math.BigDecimal)6 ArrayList (java.util.ArrayList)6 Date (java.util.Date)5 Test (org.junit.Test)5 Calendar (java.util.Calendar)2 DomainVO (com.cloud.domain.DomainVO)1 UserVO (com.cloud.user.UserVO)1 DeferredQuotaEmail (org.apache.cloudstack.quota.QuotaAlertManagerImpl.DeferredQuotaEmail)1 QuotaEmailTemplatesVO (org.apache.cloudstack.quota.vo.QuotaEmailTemplatesVO)1