package com.pb;
import com.pb.common.Executable;
import org.apache.kafka.clients.producer.KafkaProducer;
import org.apache.kafka.clients.producer.Producer;
import org.apache.kafka.clients.producer.ProducerRecord;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Properties;
public class ReadCSV implements Executable {
///读csv文件打到kafka的topic
@Override
public void execute(String …parameters) {
// public void execute(String[] parameters) {
Properties props = new Properties(); props.put(“bootstrap.servers”,ReadUtil.getProperties(“brokerUrl”));
props.put(“acks”, ReadUtil.getProperties(“acks”));
props.put(“retries”, ReadUtil.getProperties(“retries”));
props.put(“batch.size”, ReadUtil.getProperties(“batch.size”));
props.put(“linger.ms”, ReadUtil.getProperties(“linger.ms”));
props.put(“buffer.memory”, ReadUtil.getProperties(“buffer.memory”));
props.put(“key.serializer”, ReadUtil.getProperties(“key.serializer”));
props.put(“value.serializer”, ReadUtil.getProperties(“value.serializer”));
FileInputStream csvFile = null;
InputStreamReader is = null;
BufferedReader br = null;
Producer<String, String> producer = new KafkaProducer<String, String>(props);
try {
csvFile = new FileInputStream(“users.csv”);
is = new InputStreamReader(csvFile);
br = new BufferedReader(is);
String line = br.readLine();
Long position = 0L;
int i = 0;
while (line != null) {
position += line.length();
producer.send(new ProducerRecord<String, String>(“streams-plaintext-input”, Long.toString(position), line));
line = br.readLine();
i++;
}
System.out.println(i);
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
csvFile.close();
is.close();
br.close();
producer.flush();
producer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}}
驱动
main
package com.pb.main;
import com.pb.common.Executable;
import java.util.Arrays;
public class Start {
public static void main(String[] args) throws Exception{
//public void run(String… args) {
String className=args[0];
//
String[] parameters= Arrays.copyOfRange(args,1,args.length);
//
Object object=Class.forName(className).newInstance();
if(object instanceof Executable){
((Executable) object).execute();
// }
for(String str :parameters){
System.out.println(str);
}
}
}
}
接口
package com.pb.common;
public interface Executable {
// void execute(String[] parameters) throws Exception;
//void execute() throws Exception;
void execute(String …parameters) throws Exception;
}