mirror of https://github.com/apache/cloudstack.git
88 lines
4.1 KiB
Java
88 lines
4.1 KiB
Java
// 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 javax.ejb.Local;
|
|
|
|
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
|
|
@Local(value = {UploadDao.class})
|
|
public class UploadDaoImpl extends GenericDaoBase<UploadVO, Long> implements UploadDao {
|
|
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);
|
|
}
|
|
} |