RunOnce.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.contrib.failmon;
  19. import java.util.ArrayList;
  20. /**********************************************************
  21. * Runs a set of monitoring jobs once for the local node. The set of
  22. * jobs to be run is the intersection of the jobs specifed in the
  23. * configuration file and the set of jobs specified in the --only
  24. * command line argument.
  25.  **********************************************************/ 
  26. public class RunOnce {
  27.   LocalStore lstore;
  28.   ArrayList<MonitorJob> monitors;
  29.   
  30.   boolean uploading = true;
  31.   
  32.   public RunOnce(String confFile) {
  33.     
  34.     Environment.prepare(confFile);
  35.     
  36.     String localTmpDir;
  37.     
  38.     // running as a stand-alone application
  39.     localTmpDir = System.getProperty("java.io.tmpdir");
  40.     Environment.setProperty("local.tmp.dir", localTmpDir);
  41.         
  42.     monitors = Environment.getJobs();
  43.     lstore = new LocalStore();
  44.     uploading  = true;
  45.   }
  46.   private void filter (String [] ftypes) {
  47.     ArrayList<MonitorJob> filtered = new ArrayList<MonitorJob>();
  48.     boolean found;
  49.     
  50.     // filter out unwanted monitor jobs
  51.     for (MonitorJob job : monitors) {
  52.       found = false;
  53.       for (String ftype : ftypes)
  54. if (job.type.equalsIgnoreCase(ftype))
  55.     found = true;
  56.       if (found)
  57. filtered.add(job);
  58.     }
  59.     // disable uploading if not requested
  60.     found = false;
  61.     for (String ftype : ftypes)
  62.       if (ftype.equalsIgnoreCase("upload"))
  63. found = true;
  64.     if (!found)
  65.       uploading = false;
  66.     
  67.     monitors = filtered;
  68.   }
  69.   
  70.   private void run() {
  71.     
  72.     Environment.logInfo("Failmon started successfully.");
  73.     for (int i = 0; i < monitors.size(); i++) {
  74.       Environment.logInfo("Calling " + monitors.get(i).job.getInfo() + "...t");
  75.       monitors.get(i).job.monitor(lstore);
  76.     }
  77.     if (uploading)
  78.       lstore.upload();
  79.     lstore.close();
  80.   }
  81.   public void cleanup() {
  82.     // nothing to be done
  83.   }
  84.   
  85.   public static void main (String [] args) {
  86.     String configFilePath = "./conf/failmon.properties";
  87.     String [] onlyList = null;
  88.     
  89.     // Parse command-line parameters
  90.     for (int i = 0; i < args.length - 1; i++) {
  91.       if (args[i].equalsIgnoreCase("--config"))
  92. configFilePath = args[i + 1];
  93.       else if (args[i].equalsIgnoreCase("--only"))
  94. onlyList = args[i + 1].split(",");
  95.     }
  96.     RunOnce ro = new RunOnce(configFilePath);
  97.     // only keep the requested types of jobs
  98.     if (onlyList != null)
  99.       ro.filter(onlyList);
  100.     // run once only
  101.     ro.run();
  102.   }
  103. }