| // 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.upload; |
| |
| |
| import com.cloud.agent.api.storage.UploadAnswer; |
| import com.cloud.agent.api.storage.UploadProgressCommand.RequestType; |
| import com.cloud.storage.VMTemplateStorageResourceAssoc.Status; |
| import org.apache.logging.log4j.Level; |
| |
| public abstract class UploadActiveState extends UploadState { |
| |
| public UploadActiveState(UploadListener ul) { |
| super(ul); |
| } |
| |
| @Override |
| public String handleAbort() { |
| return Status.ABANDONED.toString(); |
| } |
| |
| @Override |
| public String handleDisconnect() { |
| |
| return Status.UPLOAD_ERROR.toString(); |
| } |
| |
| @Override |
| public String handleAnswer(UploadAnswer answer) { |
| if (logger.isDebugEnabled()) { |
| logger.debug("handleAnswer, answer status=" + answer.getUploadStatus() + ", curr state=" + getName()); |
| } |
| switch (answer.getUploadStatus()) { |
| case UPLOAD_IN_PROGRESS: |
| getUploadListener().scheduleStatusCheck(RequestType.GET_STATUS); |
| return Status.UPLOAD_IN_PROGRESS.toString(); |
| case UPLOADED: |
| getUploadListener().scheduleImmediateStatusCheck(RequestType.PURGE); |
| getUploadListener().cancelTimeoutTask(); |
| return Status.UPLOADED.toString(); |
| case NOT_UPLOADED: |
| getUploadListener().scheduleStatusCheck(RequestType.GET_STATUS); |
| return Status.NOT_UPLOADED.toString(); |
| case UPLOAD_ERROR: |
| getUploadListener().cancelStatusTask(); |
| getUploadListener().cancelTimeoutTask(); |
| return Status.UPLOAD_ERROR.toString(); |
| case UNKNOWN: |
| getUploadListener().cancelStatusTask(); |
| getUploadListener().cancelTimeoutTask(); |
| return Status.UPLOAD_ERROR.toString(); |
| default: |
| return null; |
| } |
| } |
| |
| @Override |
| public String handleTimeout(long updateMs) { |
| if (logger.isTraceEnabled()) { |
| getUploadListener().log("handleTimeout, updateMs=" + updateMs + ", curr state= " + getName(), Level.TRACE); |
| } |
| String newState = getName(); |
| if (updateMs > 5 * UploadListener.STATUS_POLL_INTERVAL) { |
| newState = Status.UPLOAD_ERROR.toString(); |
| getUploadListener().log("timeout: transitioning to upload error state, currstate=" + getName(), Level.DEBUG); |
| } else if (updateMs > 3 * UploadListener.STATUS_POLL_INTERVAL) { |
| getUploadListener().cancelStatusTask(); |
| getUploadListener().scheduleImmediateStatusCheck(RequestType.GET_STATUS); |
| getUploadListener().scheduleTimeoutTask(3 * UploadListener.STATUS_POLL_INTERVAL); |
| getUploadListener().log(getName() + " first timeout: checking again ", Level.DEBUG); |
| } else { |
| getUploadListener().scheduleTimeoutTask(3 * UploadListener.STATUS_POLL_INTERVAL); |
| } |
| return newState; |
| } |
| |
| @Override |
| public void onEntry(String prevState, UploadEvent event, Object evtObj) { |
| super.onEntry(prevState, event, evtObj); |
| |
| if (event == UploadEvent.UPLOAD_ANSWER) { |
| getUploadListener().setLastUpdated(); |
| } |
| |
| } |
| |
| @Override |
| public void onExit() { |
| } |
| } |