flume高并发优化——(13)扩展三级文件配置&利用Headers扩展属性

        上篇博客中,我们对flume进行了文件组进行了扩展,但是我们现在的配置还是针对某组文件单独配置,这样的维护成本还是太高,为了持续优化,我们对常见的三级文件进行优化(**logs/**project/**type/*.log)利用公司内部的约定,大家一起为简单配置努力。

1,规范目录结构

        在本项目约定下,日志结构采取如下:

        /logs/*project/type-tomcat/*.log


2,抽象日志配置

        在这个配置下,我们抽象成两级配置:

        第一级:根目录

                /logs

        第二级:业务目录

                /*project

        第三级:类型目录(不可扩展)

                一般为业务日志,数据日志

        source config

agent1.sources.s1.type = org.apache.flume.source.ExecTailSource
agent1.sources.s1.basefilepath=/export/home/tomcat/logs
agent1.sources.s1.filepath=account.soa|user.soa
agent1.sources.s1.apilogs=all
agent1.sources.s1.operatelogs=no
agent1.sources.s1.tomcatlogs=no
agent1.sources.s1.tailing=all
agent1.sources.s1.filenameRegExp=(.log{1})$
agent1.sources.s1.readinterval=300
agent1.sources.s1.startAtBeginning=false
agent1.sources.s1.restart=true
agent1.sources.s1.contextIsJson=false
agent1.sources.s1.contextIsFlumeLog=true



3,Headers定制sink

        3.1 elasticsearch sink

                indexType:The type to index the document to, defaults to ‘log’ Arbitrary header substitution is supported, eg. %{header} replaces with value of named event header。


        3.2 kafka sink

                topic:The topic in Kafka to which the messages will be published. If this parameter is configured, messages will be published to this topic. If the event header contains a “topic” field, the event will be published to that topic overriding the topic configured here.


4,demo

        4.1 elasticsearch sink config & set headers

        flume config

agent1.sinks.k1.type=org.apache.flume.sink.elasticsearch.ElasticSearchSink
agent1.sinks.k1.batchSize=5000
agent1.sinks.k1.hostNames=192.168.2.16:9300
agent1.sinks.k1.indexType =log_file_tender
agent1.sinks.k1.indexName=logstash_test
agent1.sinks.k1.clusterName=unifyloggingplatform
agent1.sinks.k1.serializer=org.apache.flume.sink.elasticsearch.ElasticSearchLogStashEventSerializer
agent1.sinks.k1.indexNameBuilder=org.apache.flume.sink.elasticsearch.TimeBasedIndexNameBuilder
agent1.sinks.k1.dateFormat=dd

        set headers

 Event oneEvent = EventBuilder.withBody(bodyjson.getBytes(charset));
 oneEvent.getHeaders().put("UlogDomain",domain);

        4.2 kafka sink config & set headers

        flume config

a1.sinks.k1.type = org.apache.flume.sink.kafka.KafkaSink
a1.sinks.k1.topic = mytopic
a1.sinks.k1.brokerList = localhost:9092
a1.sinks.k1.requiredAcks = 1
a1.sinks.k1.batchSize = 20
a1.sinks.k1.channel = c1

        set headers

 Event oneEvent = EventBuilder.withBody(bodyjson.getBytes(charset));
 oneEvent.getHeaders().put("UlogDomain",domain);
 oneEvent.getHeaders().put("topic","file_channel_topic_"+domain);

总结:

        在全局的消息体中,保留一个map,让用户自定义,以便在后期的配置中,预留对象,这种设计思想,既满足了灵活又满足了规范,我们在以后的设计中,也要充分借鉴,为用户想到用户想要的,给用户留下用户以后想加入的,方便用户扩展!


全局代码实现:

项目地址:https://github.com/xvshu/flume-files-source

/*
 * 作者:许恕
 * 时间:2016年5月3日
 * 功能:实现tail 某目录下的所有符合正则条件的文件
 * Email:[email protected]
 * To detect all files in a folder
 */

package org.apache.flume.source;

import com.dianping.cat.Cat;
import com.dianping.cat.message.Transaction;
import com.el.ump.profiler.Profiler;
import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.ThreadFactoryBuilder;
import org.apache.flume.Context;
import org.apache.flume.Event;
import org.apache.flume.EventDrivenSource;
import org.apache.flume.SystemClock;
import org.apache.flume.channel.ChannelProcessor;
import org.apache.flume.conf.Configurable;
import org.apache.flume.event.EventBuilder;
import org.apache.flume.instrumentation.SourceCounter;
import org.apache.flume.source.utils.MsgBuildeJson;
import org.mortbay.util.ajax.JSON;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.nio.charset.Charset;
import java.util.*;
import java.util.concurrent.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 *  step:
 *    1,config one path
 *    2,find all file with RegExp
 *    3,tail one children file
 *    4,batch to channal
 *
 *  demo:
 *    demo.sources.s1.type = org.apache.flume.source.ExecTailSource
 *    demo.sources.s1.filepath=/export/home/tomcat/logs/auth.el.net/
 *    demo.sources.s1.filenameRegExp=(.log{1})$
 *    demo.sources.s1.tailing=true
 *    demo.sources.s1.readinterval=300
 *    demo.sources.s1.startAtBeginning=false
 *    demo.sources.s1.restart=true
 */
public class ExecTailSource extends AbstractSource implements EventDrivenSource,
        Configurable {

  private static final Logger logger = LoggerFactory
          .getLogger(ExecTailSource.class);

  private SourceCounter sourceCounter;
  private ExecutorService executor;
  private List listRuners;
  private List> listFuture;
  private long restartThrottle;
  private boolean restart = true;
  private boolean logStderr;
  private Integer bufferCount;
  private long batchTimeout;
  private Charset charset;
  private String  basefilepath;
  private String filepath;
  private String filenameRegExp;

  private String apilogs;
  private String operatelogs;
  private String tomcatlogs;

  private List apilogsList = new ArrayList();
  private List operatelogsList = new ArrayList();
  private List tomcatlogsList =  new ArrayList();

  private String tailing;
  private Integer readinterval;
  private boolean startAtBeginning;
  private boolean contextIsJson;
  private String fileWriteJson;
  private Long flushTime;
  private boolean contextIsFlumeLog;
  private String domain;
  private String msgTypeConfig;


  @Override
  public void start() {
    logger.info("=start=> flume tail source start begin time:"+new Date().toString());
    logger.info("ExecTail source starting with filepath:{}", filepath);

    List listFiles = getFileList();
    if(listFiles==null || listFiles.isEmpty()){
      logger.info("The filepath's file not have fiels with filenameRegExp");
      sourceCounter.start();
      super.start();
      return;
    }

    Properties prop=null;

    try{
      prop = new Properties();//属性集合对象
      FileInputStream fis = new FileInputStream(fileWriteJson);//属性文件流
      prop.load(fis);
    }catch(Exception ex){
      logger.error("==>",ex);
    }



    executor = Executors.newFixedThreadPool(listFiles.size());

    listRuners = new ArrayList();
    listFuture = new ArrayList>();

    logger.info("files size is {} ", listFiles.size());
    // FIXME: Use a callback-like executor / future to signal us upon failure.
    for(String oneFilePath : listFiles){
      ExecRunnable runner = new ExecRunnable(getChannelProcessor(), sourceCounter,
              restart, restartThrottle, logStderr, bufferCount, batchTimeout,
              charset,oneFilePath,tailing,readinterval,startAtBeginning,contextIsJson,
              prop,fileWriteJson,flushTime,contextIsFlumeLog,domain);
      listRuners.add(runner);
      Future runnerFuture = executor.submit(runner);
      listFuture.add(runnerFuture);
      logger.info("{} is begin running",oneFilePath);
    }

    /*
     * NB: This comes at the end rather than the beginning of the method because
     * it sets our state to running. We want to make sure the executor is alive
     * and well first.
     */
    sourceCounter.start();
    super.start();
    logger.info("=start=> flume tail source start end time:"+new Date().toString());
    logger.debug("ExecTail source started");
  }

  @Override
  public void stop() {

    logger.info("=stop=> flume tail source stop begin time:"+new Date().toString());
    if(listRuners !=null && !listRuners.isEmpty()){
      for(ExecRunnable oneRunner : listRuners){
        if(oneRunner != null) {
          oneRunner.setRestart(false);
          oneRunner.kill();
        }
      }
    }


    if(listFuture !=null && !listFuture.isEmpty()){
      for(Future oneFuture : listFuture){
        if (oneFuture != null) {
          logger.debug("Stopping ExecTail runner");
          oneFuture.cancel(true);
          logger.debug("ExecTail runner stopped");
        }
      }
    }

    executor.shutdown();
    while (!executor.isTerminated()) {
      logger.debug("Waiting for ExecTail executor service to stop");
      try {
        executor.awaitTermination(500, TimeUnit.MILLISECONDS);
      } catch (InterruptedException e) {
        logger.debug("Interrupted while waiting for ExecTail executor service "
                + "to stop. Just exiting.");
        Thread.currentThread().interrupt();
      }
    }




    sourceCounter.stop();
    super.stop();
    logger.info("=stop=> flume tail source stop end time:"+new Date().toString());

  }

  @Override
  public void configure(Context context) {

    basefilepath=context.getString(ExecTailSourceConfigurationConstants.BASE_FILE_PATH,
            ExecTailSourceConfigurationConstants.DEFAULT_BASE_FILE_PATH);

    filepath = context.getString("filepath");
    Preconditions.checkState(filepath != null,
            "The parameter filepath must be specified");
    logger.info("The parameter filepath is {}" ,filepath);

    apilogs = context.getString(ExecTailSourceConfigurationConstants.APILOGS,
            ExecTailSourceConfigurationConstants.DEFAULT_CHILDREN_FILE);

    apilogsList.addAll(Arrays.asList(apilogs.split("\\|")));

    operatelogs = context.getString(ExecTailSourceConfigurationConstants.OPERATELOGS,
            ExecTailSourceConfigurationConstants.DEFAULT_CHILDREN_FILE);
    operatelogsList.addAll(Arrays.asList(operatelogs.split("\\|")));

    tomcatlogs = context.getString(ExecTailSourceConfigurationConstants.TOMCATLOGS,
            ExecTailSourceConfigurationConstants.DEFAULT_CHILDREN_FILE);
    tomcatlogsList.addAll(Arrays.asList(tomcatlogs.split("\\|")));

    filenameRegExp = context.getString("filenameRegExp");
    Preconditions.checkState(filenameRegExp != null,
            "The parameter filenameRegExp must be specified");
    logger.info("The parameter filenameRegExp is {}" ,filenameRegExp);

    msgTypeConfig=context.getString(ExecTailSourceConfigurationConstants.CONFIG_MSGTYPECONFIG_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_MSGTYPECONFIG);

    String[] defultTypes = ExecTailSourceConfigurationConstants.DEFAULT_MSGTYPECONFIG_DEFULT.split("\\,");
    for(String oneType : defultTypes){
      String[] oneTypeMap = oneType.split("\\:");
      MsgBuildeJson.MsgTypes.put(oneTypeMap[0],oneTypeMap[1].split("\\|"));
    }

    try {
      if (msgTypeConfig != null && !msgTypeConfig.trim().isEmpty()) {
        String[] userTypes = msgTypeConfig.split("\\,");
        for(String oneType : defultTypes){
          String[] oneTypeMap = oneType.split("\\:");
          if(oneTypeMap.length>=2){
            MsgBuildeJson.MsgTypes.put(oneTypeMap[0],oneTypeMap[1].split("\\|"));
          }
        }
      }
    }catch (Exception ex){
      ex.printStackTrace();
    }

    logger.info("=MsgBuildeJson.MsgTypes is =>"+ JSON.toString(MsgBuildeJson.MsgTypes));


    MsgBuildeJson.MsgIntAtti.addAll(Arrays.asList(ExecTailSourceConfigurationConstants.MAP_INT_ATTRIBUTE.split("\\,")));

    contextIsJson= context.getBoolean(ExecTailSourceConfigurationConstants.CONFIG_CONTEXTISJSON_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_CONTEXTISJSON);

    contextIsFlumeLog=context.getBoolean(ExecTailSourceConfigurationConstants.CONFIG_CONTEXTISFLUMELOG_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_CONTEXTISFLUMELOG);

    domain=context.getString(ExecTailSourceConfigurationConstants.CONFIG_DOMIAN_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_DOMAIN);

    fileWriteJson= context.getString(ExecTailSourceConfigurationConstants.CONFIG_FILEWRITEJSON_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_FILEWRITEJSON);

    flushTime= context.getLong(ExecTailSourceConfigurationConstants.CONFIG_FLUSHTIME_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_FLUSHTIME);

    restartThrottle = context.getLong(ExecTailSourceConfigurationConstants.CONFIG_RESTART_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_RESTART_THROTTLE);

    tailing = context.getString(ExecTailSourceConfigurationConstants.CONFIG_TAILING_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_ISTAILING_TRUE);

    readinterval=context.getInteger(ExecTailSourceConfigurationConstants.CONFIG_READINTERVAL_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_READINTERVAL);

    startAtBeginning=context.getBoolean(ExecTailSourceConfigurationConstants.CONFIG_STARTATBEGINNING_THROTTLE,
            ExecTailSourceConfigurationConstants.DEFAULT_STARTATBEGINNING);

    restart = context.getBoolean(ExecTailSourceConfigurationConstants.CONFIG_RESTART,
            ExecTailSourceConfigurationConstants.DEFAULT_RESTART_TRUE);

    logStderr = context.getBoolean(ExecTailSourceConfigurationConstants.CONFIG_LOG_STDERR,
            ExecTailSourceConfigurationConstants.DEFAULT_LOG_STDERR);

    bufferCount = context.getInteger(ExecTailSourceConfigurationConstants.CONFIG_BATCH_SIZE,
            ExecTailSourceConfigurationConstants.DEFAULT_BATCH_SIZE);

    batchTimeout = context.getLong(ExecTailSourceConfigurationConstants.CONFIG_BATCH_TIME_OUT,
            ExecTailSourceConfigurationConstants.DEFAULT_BATCH_TIME_OUT);

    charset = Charset.forName(context.getString(ExecTailSourceConfigurationConstants.CHARSET,
            ExecTailSourceConfigurationConstants.DEFAULT_CHARSET));


    if (sourceCounter == null) {
      sourceCounter = new SourceCounter(getName());
    }
  }

  /**
   * 获取指定路径下的所有文件列表
   *
   * @return
   */
  public  List getFileList() {
    List fatherFilesList = new ArrayList();
    List listFile = new ArrayList();
    String[] baseFatherFiles =basefilepath.split("\\|");
    for(String  oneBaseFile : baseFatherFiles){
      fatherFilesList.addAll(getPassDomainDir(oneBaseFile));
    }
    logger.info("=getFileList=> fatherfiles is "+JSON.toString(fatherFilesList));
    if(fatherFilesList != null || fatherFilesList.size()>0){
      for(String oneDir :fatherFilesList ){
        listFile.addAll(getChildrenFileList(oneDir));
      }
    }
    return listFile;
  }

  public List getPassDomainDir(String dir){
    String[] passFileNames = filepath.split("\\|");
    List listReturn  =  new ArrayList();
    ArrayList PassFileListName = new ArrayList(Arrays.asList(passFileNames));
    File dirFile = new File(dir);
    File[] files = dirFile.listFiles();
    for(File oneCheckfile :files ){
      if(PassFileListName.contains("all")){
        listReturn.addAll(getFilesByConfig(oneCheckfile.getAbsolutePath()));
      } else if(PassFileListName.contains(getDomainName(oneCheckfile.getAbsolutePath()))){
        listReturn.addAll(getFilesByConfig(oneCheckfile.getAbsolutePath()));
      }
    }
    return listReturn;
  }

  private List  getFilesByConfig(String dir){
    List listReturn  =  new ArrayList();
    String apiFile =  getPassApiDir(dir);
    String operateFile =  getPassOperateDir(dir);
    String tomcatFile =  getPassTomcatDir(dir);


    if(apiFile != null && !apiFile.isEmpty()){
      listReturn.add(apiFile);
    }
    if(operateFile != null && !operateFile.isEmpty()){
      listReturn.add(operateFile);
    }
    if(tomcatFile != null && !tomcatFile.isEmpty()){
      listReturn.add(tomcatFile);
    }
    return listReturn;
  }

  public String getPassApiDir(String dir){
    if(apilogsList.contains("all")){
      dir+="/apilogs";
    }else if(apilogsList.contains( getDomainName(dir))){
      dir+="/apilogs";
    }else{
      return null;
    }

    return dir;
  }

  public String getPassOperateDir(String dir){
    if(operatelogsList.contains("all")){
      dir+="/operatelogs";
    }else if(operatelogsList.contains( getDomainName(dir))){
      dir+="/operatelogs";
    }else{
      return null;
    }
    return dir;
  }

  public String getPassTomcatDir(String dir){
    if(tomcatlogsList.contains("all")){
      dir+="/tomcatlogs";
    }else if(tomcatlogsList.contains( getDomainName(dir))){
      dir+="/tomcatlogs";
    }else{
      return null;
    }
    return dir;
  }

  private static String getDomainName(String filePath){
    String[] strs = filePath.split("/");
    String domain ;
    domain=strs[strs.length-1];
    if(domain==null || domain.isEmpty()){
      domain=filePath;
    }
    return domain;
  }

  public  List getChildrenFileList(String dir){

      List listFile = new ArrayList();
      File dirFile = new File(dir);
      //如果不是目录文件,则直接返回
      if (dirFile.isDirectory()) {
        //获得文件夹下的文件列表,然后根据文件类型分别处理
        File[] files = dirFile.listFiles();
        if (null != files && files.length > 0) {
          //根据时间排序
          Arrays.sort(files, new Comparator() {
            public int compare(File f1, File f2) {
              return (int) (f1.lastModified() - f2.lastModified());
            }

            public boolean equals(Object obj) {
              return true;
            }
          });
          for (File file : files) {
            //如果不是目录,直接添加
            if (!file.isDirectory()) {
              String oneFileName = file.getName();
              if(match(filenameRegExp,oneFileName)){
                listFile.add(file.getAbsolutePath());
                logger.info("filename:{} is pass",oneFileName);
              }
            } else {
              //对于目录文件,递归调用
              listFile.addAll(getChildrenFileList(file.getAbsolutePath()));
            }
          }
        }
      }else{
        logger.info("FilePath:{} is not Directory",dir);
      }
      return listFile;

  }



  /**
   * @param regex
   * 正则表达式字符串
   * @param str
   * 要匹配的字符串
   * @return 如果str 符合 regex的正则表达式格式,返回true, 否则返回 false;
   */
  private boolean match(String regex, String str) {
    Pattern pattern = Pattern.compile(regex);
    Matcher matcher = pattern.matcher(str);
    return matcher.find();
  }


  private static class ExecRunnable implements Runnable {

    public ExecRunnable(ChannelProcessor channelProcessor,
                        SourceCounter sourceCounter, boolean restart, long restartThrottle,
                        boolean logStderr, int bufferCount, long batchTimeout,
                        Charset charset, String filepath,
                        String tailing, Integer readinterval,
                        boolean startAtBeginning, boolean contextIsJson,
                        Properties prop, String fileWriteJson, Long flushTime,
                        boolean contextIsFlumeLog, String domain) {

      this.channelProcessor = channelProcessor;
      this.sourceCounter = sourceCounter;
      this.restartThrottle = restartThrottle;
      this.bufferCount = bufferCount;
      this.batchTimeout = batchTimeout;
      this.restart = restart;
      this.logStderr = logStderr;
      this.charset = charset;
      this.filepath=filepath;
      this.logfile=new File(filepath);
      this.tailing=tailing;
      if(tailing.equals("all")){
        this.isTailing=true;
      }else{
        this.isTailing = Arrays.asList( tailing.split("\\|")).contains(getDomainName(filepath));
      }
      this.readinterval=readinterval;
      this.startAtBeginning=startAtBeginning;
      this.contextIsJson=contextIsJson;
      this.prop = prop;
      this.fileWriteJson=fileWriteJson;
      this.flushTime=flushTime;
      this.contextIsFlumeLog=contextIsFlumeLog;
      this.domain=getDomain(filepath);
    }



    private final ChannelProcessor channelProcessor;
    private final SourceCounter sourceCounter;
    private volatile boolean restart;
    private final long restartThrottle;
    private final int bufferCount;
    private long batchTimeout;
    private final boolean logStderr;
    private final Charset charset;
    private SystemClock systemClock = new SystemClock();
    private Long lastPushToChannel = systemClock.currentTimeMillis();
    ScheduledExecutorService timedFlushService;
    ScheduledFuture future;
    private String filepath;
    private boolean contextIsJson;
    private Properties prop;
    private long timepoint;
    private String fileWriteJson;
    private Long flushTime;
    private String domain;

    /**
     * 当读到文件结尾后暂停的时间间隔
     */
    private long readinterval = 500;

    /**
     * 设置日志文件
     */
    private File logfile;

    /**
     * 设置是否从头开始读
     */
    private boolean startAtBeginning = false;

    /**
     * 设置tail运行标记
     */
    private String tailing = "all";
    private boolean  isTailing = false;

    private boolean contextIsFlumeLog=false;

    private static String getDomain(String filePath){
      String[] strs = filePath.split("/");
      String domain ;
      domain=strs[strs.length-3];
      if(domain==null || domain.isEmpty()){
        domain=filePath;
      }
      return domain;
    }

    @Override
    public void run() {
      Transaction tt = null;
      try {
        do {
          beginReadLog();
        } while (restart);
      }catch (Exception ex){
        logger.error("=run=>",ex);
        try{
          tt=  Profiler.registerInfo("service", "org.apache.flume.source.ExecTailSource.ExecRunnable.run");
          Profiler.functionError(tt, ex);
        }catch (Exception exp){
          logger.error("=run=>tt error ",exp);
        }
      }finally {
        if(tt!=null){
          Profiler.registerEnd(tt);
        }
      }
    }

    private void beginReadLog(){
        logger.info("=run=> flume tail source run start time:"+new Date().toString());
        timepoint=System.currentTimeMillis();
        Long filePointer = null;
        if (this.startAtBeginning) { //判断是否从头开始读文件
          filePointer =0L;
        } else {
          if(prop!=null || prop.contains(filepath)){

            try {
              filePointer = Long.valueOf((String) prop.get(filepath));
              logger.info("=ExecRunnable.run=>filePointer get from  Properties");
            }catch (Exception ex){
              logger.error("=ExecRunnable.run=>",ex);
              logger.info("=ExecRunnable.run=> error filePointer get from file size");
              filePointer=null;
            }
          }
          if(filePointer ==null){
            filePointer = this.logfile.length(); //指针标识从文件的当前长度开始。
            logger.info("=ExecRunnable.run=>filePointer get from file size");
          }

        }

        final List eventList = new ArrayList();

        timedFlushService = Executors.newSingleThreadScheduledExecutor(
                new ThreadFactoryBuilder().setNameFormat(
                        "timedFlushExecService" +
                                Thread.currentThread().getId() + "-%d").build());
        RandomAccessFile randomAccessFile = null;
        try {

          randomAccessFile= new RandomAccessFile(logfile, "r"); //创建随机读写文件
          future = timedFlushService.scheduleWithFixedDelay(new Runnable() {
                                                              @Override
                                                              public void run() {
                                                                try {
                                                                  synchronized (eventList) {
                                                                    if(!eventList.isEmpty() && timeout()) {
                                                                      flushEventBatch(eventList);
                                                                    }
                                                                  }
                                                                } catch (Exception e) {
                                                                  logger.error("Exception occured when processing event batch", e);
                                                                  if(e instanceof InterruptedException) {
                                                                    Thread.currentThread().interrupt();
                                                                  }
                                                                }
                                                              }
                                                            },
                  batchTimeout, batchTimeout, TimeUnit.MILLISECONDS);

          while (this.isTailing) {
            long fileLength = this.logfile.length();
            if (fileLength < filePointer) {
              randomAccessFile = new RandomAccessFile(logfile, "r");
              filePointer = 0l;
            }
            if (fileLength > filePointer) {
              randomAccessFile.seek(filePointer);
              String line = randomAccessFile.readLine();
              if(line!=null){
                line = new String(line.getBytes(ExecTailSourceConfigurationConstants.CHARSET_RANDOMACCESSFILE),charset);
                line = line.replaceAll("\"","\'");
              }

              while (line != null) {

                //送channal
                synchronized (eventList) {
                  sourceCounter.incrementEventReceivedCount();


                  String bodyjson = "";
                  if (!contextIsJson) {
                    bodyjson = MsgBuildeJson.buildeJson(contextIsFlumeLog,line,filepath,domain);
                    if(bodyjson.indexOf("{")>0){
                      bodyjson = bodyjson.substring(bodyjson.indexOf("{"),bodyjson.length());
                    }
                  }else{
                    bodyjson = MsgBuildeJson.changeDomain(line.toString(),domain);
                  }

                  Event oneEvent = EventBuilder.withBody(bodyjson.getBytes(charset));
                  oneEvent.getHeaders().put("UlogDomain",domain);
                  oneEvent.getHeaders().put("topic","file_channel_topic_"+domain);
                  eventList.add(oneEvent);
                  if (eventList.size() >= bufferCount || timeout()) {
                    flushEventBatch(eventList);
                  }
                }

                //读下一行
                line = randomAccessFile.readLine();
                if(line!=null){
                  line = new String(line.getBytes(ExecTailSourceConfigurationConstants.CHARSET_RANDOMACCESSFILE),charset);
                  line = line.replaceAll("\"","\'");
                }

                try{
                  Cat.logMetricForCount("count_file_readline_"+domain);
                }catch (Exception ex){
                  logger.error("=run=>count_file_readline_"+domain,ex);
                }

                try {
                  Long nowfilePointer = randomAccessFile.getFilePointer();
                  if (!nowfilePointer.equals(filePointer)) {
                    filePointer = nowfilePointer;
                    if (System.currentTimeMillis() - timepoint > flushTime) {
                      timepoint = System.currentTimeMillis();
                      prop.setProperty(filepath, filePointer.toString());
                      FileOutputStream fos = new FileOutputStream(fileWriteJson);
                      if (fos != null) {
                        prop.store(fos, "Update '" + filepath + "' value");
                      }
                      fos.close();

                    }
                  }
                }catch(Exception ex){
                  ex.printStackTrace();
                }
              }

            }
            Thread.sleep(this.readinterval);
          }

          synchronized (eventList) {
            if(!eventList.isEmpty()) {
              flushEventBatch(eventList);
            }
          }

        } catch (Exception e) {
          logger.error("Failed while running filpath: " + filepath, e);
          if(e instanceof InterruptedException) {
            Thread.currentThread().interrupt();
          }
        } finally {

          if(randomAccessFile!=null){
            try {
              randomAccessFile.close();
            } catch (IOException ex) {
              logger.error("Failed to close reader for ExecTail source", ex);
            }
          }

        }
        logger.info("=run=> flume tail source run restart:"+restart);
        if(restart) {
          logger.info("=run=> flume tail source run restart time:"+new Date().toString());
          logger.info("Restarting in {}ms", restartThrottle);
          try {
            Thread.sleep(restartThrottle);
          } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
          }
        } else {
          logger.info("filepath [" + filepath + "] exited with restart[" + restart+"]");
        }

    }

    private void flushEventBatch(List eventList){
      channelProcessor.processEventBatch(eventList);
      sourceCounter.addToEventAcceptedCount(eventList.size());
      eventList.clear();
      lastPushToChannel = systemClock.currentTimeMillis();

      try{
        Cat.logMetricForCount("count_file_flushEventBatch_"+domain);
      }catch (Exception ex){
        logger.error("=run=>count_file_flushEventBatch_"+domain,ex);
      }
    }

    private HashMap ParseFlumeLog(String log,HashMap logMap){
      String[] strLogs = log.split("\\|");
      logMap.put("className",strLogs[0]);
      logMap.put("methodName",strLogs[1]);
      logMap.put("level",strLogs[2]);
      logMap.put("treeId",strLogs[3]);
      logMap.put("requestId",strLogs[4]);
      logMap.put("transactionId",strLogs[5]);
      return logMap;
    }

    private boolean timeout(){
      return (systemClock.currentTimeMillis() - lastPushToChannel) >= batchTimeout;
    }

    private static String[] formulateShellCommand(String shell, String command) {
      String[] shellArgs = shell.split("\\s+");
      String[] result = new String[shellArgs.length + 1];
      System.arraycopy(shellArgs, 0, result, 0, shellArgs.length);
      result[shellArgs.length] = command;
      return result;
    }

    public int kill() {
      logger.info("=kill=> flume tail source kill start time:"+new Date().toString());
      this.tailing="no";
      synchronized (this.getClass()) {
        try {
          // Stop the Thread that flushes periodically
          if (future != null) {
            future.cancel(true);
          }

          if (timedFlushService != null) {
            timedFlushService.shutdown();
            while (!timedFlushService.isTerminated()) {
              try {
                timedFlushService.awaitTermination(500, TimeUnit.MILLISECONDS);
              } catch (InterruptedException e) {
                logger.debug("Interrupted while waiting for ExecTail executor service "
                        + "to stop. Just exiting.");
                Thread.currentThread().interrupt();
              }
            }
          }
          logger.info("=kill=> flume tail source kill end time:" + new Date().toString());
          return Integer.MIN_VALUE;
        } catch (Exception ex) {
          logger.error("=kill=>", ex);
          Thread.currentThread().interrupt();
        }
      }
      logger.info("=kill=> flume tail source kill end time:"+new Date().toString());
      return Integer.MIN_VALUE / 2;
    }
    public void setRestart(boolean restart) {
      this.restart = restart;
    }
  }
  private static class StderrReader extends Thread {
    private BufferedReader input;
    private boolean logStderr;

    protected StderrReader(BufferedReader input, boolean logStderr) {
      this.input = input;
      this.logStderr = logStderr;
    }



    @Override
    public void run() {
      try {
        int i = 0;
        String line = null;
        while((line = input.readLine()) != null) {
          if(logStderr) {
            // There is no need to read 'line' with a charset
            // as we do not to propagate it.
            // It is in UTF-16 and would be printed in UTF-8 format.
            logger.info("StderrLogger[{}] = '{}'", ++i, line);
          }
        }
      } catch (IOException e) {
        logger.info("StderrLogger exiting", e);
      } finally {
        try {
          if(input != null) {
            input.close();
          }
        } catch (IOException ex) {
          logger.error("Failed to close stderr reader for ExecTail source", ex);
        }
      }
    }
  }
}



你可能感兴趣的:(flume,source,elasticsearch,flume高并发优化,扩展,flume,kakfa,elasticsearch)