Flume(ng) 自定义sink实现和属性注入

转自：http://www.aboutyun.com/thread-7884-1-1.html

问题导读：
1.如何实现flume端自定一个sink，来按照我们的规则来保存日志？
2.想从flume的配置文件中获取rootPath的值，该如何配置？






最近需要利用flume来做收集远端日志，所以学习一些flume最基本的用法。这里仅作记录。

远端日志收集的整体思路是远端自定义实现log4j的appender把消息发送到flume端，flume端自定义实现一个sink来按照我们的规则保存日志。

自定义Sink代码：

1. public class LocalFileLogSink extends AbstractSink implements Configurable {
   
2.      private static final Logger logger = LoggerFactory
   
3.               . getLogger(LocalFileLogSink .class );
   
4.             private static final String PROP_KEY_ROOTPATH = "rootPath";
   
5.       private String rootPath;
   
6.      @Override
   
7.      public void configure(Context context) {
   
8.           String rootPath = context.getString(PROP_KEY_ROOTPATH );
   
9.           setRootPath(rootPath);
   
10.      }
    
11.            
    
12.           @Override
    
13.           public Status process() throws EventDeliveryException {
    
14.            logger .debug("Do process" );
    
15.        ｝
    
16. }

复制代码

实现Configurable接口，即可在初始化时，通过configure方法从context中获取配置的参数的值。这里，我们是想从flume的配置文件中获取rootPath的值，也就是日志保存的根路径。在flume-conf.properties中配置如下：

1. agent.sinks = loggerSink
   
2. agent.sinks.loggerSink.rootPath = ./logs

复制代码

loggerSink是自定义sink的名称，我们取值时的key，只需要loggerSink后面的部分即可，即这里的rootPath。

实际业务逻辑的执行，是通过继承复写AbstractSink中的process方法实现。从基类的getChannel方法中获取信道，从中取出Event处理即可。

1. Channel ch = getChannel();
   
2.            Transaction txn = ch.getTransaction();
   
3.          txn.begin();
   
4.           try {
   
5.               logger .debug("Get event." );
   
6.              Event event = ch.take();
   
7.              txn.commit();
   
8.              status = Status. READY ;
   
9.              return status;
   
10.                    ｝finally {
    
11.              Log. info( "trx close.");
    
12.              txn.close();
    
13.          }

复制代码