Project: Flume-Hive
/**
 * Licensed to Cloudera, Inc. under one 
 * or more contributor license agreements.  See the NOTICE file 
 * distributed with this work for additional information 
 * regarding copyright ownership.  Cloudera, Inc. 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.cloudera.flume.master.commands; 
 
import java.io.IOException; 
 
import org.slf4j.Logger; 
import org.slf4j.LoggerFactory; 
 
import com.cloudera.flume.master.Command; 
import com.cloudera.flume.master.Execable; 
import com.cloudera.flume.master.FlumeMaster; 
import com.cloudera.flume.master.MasterExecException; 
import com.google.common.base.Preconditions; 
 
/**
 * This implements the "setChokeLimit" command 
 */
 
public class SetChokeLimitForm { 
  static final Logger LOG = LoggerFactory.getLogger(SetChokeLimitForm.class); 
  String logicalNode; 
 
  public String getLogicalNode() { 
    return logicalNode; 
  } 
 
  public void setLogicalNode(String logicalNode) { 
    this.logicalNode = logicalNode; 
  } 
 
  /**
   * Convert this bean into a command. 
   */
 
  public Command toCommand() { 
    String[] args = { "physicalnode""chokeID""limit" }; 
    return new Command("setChokeLimit", args); 
  } 
 
  /**
   * Build an execable that will execute the command. 
   */
 
  public static Execable buildExecable() { 
    return new Execable() { 
      // TODO(Vibhor): Once the PhysicalNodeThrottling patch is in, change the 
      // preconditions here and add the ability to get the physicalNode 
      // throttling limit. 
      @Override 
      public void exec(String[] args) throws MasterExecException, IOException { 
        // first check the length of the arguments 
        Preconditions.checkArgument(args.length > 2
            "Usage: setChokeLimit physicalNode chokeID limit"); 
 
        String physicalNodeName = args[0]; 
        // issue a polite warning if the physicalnode does not exist yet 
        if (FlumeMaster.getInstance().getSpecMan().getLogicalNode( 
            physicalNodeName).isEmpty()) { 
          LOG.warn("PhysicalNode: " + physicalNodeName + " not present yet!"); 
        } 
 
        int limit = 0
        String chokerName = args[1]; 
        try { 
          limit = Integer.parseInt(args[2]); 
        } catch (NumberFormatException e) { 
          LOG.error("Limit not given in the right format"); 
          throw new MasterExecException("Limit not given in the right format"
              e); 
        } 
 
        Preconditions.checkState(limit >= 0"Limit has to be at least 0"); 
        // only works in memory!! not in zookeeper. 
        FlumeMaster.getInstance().getSpecMan().addChokeLimit(physicalNodeName, 
            chokerName, limit); 
      } 
    }; 
  } 
}