| // 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.storage.dao; |
| |
| import java.util.List; |
| |
| |
| import org.apache.log4j.Logger; |
| import org.springframework.stereotype.Component; |
| |
| import com.cloud.storage.Upload.Mode; |
| import com.cloud.storage.Upload.Status; |
| import com.cloud.storage.UploadVO; |
| import com.cloud.utils.db.GenericDaoBase; |
| import com.cloud.utils.db.SearchBuilder; |
| import com.cloud.utils.db.SearchCriteria; |
| |
| @Component |
| public class UploadDaoImpl extends GenericDaoBase<UploadVO, Long> implements UploadDao { |
| public static final Logger s_logger = Logger.getLogger(UploadDaoImpl.class.getName()); |
| protected final SearchBuilder<UploadVO> typeUploadStatusSearch; |
| protected final SearchBuilder<UploadVO> typeHostAndUploadStatusSearch; |
| protected final SearchBuilder<UploadVO> typeModeAndStatusSearch; |
| |
| protected static final String UPDATE_UPLOAD_INFO = "UPDATE upload SET upload_state = ?, upload_pct= ?, last_updated = ? " |
| + ", upload_error_str = ?, upload_job_id = ? " + "WHERE host_id = ? and type_id = ? and type = ?"; |
| |
| protected static final String UPLOADS_STATE_DC = "SELECT * FROM upload t, host h where t.host_id = h.id and h.data_center_id=? " |
| + " and t.type_id=? and t.upload_state = ?"; |
| |
| public UploadDaoImpl() { |
| typeUploadStatusSearch = createSearchBuilder(); |
| typeUploadStatusSearch.and("type_id", typeUploadStatusSearch.entity().getTypeId(), SearchCriteria.Op.EQ); |
| typeUploadStatusSearch.and("upload_state", typeUploadStatusSearch.entity().getUploadState(), SearchCriteria.Op.EQ); |
| typeUploadStatusSearch.and("type", typeUploadStatusSearch.entity().getType(), SearchCriteria.Op.EQ); |
| typeUploadStatusSearch.done(); |
| |
| typeHostAndUploadStatusSearch = createSearchBuilder(); |
| typeHostAndUploadStatusSearch.and("host_id", typeHostAndUploadStatusSearch.entity().getDataStoreId(), SearchCriteria.Op.EQ); |
| typeHostAndUploadStatusSearch.and("upload_state", typeHostAndUploadStatusSearch.entity().getUploadState(), SearchCriteria.Op.EQ); |
| typeHostAndUploadStatusSearch.done(); |
| |
| typeModeAndStatusSearch = createSearchBuilder(); |
| typeModeAndStatusSearch.and("mode", typeModeAndStatusSearch.entity().getMode(), SearchCriteria.Op.EQ); |
| typeModeAndStatusSearch.and("upload_state", typeModeAndStatusSearch.entity().getUploadState(), SearchCriteria.Op.EQ); |
| typeModeAndStatusSearch.done(); |
| |
| } |
| |
| @Override |
| public List<UploadVO> listByTypeUploadStatus(long typeId, UploadVO.Type type, UploadVO.Status uploadState) { |
| SearchCriteria<UploadVO> sc = typeUploadStatusSearch.create(); |
| sc.setParameters("type_id", typeId); |
| sc.setParameters("type", type); |
| sc.setParameters("upload_state", uploadState.toString()); |
| return listBy(sc); |
| } |
| |
| @Override |
| public List<UploadVO> listByHostAndUploadStatus(long sserverId, Status uploadState) { |
| SearchCriteria<UploadVO> sc = typeHostAndUploadStatusSearch.create(); |
| sc.setParameters("host_id", sserverId); |
| sc.setParameters("upload_state", uploadState.toString()); |
| return listBy(sc); |
| } |
| |
| @Override |
| public List<UploadVO> listByModeAndStatus(Mode mode, Status uploadState) { |
| SearchCriteria<UploadVO> sc = typeModeAndStatusSearch.create(); |
| sc.setParameters("mode", mode.toString()); |
| sc.setParameters("upload_state", uploadState.toString()); |
| return listBy(sc); |
| } |
| } |