• flume自定义sink之文件


    package me;
    
    import java.io.File;
    import java.io.FileNotFoundException;
    import java.io.FileOutputStream;
    import java.io.IOException;
    
    import org.apache.flume.Channel;
    import org.apache.flume.Context;
    import org.apache.flume.Event;
    import org.apache.flume.EventDeliveryException;
    import org.apache.flume.Transaction;
    import org.apache.flume.conf.Configurable;
    import org.apache.flume.sink.AbstractSink;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    
    public class MySinks extends AbstractSink implements Configurable {
        private static final Logger logger = LoggerFactory.getLogger(MySink.class);
        private static final String PROP_KEY_ROOTPATH = "fileName";
        private String fileName;
    
        @Override
        public void configure(Context context) {
            // TODO Auto-generated method stub
            fileName = context.getString(PROP_KEY_ROOTPATH);
        }
    
        @Override
        public Status process() throws EventDeliveryException {
            // TODO Auto-generated method stub
            Channel ch = getChannel();
            Transaction txn = ch.getTransaction();
            Event event =null;
            txn.begin();
            while(true){
                event = ch.take();
                if (event!=null) {
                    break;
                }
            }
            try {
    
                logger.debug("Get event.");
    
                String body = new String(event.getBody());
                System.out.println("event.getBody()-----" + body);
                
                String res = body + ":" + System.currentTimeMillis() + "
    ";
                File file = new File(fileName);
                FileOutputStream fos = null;
                try {
                    fos = new FileOutputStream(file, true);
                } catch (FileNotFoundException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                try {
                    fos.write(res.getBytes());
                } catch (IOException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                try {
                    fos.close();
                } catch (IOException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                txn.commit();
                return Status.READY;
            } catch (Throwable th) {
                txn.rollback();
    
                if (th instanceof Error) {
                    throw (Error) th;
                } else {
                    throw new EventDeliveryException(th);
                }
            } finally {
                txn.close();
            }
    
        }
    
    }
    
    agent.sources = r1
    agent.sinks = k1
    agent.channels = c1
    # Describe/configure the source
    agent.sources.r1.type = netcat  
    agent.sources.r1.bind = localhost  
    agent.sources.r1.port = 5678  
    agent.sources.r1.channels = c1  
    # Describe the sink
    agent.sinks.k1.type = me.MySinks
    agent.sinks.k1.fileName=/Users/lihu/Desktop/crawla/mysinks.txt
    # Use a channel which buffers events in memory
    agent.channels.c1.type = memory
    agent.channels.c1.capacity = 1000
    agent.channels.c1.transactionCapacity = 100
    # Bind the source and sink to the channel
    agent.sources.r1.channels = c1
    agent.sinks.k1.channel = c1
    

     

  • 相关阅读:
    (转)Computer Vision Open Source Algorithm Implementations
    关于项目依赖项
    fatal error CVT1100: duplicate resource. type:MANIFEST, name:1, language:0×0409
    转:error LNK2005
    C++ string类常用函数
    [转载]同步synchronized方法和代码块
    Java语言基础常用对象API(二)泛型、Map集合
    Java语言基础IO流(输入输出流) 字符流
    Java语言基础常用对象API(二)集合框架
    Java语言基础IO流(输入输出流) 字节流、转换流
  • 原文地址:https://www.cnblogs.com/sunyaxue/p/6645753.html
Copyright © 2020-2023  润新知