package com.kafka.myparitioner;

import kafka.producer.Partitioner;

import kafka.utils.VerifiableProperties;

//设定依据key将当前这条消息发送到哪个partition的规则

public class CidPartitioner implements Partitioner {

    public CidPartitioner(VerifiableProperties props) {

          //注意 ： 构造函数的函数体没有东西，但是不能没有构造函数

    }  

    @Override

    public int partition(Object key, int numPartitions) {

        try {

            long partitionNum = Long.parseLong((String) key);

            return (int) Math.abs(partitionNum % numPartitions);

        } catch (Exception e) {

            return Math.abs(key.hashCode() % numPartitions);

        }

    }

}

import java.io.IOException;

import java.io.InputStream;

import java.net.ServerSocket;

import java.net.Socket;

import java.util.Properties;

import java.util.regex.Pattern;

import kafka.producer.KeyedMessage;

import kafka.producer.ProducerConfig;

//与KafkaReceiverLTELogSocket的区别在于，指定了消息的partition分配规则

public class KafkaReceiveLTELogSocketPartition extends Thread{

    //按照一定的时间间隔发送LTE信令数据

    String regEx ="[^0-9.\\+\\-\\s+\\,E]";

    Pattern p = Pattern.compile(regEx); 

    //第一个类型代表key的类型，第二个代表消息的类型

    private final kafka.javaapi.producer.Producer<String, String> producer;

    private final String topic;

    private final Properties props = new Properties();

    private final int port = 12345; 

    public KafkaReceiveLTELogSocketPartition(String topic) {

        props.put("serializer.class", "kafka.serializer.StringEncoder");

        props.put("metadata.broker.list", "192.168.1.164:9093"); // 配置kafka端口

        props.put("partitioner.class","com.kafka.myparitioner.CidPartitioner");

        //props.put("zk.connect", "192.168.1.164:2181");//连接zk,新的版本好像不需要

        producer = new kafka.javaapi.producer.Producer<String, String>(new ProducerConfig(props));

        this.topic = topic;

    }

    public void receiveAndWrite2(String outputFileName , int port) throws IOException{

        ServerSocket serverSocket = new ServerSocket(port);

        Socket socket = serverSocket.accept();

        StringBuilder sb = new StringBuilder();

        try{

            while(true){

                InputStream istream = socket.getInputStream();

                int count = 0;

                while (count == 0) {

                    count = istream.available();

                }

                byte[] b = new byte[count];

                istream.read(b);

                for(int i = 0 ; i < count ; i ++){

                    if(b[i]=='\n'){ //当遇到流中的换行符时，说明已经获取一条完整的信息，发送

                        String str = sb.toString();

                        //获取key_cid_str

                        String key_cid_str = str.substring(str.indexOf(":")+1, str.indexOf(","));

                        System.out.println("接收长度："+str.length());

                        System.out.println(str);

                        //第一个参数代表key的类型，第二个参数代表message的类型

                        producer.send(new KeyedMessage<String, String>(topic,key_cid_str,str));

                        sb = new StringBuilder();

                    }else{

                        sb.append(Character.toChars(b[i]));

                    }

                }

            }

        }finally{

            // 关闭socket,不要再while中关闭，否则发送方每次都要重建连接

            socket.close();

            serverSocket.close();

        }

    }

    @Override

    public void run() {

        String filename = "JSON1_Yanming_DriveTesting_09-04.16-17.16-27_TIME.json";

        String outputFileName  = ""+filename;

        try {

            receiveAndWrite2(outputFileName,port);

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    public static void main(String[] args) {

        String topic = "kafka_flume_topic";

        new KafkaReceiveLTELogSocketPartition(topic).start();

    }

}

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Properties;

import kafka.consumer.ConsumerConfig;

import kafka.consumer.ConsumerIterator;

import kafka.consumer.KafkaStream;

import kafka.javaapi.consumer.ConsumerConnector;

import kafka.message.MessageAndMetadata;

public class KafkaConsumer extends Thread {

    private final ConsumerConnector consumer;

    private final String topic;

    public KafkaConsumer(String topic) {

        consumer = kafka.consumer.Consumer.createJavaConsumerConnector(createConsumerConfig());

        this.topic = topic;

    }

    private static ConsumerConfig createConsumerConfig() {

        Properties props = new Properties();

        props.put("zookeeper.connect", "192.168.1.164:2181"); // zookeeper的地址

        props.put("group.id", "group2"); // 组ID

        //zk连接超时

        props.put("zookeeper.session.timeout.ms", "40000");

        props.put("zookeeper.sync.time.ms", "200");

        props.put("auto.commit.interval.ms", "1000");

        return new ConsumerConfig(props);

    }

    @Override

    public void run() {

        Map<String, Integer> topicCountMap = new HashMap<String, Integer>();

        //设定每个topic开几个线程

        topicCountMap.put(topic, new Integer(1));

        Map<String, List<KafkaStream<byte[], byte[]>>> consumerMap     = consumer.createMessageStreams(topicCountMap);

        KafkaStream<byte[], byte[]> stream = consumerMap.get(topic).get(0);

        ConsumerIterator<byte[], byte[]> it = stream.iterator();

        while (it.hasNext()) {

            MessageAndMetadata<byte[], byte[]> message = it.next();

            String topic = message.topic();

            int partition = message.partition();

            long offset = message.offset();

            String key = new String(message.key());

            String msg = new String(message.message());

            // 在这里处理消息,这里仅简单的输出

            // 如果消息消费失败，可以将已上信息打印到日志中，活着发送到报警短信和邮件中，以便后续处理

            System.out.println( " thread : " + Thread.currentThread().getName()

                    + ", topic : " + topic + ", partition : " + partition + ", offset : " + offset + " , key : "

                    + key + " , mess : " + msg);

        }

    }

}