用MapReduce把hdfs数据写入HBase中

1.使用Map+Reduce方式

public class MapReduceImport {

/**
* Mapper
*/
static class HMapper extends Mapper<LongWritable, Text, LongWritable, Text> {
Text v2 = new Text();

protected void map(LongWritable key, Text value, Context context) throws java.io.IOException, InterruptedException {
String[] splited = value.toString().split(" ");
if (splited.length != 6)//清洗不符合标准的数据
return;
try {
//GetRowKey.getRowKeyString方法是自己定义生成rowkey的方法
//rowkey设计为IP_TimeStamp这种方式
v2.set(GetRowKey.getRowKeyString(splited[2], splited[4]) + " " + value.toString());
context.write(key, v2);
} catch (NumberFormatException e) {
System.out.println("出错了" + e.getMessage());
}
}
}

/**
* Reducer
*/
static class HReducer extends TableReducer<LongWritable, Text, NullWritable> {

protected void reduce(LongWritable key, java.lang.Iterable<Text> values, Context context) throws java.io.IOException, InterruptedException {
for (Text text : values) {
String[] splited = text.toString().split(" ");

Put put = new Put(Bytes.toBytes(splited[0]));

for (int j = 1; j < splited.length; j++) {
put.addColumn(Bytes.toBytes(HConfiguration.colFamily), Bytes.toBytes("log" + j), Bytes.toBytes(splited[j]));
}

context.write(NullWritable.get(), put);
}
}
}

/**
* Main
*
* @param args
* @throws Exception
*/
public static void main(String[] args) throws Exception {

Configuration configuration = new Configuration();
//设置zookeeper
configuration.set("hbase.zookeeper.quorum", HConfiguration.hbase_zookeeper_quorum);
configuration.set("hbase.zookeeper.property.clientPort", "2181");
//设置hbase表名称
configuration.set(TableOutputFormat.OUTPUT_TABLE, HConfiguration.tableName);
//将该值改大，防止hbase超时退出
configuration.set("dfs.socket.timeout", "180000");

MRDriver myDriver = MRDriver.getInstance();

try {
myDriver.createTableIfExistDelete(HConfiguration.tableName, HConfiguration.colFamily);
} catch (Exception e) {
e.printStackTrace();
}

Job job = new Job(configuration, "Map+ReduceImport");

job.setMapperClass(HMapper.class);
job.setReducerClass(HReducer.class);

job.setMapOutputKeyClass(LongWritable.class);
job.setMapOutputValueClass(Text.class);

job.setInputFormatClass(TextInputFormat.class);
//不再设置输出路径，而是设置输出格式类型TableOutputFormat
job.setOutputFormatClass(TableOutputFormat.class);

FileInputFormat.setInputPaths(job, HConfiguration.mapreduce_inputPath);

job.waitForCompletion(true);
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

———————-分—-割—-线————————–

2.只使用Map的方式

public class OnlyMapImport {
/**
* Mapper
*/
static class ImportMapper extends Mapper<LongWritable, Text, ImmutableBytesWritable, Put> {

@Override
public void map(LongWritable offset, Text value, Context context) {
String[] splited = value.toString().split(" ");
if (splited.length != 6)
return;
try {
//GetRowKey.getRowKeyString方法是自己定义生成rowkey的方法
//rowkey设计为IP_TimeStamp这种方式
byte[] rowkey = Bytes.toBytes(GetRowKey.getRowKeyString(splited[2], splited[4]));
Put put = new Put(rowkey);

for (int j = 0; j < splited.length; j++) {
put.addColumn(Bytes.toBytes(HConfiguration.colFamily), Bytes.toBytes("log" + j), Bytes.toBytes(splited[j]));
}

context.write(new ImmutableBytesWritable(rowkey), put);

} catch (NumberFormatException e) {
System.out.println("出错了" + e.getMessage());
} catch (IOException e) {
e.printStackTrace();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}

/**
* Main
*
* @param args
* @throws Exception
*/
public static void main(String[] args) throws Exception {

Configuration configuration = new Configuration();
//设置zookeeper
configuration.set("hbase.zookeeper.quorum", HConfiguration.hbase_zookeeper_quorum);
configuration.set("hbase.zookeeper.property.clientPort", "2181");
//设置hbase表名称
configuration.set(TableOutputFormat.OUTPUT_TABLE, HConfiguration.tableName);
//将该值改大，防止hbase超时退出
configuration.set("dfs.socket.timeout", "180000");

MRDriver myDriver = MRDriver.getInstance();

try {
myDriver.createTableIfExistDelete(HConfiguration.tableName, HConfiguration.colFamily);
} catch (Exception e) {
e.printStackTrace();
}

Job job = new Job(configuration, "HBaseBatchImport");

job.setJarByClass(OnlyMapImport.class);
job.setMapperClass(ImportMapper.class);
//设置map的输出，不设置reduce的输出类型
job.setMapOutputKeyClass(ImmutableBytesWritable.class);
job.setMapOutputValueClass(Writeable.class);
job.setNumReduceTasks(0);

job.setInputFormatClass(TextInputFormat.class);
//不再设置输出路径，而是设置输出格式类型
job.setOutputFormatClass(TableOutputFormat.class);

FileInputFormat.setInputPaths(job, HConfiguration.mapreduce_inputPath);

job.waitForCompletion(true);
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

经过测试，导入时间明显减少。