业务需求说明,由于公司数据中心处于刚开始部署的阶段,这需要涉及其它部分将数据全部汇总到数据中心,这实现的方式是同上传json文件,通过采用socket&serversocket实现传输。
其中,服务端采用多线程的方式,实现多用户传输的目的。并且实现可以将数据写入到hbase中。
具体步骤如下:
1、首先编写客户端的代码:
package com.yiban.datacenter.ToHbaseFromJson;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.Socket;
import java.net.UnknownHostException;
public class hbaseclient {
public static void main(String[] args) {
// TODO Auto-generated method stub
try {
// 创建客户端的socket
Socket s = new Socket("192.168.27.47", 22222);
// 首先确认连接上了我的服务器,通过接受服务器发送的确认信息
BufferedWriter firstclientwrite = new BufferedWriter(
new OutputStreamWriter(s.getOutputStream()));
firstclientwrite.write("我准备向你发送数据了,你准备好接收了吗?");
firstclientwrite.newLine();
firstclientwrite.flush();
// 经通道内的字节输入流进行一个封装程字符流,方便直接输出
BufferedReader testconnection = new BufferedReader(
new InputStreamReader(s.getInputStream()));
// 输出结果
String sss = testconnection.readLine();
System.out.println(sss);
//发送表名和列族名
firstclientwrite.write("nihao");
firstclientwrite.newLine();
firstclientwrite.flush();
//确定表发送是否成功
System.out.println(testconnection.readLine());
// 封装客户端的文本文件
BufferedReader clientread = new BufferedReader(new FileReader(
"file.json"));
// 准备将客户端的字符流写入到对应的通道内,OutputStreamWriter是将字符流转换成字节流,
// BufferedWriter封装字符流
BufferedWriter clientwirte = new BufferedWriter(
new OutputStreamWriter(s.getOutputStream()));
String line = null;
while ((line = clientread.readLine()) != null) {
clientwirte.write(line);
clientwirte.newLine();
clientwirte.flush();
}
// 提示发送完成
s.shutdownOutput();
// 准备接收一个反馈
// 经通道内的字节输入流进行一个封装程字符流,方便直接输出
BufferedReader ctread = new BufferedReader(new InputStreamReader(
s.getInputStream()));
// 输出结果
String fackcall = null;
while ((fackcall = ctread.readLine()) != null) {
System.out.println(fackcall);
}
// 释放资源
clientread.close();
s.close();
} catch (UnknownHostException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
2、服务端的代码:
(1)线程类的实现
package com.yiban.datacenter.ToHbaseFromJson;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
import java.util.NavigableMap;
import java.util.Map.Entry;
import net.sf.json.JSONArray;
import net.sf.json.JSONObject;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.HColumnDescriptor;
import org.apache.hadoop.hbase.HConstants;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.MasterNotRunningException;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.ZooKeeperConnectionException;
import org.apache.hadoop.hbase.client.Admin;
import org.apache.hadoop.hbase.client.Connection;
import org.apache.hadoop.hbase.client.ConnectionFactory;
import org.apache.hadoop.hbase.client.HBaseAdmin;
import org.apache.hadoop.hbase.client.HTable;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.client.Table;
import org.apache.hadoop.hbase.util.Bytes;
public class UserThread implements Runnable {
private String testconnect = "我准备向你发送数据了,你准备好接收了吗?"; // 这个可以用来验证用户名和密码
private static Configuration conf = HBaseConfiguration.create();
private static Connection connection = null;
// 配置信息
static {
try {
conf.set(HConstants.ZOOKEEPER_QUORUM, "192.168.27.233");
conf.setInt(HConstants.ZOOKEEPER_CLIENT_PORT, 2181);
connection = ConnectionFactory.createConnection(conf);
} catch (Exception e) {
// TODO: handle exception
e.printStackTrace();
}
}
private Socket s;
public UserThread(Socket s) {
this.s = s;
}
private String userTableName = null;
private String columnFamilyName = null;
@Override
public void run() {
// TODO Auto-generated method stub
try {
// 将通道内的字节流转换成字符流,并用bufferedreader进行封装,InputStreamReader是将字节流转换成字符流
BufferedReader serverread = new BufferedReader(
new InputStreamReader(s.getInputStream()));
// 询问客户端连接是否准备好,接受客户端的连接请求
String line = serverread.readLine(); // 阻塞
System.out.println(line);// 输出客户端的连接请求
// 将通道内的字符写入到对应的文件中,利用bufferedwrite进行封装,FileWriter是将字符流写入到文件中
BufferedWriter serverwrite = new BufferedWriter(
new OutputStreamWriter(s.getOutputStream()));
if (line.equals(testconnect)) {
serverwrite.write("连接成功,你可以发送数据了,发送数据前,请先发送你要用的数据库表名!");
serverwrite.newLine();
serverwrite.flush();
} else {
serverwrite.write("连接失败!");
serverwrite.newLine();
serverwrite.flush();
}
// 准备接收表名和列族名
userTableName = serverread.readLine();
System.out.println("tablename:" + userTableName);// 输出客户端的连接请求
// 告诉客户端,我接受成功
if (TableIsExist(userTableName)) {
serverwrite.write("接收表名成功");
serverwrite.newLine();
serverwrite.flush();
} else {
serverwrite.write("表不存在");
serverwrite.newLine();
serverwrite.flush();
}
// 循环读取客户端的数据
line = "";
StringBuffer temp = new StringBuffer(line);
while ((line = serverread.readLine()) != null) {
temp.append(line);
}
// 对json文件进行解析
JSONArray jsonArray = JSONArray.fromObject(temp.toString());
// 解析之后进行输出,在这里可以直接写入到hbase中
PrintJsonArray(jsonArray);
getAllTables(conf);
// 将接收到的数据写入hbase中的表中
insertData(jsonArray, userTableName);
// 给出一个反馈,提示数据上传成功
// 封装通道内的输出流,方便对他进行写字符数据
BufferedWriter bwserver = new BufferedWriter(
new OutputStreamWriter(s.getOutputStream()));
bwserver.write("文件上传成功!");
// bwserver.newLine();
bwserver.flush();
bwserver.close();
// 释放资源
// serverwrite.close();
s.close();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
@SuppressWarnings("deprecation")
private void insertData(JSONArray jsonArray, String userTableName) {
// TODO Auto-generated method stub
Table table = null;
try {
table = connection.getTable(TableName.valueOf(userTableName));
} catch (IOException e1) {
// TODO Auto-generated catch block
e1.printStackTrace();
}
List<Put> putlist = new ArrayList<Put>();
for (int i = 0; i < jsonArray.size(); i++) {
JSONObject jsonobject = jsonArray.getJSONObject(i);
Put put = new Put(Bytes.toBytes(jsonobject.getString("DocumentID")));// 指定行,也就是键值
// 参数分别:列族、列、值
put.add(Bytes.toBytes("info"),
Bytes.toBytes("DocumentContent"),
Bytes.toBytes(jsonobject.getString("DocumentContent")));
putlist.add(put);
}
try {
table.put(putlist);
table.close();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
private boolean TableIsExist(String userTableName2) {
boolean flag = false;
try {
// Connection connection = ConnectionFactory.createConnection(conf);
Admin ad = connection.getAdmin();
if (ad.tableExists(TableName.valueOf(userTableName2))) {
flag = true;
System.out.println("表存在");
} else {
System.out.println("表不存在");
}
} catch (Exception e) {
// TODO: handle exception
}
return flag;
// TODO Auto-generated method stub
}
private void PrintJsonArray(JSONArray jsonArray) {
int size = jsonArray.size();
System.out.println("Size: " + size);
for (int i = 0; i < size; i++) {
JSONObject jsonObject = jsonArray.getJSONObject(i);
System.out.println("[" + i + "]id=" + jsonObject.get("id"));
System.out.println("[" + i + "]name=" + jsonObject.get("name"));
System.out.println("[" + i + "]role=" + jsonObject.get("role"));
}
}
// create table
private void createTable(Configuration conf) {
// HBaseAdmin ha=new HBaseAdmin(conf);
try {
// Connection connection = ConnectionFactory.createConnection(conf);
Table table = connection.getTable(TableName.valueOf(userTableName));
Admin ad = connection.getAdmin();
// TableName name= TableName.valueOf(Bytes.toBytes(tablename));//表名
HTableDescriptor desc = new HTableDescriptor(table.getName());
HColumnDescriptor family = new HColumnDescriptor(
Bytes.toBytes(columnFamilyName));// 列簇
desc.addFamily(family);
ad.createTable(desc);
ad.close();
} catch (Exception e) {
// TODO: handle exception
e.printStackTrace();
}
}
// Hbase获取所有的表信息
public static List getAllTables(Configuration conf)
throws MasterNotRunningException, ZooKeeperConnectionException,
IOException {
HBaseAdmin ad = new HBaseAdmin(conf);
List<String> tables = null;
if (ad != null) {
try {
HTableDescriptor[] allTable = ad.listTables();
if (allTable.length > 0)
tables = new ArrayList<String>();
for (HTableDescriptor hTableDescriptor : allTable) {
tables.add(hTableDescriptor.getNameAsString());
System.out.println(hTableDescriptor.getNameAsString());
}
} catch (IOException e) {
e.printStackTrace();
}
}
return tables;
}
// 按顺序输出
public void printResult(Result rs) {
if (rs.isEmpty()) {
System.out.println("result is empty!");
return;
}
// new API and print Map of families to all versions of its qualifiers
// and values.
NavigableMap<byte[], NavigableMap<byte[], NavigableMap<Long, byte[]>>> temps = rs
.getMap();
String rowkey = Bytes.toString(rs.getRow()); // actain rowkey
System.out.println("rowkey->" + rowkey);
for (Entry<byte[], NavigableMap<byte[], NavigableMap<Long, byte[]>>> temp : temps
.entrySet()) {
System.out.print("\tfamily->" + Bytes.toString(temp.getKey()));
for (Entry<byte[], NavigableMap<Long, byte[]>> value : temp
.getValue().entrySet()) {
System.out.print("\tcol->" + Bytes.toString(value.getKey()));
for (Entry<Long, byte[]> va : value.getValue().entrySet()) {
System.out.print("\tvesion->" + va.getKey());
System.out.print("\tvalue->"
+ Bytes.toString(va.getValue()));
System.out.println();
}
}
}
}
}
(2)主函数的实现
package com.yiban.datacenter.ToHbaseFromJson;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.HConstants;
public class HbaseServer {
public static void main(String[] args) {
// TODO Auto-generated method stub
try {
@SuppressWarnings("resource")
ServerSocket ss=new ServerSocket(22222);
while(true){
Socket s=ss.accept();
new Thread(new UserThread3(s)).start();
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}