UpgradeManager.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.hdfs.server.common;
  19. import java.io.IOException;
  20. import java.util.SortedSet;
  21. import org.apache.hadoop.hdfs.protocol.FSConstants;
  22. import org.apache.hadoop.hdfs.server.protocol.UpgradeCommand;
  23. /**
  24.  * Generic upgrade manager.
  25.  * 
  26.  * {@link #broadcastCommand} is the command that should be 
  27.  *
  28.  */
  29. public abstract class UpgradeManager {
  30.   protected SortedSet<Upgradeable> currentUpgrades = null;
  31.   protected boolean upgradeState = false; // true if upgrade is in progress
  32.   protected int upgradeVersion = 0;
  33.   protected UpgradeCommand broadcastCommand = null;
  34.   public synchronized UpgradeCommand getBroadcastCommand() {
  35.     return this.broadcastCommand;
  36.   }
  37.   public boolean getUpgradeState() {
  38.     return this.upgradeState;
  39.   }
  40.   public int getUpgradeVersion(){
  41.     return this.upgradeVersion;
  42.   }
  43.   public void setUpgradeState(boolean uState, int uVersion) {
  44.     this.upgradeState = uState;
  45.     this.upgradeVersion = uVersion;
  46.   }
  47.   public SortedSet<Upgradeable> getDistributedUpgrades() throws IOException {
  48.     return UpgradeObjectCollection.getDistributedUpgrades(
  49.                                             getUpgradeVersion(), getType());
  50.   }
  51.   public short getUpgradeStatus() {
  52.     if(currentUpgrades == null)
  53.       return 100;
  54.     return currentUpgrades.first().getUpgradeStatus();
  55.   }
  56.   public boolean initializeUpgrade() throws IOException {
  57.     currentUpgrades = getDistributedUpgrades();
  58.     if(currentUpgrades == null) {
  59.       // set new upgrade state
  60.       setUpgradeState(false, FSConstants.LAYOUT_VERSION);
  61.       return false;
  62.     }
  63.     Upgradeable curUO = currentUpgrades.first();
  64.     // set and write new upgrade state into disk
  65.     setUpgradeState(true, curUO.getVersion());
  66.     return true;
  67.   }
  68.   public boolean isUpgradeCompleted() {
  69.     if (currentUpgrades == null) {
  70.       return true;
  71.     }
  72.     return false;
  73.   }
  74.   public abstract HdfsConstants.NodeType getType();
  75.   public abstract boolean startUpgrade() throws IOException;
  76.   public abstract void completeUpgrade() throws IOException;
  77. }