| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, |
| // software distributed under the License is distributed on an |
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| // KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations |
| // under the License. |
| package com.cloud.alert.dao; |
| |
| import java.util.Date; |
| import java.util.List; |
| |
| |
| import org.springframework.stereotype.Component; |
| |
| import com.cloud.alert.AlertVO; |
| import com.cloud.utils.db.Filter; |
| import com.cloud.utils.db.GenericDaoBase; |
| import com.cloud.utils.db.SearchBuilder; |
| import com.cloud.utils.db.SearchCriteria; |
| import com.cloud.utils.db.SearchCriteria.Op; |
| import com.cloud.utils.db.TransactionLegacy; |
| |
| @Component |
| public class AlertDaoImpl extends GenericDaoBase<AlertVO, Long> implements AlertDao { |
| |
| protected final SearchBuilder<AlertVO> AlertSearchByIdsAndType; |
| |
| public AlertDaoImpl() { |
| AlertSearchByIdsAndType = createSearchBuilder(); |
| AlertSearchByIdsAndType.and("id", AlertSearchByIdsAndType.entity().getId(), Op.IN); |
| AlertSearchByIdsAndType.and("type", AlertSearchByIdsAndType.entity().getType(), Op.EQ); |
| AlertSearchByIdsAndType.and("createdDateB", AlertSearchByIdsAndType.entity().getCreatedDate(), Op.BETWEEN); |
| AlertSearchByIdsAndType.and("createdDateL", AlertSearchByIdsAndType.entity().getCreatedDate(), Op.LTEQ); |
| AlertSearchByIdsAndType.and("data_center_id", AlertSearchByIdsAndType.entity().getDataCenterId(), Op.EQ); |
| AlertSearchByIdsAndType.and("archived", AlertSearchByIdsAndType.entity().getArchived(), Op.EQ); |
| AlertSearchByIdsAndType.done(); |
| } |
| |
| @Override |
| public AlertVO getLastAlert(short type, long dataCenterId, Long podId, Long clusterId) { |
| Filter searchFilter = new Filter(AlertVO.class, "createdDate", Boolean.FALSE, Long.valueOf(0), Long.valueOf(1)); |
| SearchCriteria<AlertVO> sc = createSearchCriteria(); |
| |
| sc.addAnd("type", SearchCriteria.Op.EQ, Short.valueOf(type)); |
| sc.addAnd("dataCenterId", SearchCriteria.Op.EQ, Long.valueOf(dataCenterId)); |
| sc.addAnd("archived", SearchCriteria.Op.EQ, false); |
| if (podId != null) { |
| sc.addAnd("podId", SearchCriteria.Op.EQ, podId); |
| } |
| if (clusterId != null) { |
| sc.addAnd("clusterId", SearchCriteria.Op.EQ, clusterId); |
| } |
| |
| List<AlertVO> alerts = listBy(sc, searchFilter); |
| if ((alerts != null) && !alerts.isEmpty()) { |
| return alerts.get(0); |
| } |
| return null; |
| } |
| |
| @Override |
| public AlertVO getLastAlert(short type, long dataCenterId, Long podId) { |
| Filter searchFilter = new Filter(AlertVO.class, "createdDate", Boolean.FALSE, Long.valueOf(0), Long.valueOf(1)); |
| SearchCriteria<AlertVO> sc = createSearchCriteria(); |
| |
| sc.addAnd("type", SearchCriteria.Op.EQ, Short.valueOf(type)); |
| sc.addAnd("dataCenterId", SearchCriteria.Op.EQ, Long.valueOf(dataCenterId)); |
| sc.addAnd("archived", SearchCriteria.Op.EQ, false); |
| if (podId != null) { |
| sc.addAnd("podId", SearchCriteria.Op.EQ, podId); |
| } |
| |
| List<AlertVO> alerts = listBy(sc, searchFilter); |
| if ((alerts != null) && !alerts.isEmpty()) { |
| return alerts.get(0); |
| } |
| return null; |
| } |
| |
| @Override |
| public boolean archiveAlert(List<Long> ids, String type, Date startDate, Date endDate, Long zoneId) { |
| SearchCriteria<AlertVO> sc = AlertSearchByIdsAndType.create(); |
| |
| if (ids != null) { |
| sc.setParameters("id", ids.toArray(new Object[ids.size()])); |
| } |
| if (type != null) { |
| sc.setParameters("type", type); |
| } |
| if (zoneId != null) { |
| sc.setParameters("data_center_id", zoneId); |
| } |
| if (startDate != null && endDate != null) { |
| sc.setParameters("createdDateB", startDate, endDate); |
| } else if (endDate != null) { |
| sc.setParameters("createdDateL", endDate); |
| } |
| sc.setParameters("archived", false); |
| |
| boolean result = true; |
| ; |
| List<AlertVO> alerts = listBy(sc); |
| if (ids != null && alerts.size() < ids.size()) { |
| result = false; |
| return result; |
| } |
| if (alerts != null && !alerts.isEmpty()) { |
| TransactionLegacy txn = TransactionLegacy.currentTxn(); |
| txn.start(); |
| for (AlertVO alert : alerts) { |
| alert = lockRow(alert.getId(), true); |
| alert.setArchived(true); |
| update(alert.getId(), alert); |
| txn.commit(); |
| } |
| txn.close(); |
| } |
| return result; |
| } |
| |
| @Override |
| public boolean deleteAlert(List<Long> ids, String type, Date startDate, Date endDate, Long zoneId) { |
| SearchCriteria<AlertVO> sc = AlertSearchByIdsAndType.create(); |
| |
| if (ids != null) { |
| sc.setParameters("id", ids.toArray(new Object[ids.size()])); |
| } |
| if (type != null) { |
| sc.setParameters("type", type); |
| } |
| if (zoneId != null) { |
| sc.setParameters("data_center_id", zoneId); |
| } |
| if (startDate != null && endDate != null) { |
| sc.setParameters("createdDateB", startDate, endDate); |
| } else if (endDate != null) { |
| sc.setParameters("createdDateL", endDate); |
| } |
| sc.setParameters("archived", false); |
| |
| boolean result = true; |
| List<AlertVO> alerts = listBy(sc); |
| if (ids != null && alerts.size() < ids.size()) { |
| result = false; |
| return result; |
| } |
| remove(sc); |
| return result; |
| } |
| |
| @Override |
| public List<AlertVO> listOlderAlerts(Date oldTime) { |
| if (oldTime == null) |
| return null; |
| SearchCriteria<AlertVO> sc = createSearchCriteria(); |
| sc.addAnd("createdDate", SearchCriteria.Op.LT, oldTime); |
| sc.addAnd("archived", SearchCriteria.Op.EQ, false); |
| return listIncludingRemovedBy(sc, null); |
| } |
| |
| } |