mirror of https://github.com/apache/cloudstack.git
98 lines
3.8 KiB
Java
98 lines
3.8 KiB
Java
/**
|
|
* Copyright (C) 2010 Cloud.com, Inc. All rights reserved.
|
|
*
|
|
* This software is licensed under the GNU General Public License v3 or later.
|
|
*
|
|
* It is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
package com.cloud.network.security.dao;
|
|
|
|
import java.util.List;
|
|
|
|
import javax.ejb.Local;
|
|
|
|
import com.cloud.network.security.SecurityGroupVO;
|
|
import com.cloud.utils.db.GenericDaoBase;
|
|
import com.cloud.utils.db.SearchBuilder;
|
|
import com.cloud.utils.db.SearchCriteria;
|
|
|
|
@Local(value={SecurityGroupDao.class})
|
|
public class SecurityGroupDaoImpl extends GenericDaoBase<SecurityGroupVO, Long> implements SecurityGroupDao {
|
|
private SearchBuilder<SecurityGroupVO> AccountIdSearch;
|
|
private SearchBuilder<SecurityGroupVO> AccountIdNameSearch;
|
|
private SearchBuilder<SecurityGroupVO> AccountIdNamesSearch;
|
|
|
|
protected SecurityGroupDaoImpl() {
|
|
AccountIdSearch = createSearchBuilder();
|
|
AccountIdSearch.and("accountId", AccountIdSearch.entity().getAccountId(), SearchCriteria.Op.EQ);
|
|
AccountIdSearch.done();
|
|
|
|
AccountIdNameSearch = createSearchBuilder();
|
|
AccountIdNameSearch.and("accountId", AccountIdNameSearch.entity().getAccountId(), SearchCriteria.Op.EQ);
|
|
AccountIdNameSearch.and("name", AccountIdNameSearch.entity().getName(), SearchCriteria.Op.EQ);
|
|
|
|
AccountIdNamesSearch = createSearchBuilder();
|
|
AccountIdNamesSearch.and("accountId", AccountIdNamesSearch.entity().getAccountId(), SearchCriteria.Op.EQ);
|
|
AccountIdNamesSearch.and("groupNames", AccountIdNamesSearch.entity().getName(), SearchCriteria.Op.IN);
|
|
AccountIdNameSearch.done();
|
|
}
|
|
|
|
@Override
|
|
public List<SecurityGroupVO> listByAccountId(long accountId) {
|
|
SearchCriteria<SecurityGroupVO> sc = AccountIdSearch.create();
|
|
sc.setParameters("accountId", accountId);
|
|
return listBy(sc);
|
|
}
|
|
|
|
@Override
|
|
public boolean isNameInUse(Long accountId, Long domainId, String name) {
|
|
SearchCriteria<SecurityGroupVO> sc = createSearchCriteria();
|
|
sc.addAnd("name", SearchCriteria.Op.EQ, name);
|
|
if (accountId != null) {
|
|
sc.addAnd("accountId", SearchCriteria.Op.EQ, accountId);
|
|
} else {
|
|
sc.addAnd("domainId", SearchCriteria.Op.EQ, domainId);
|
|
sc.addAnd("accountId", SearchCriteria.Op.NULL);
|
|
}
|
|
|
|
List<SecurityGroupVO> securityGroups = listBy(sc);
|
|
return ((securityGroups != null) && !securityGroups.isEmpty());
|
|
}
|
|
|
|
@Override
|
|
public SecurityGroupVO findByAccountAndName(Long accountId, String name) {
|
|
SearchCriteria<SecurityGroupVO> sc = AccountIdNameSearch.create();
|
|
sc.setParameters("accountId", accountId);
|
|
sc.setParameters("name", name);
|
|
|
|
return findOneIncludingRemovedBy(sc);
|
|
}
|
|
|
|
@Override
|
|
public List<SecurityGroupVO> findByAccountAndNames(Long accountId, String... names) {
|
|
SearchCriteria<SecurityGroupVO> sc = AccountIdNamesSearch.create();
|
|
sc.setParameters("accountId", accountId);
|
|
|
|
sc.setParameters("groupNames", (Object [])names);
|
|
|
|
return listBy(sc);
|
|
}
|
|
@Override
|
|
public int removeByAccountId(long accountId) {
|
|
SearchCriteria sc = AccountIdSearch.create();
|
|
sc.setParameters("accountId", accountId);
|
|
return expunge(sc);
|
|
}
|
|
}
|