From a0213b9b7cc10cc1bcb4e07eacc32fb9b91f1736 Mon Sep 17 00:00:00 2001
From: "zhang.xin" <ascrutae@gmail.com>
Date: Mon, 14 Mar 2016 23:43:11 +0800
Subject: [PATCH] =?UTF-8?q?1.=E4=BF=AE=E5=A4=8DAnalysisMode=E4=BC=A0?=
 =?UTF-8?q?=E9=80=92=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/cloud/skywalking/analysis/AnalysisServerDriver.java   | 2 ++
 .../skywalking/analysis/chainbuild/ChainBuildReducer.java    | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/AnalysisServerDriver.java b/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/AnalysisServerDriver.java
index 127156dfe..c3b022497 100644
--- a/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/AnalysisServerDriver.java
+++ b/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/AnalysisServerDriver.java
@@ -52,6 +52,7 @@ public class AnalysisServerDriver extends Configured implements Tool {
     public int run(String[] args) throws Exception {
         ConfigInitializer.initialize();
         Configuration conf = new Configuration();
+        conf.set("skywalking.analysis.mode", String.valueOf(Config.AnalysisServer.IS_ACCUMULATE_MODE));
         conf.set("hbase.zookeeper.quorum", Config.HBase.ZK_QUORUM);
         conf.set("hbase.zookeeper.property.clientPort", Config.HBase.ZK_CLIENT_PORT);
         String[] otherArgs = new GenericOptionsParser(conf, args).getRemainingArgs();
@@ -62,6 +63,7 @@ public class AnalysisServerDriver extends Configured implements Tool {
 
         Job job = Job.getInstance(conf);
         job.setJarByClass(AnalysisServerDriver.class);
+
         Scan scan = buildHBaseScan(args);
 
         TableMapReduceUtil.initTableMapperJob(HBaseTableMetaData.TABLE_CALL_CHAIN.TABLE_NAME, scan, ChainBuildMapper.class,
diff --git a/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/ChainBuildReducer.java b/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/ChainBuildReducer.java
index 8ec60a9c9..3667f8e66 100644
--- a/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/ChainBuildReducer.java
+++ b/skywalking-analysis/src/main/java/com/ai/cloud/skywalking/analysis/chainbuild/ChainBuildReducer.java
@@ -3,6 +3,7 @@ package com.ai.cloud.skywalking.analysis.chainbuild;
 import com.ai.cloud.skywalking.analysis.chainbuild.entity.CallChainTree;
 import com.ai.cloud.skywalking.analysis.chainbuild.po.ChainInfo;
 import com.ai.cloud.skywalking.analysis.chainbuild.po.SpecificTimeCallTreeMergedChainIdContainer;
+import com.ai.cloud.skywalking.analysis.config.Config;
 import com.ai.cloud.skywalking.analysis.config.ConfigInitializer;
 import com.google.gson.Gson;
 import org.apache.hadoop.hbase.util.Bytes;
@@ -22,6 +23,10 @@ public class ChainBuildReducer extends Reducer<Text, Text, Text, IntWritable> {
     protected void setup(Context context) throws IOException,
             InterruptedException {
         ConfigInitializer.initialize();
+        Config.AnalysisServer.IS_ACCUMULATE_MODE = Boolean.parseBoolean(context.getConfiguration()
+                .get("skywalking.analysis.mode", "false"));
+        logger.info("Skywalking analysis mode :[{}]",
+                Config.AnalysisServer.IS_ACCUMULATE_MODE ? "ACCUMULATE" : "REWRITE");
     }
 
     @Override