1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase;
19
20 import java.util.ArrayList;
21 import java.util.Arrays;
22 import java.util.List;
23
24 import org.apache.hadoop.hbase.io.hfile.HFile;
25 import org.apache.hadoop.hbase.util.LoadTestDataGeneratorWithTags;
26 import org.apache.hadoop.hbase.util.LoadTestTool;
27 import org.junit.experimental.categories.Category;
28
29 @Category(IntegrationTests.class)
30 public class IntegrationTestIngestWithTags extends IntegrationTestIngest {
31
32 private static final char COLON = ':';
33
34 private int minTagsPerKey = 1, maxTagsPerKey = 10;
35 private int minTagLength = 16, maxTagLength = 512;
36
37 @Override
38 public void setUpCluster() throws Exception {
39 getTestingUtil(conf).getConfiguration().setInt(HFile.FORMAT_VERSION_KEY, 3);
40 super.setUpCluster();
41 }
42
43 @Override
44 protected String[] getArgsForLoadTestTool(String mode, String modeSpecificArg, long startKey,
45 long numKeys) {
46 String[] args = super.getArgsForLoadTestTool(mode, modeSpecificArg, startKey, numKeys);
47 List<String> tmp = new ArrayList<String>(Arrays.asList(args));
48
49 tmp.add(HIPHEN + LoadTestTool.OPT_GENERATOR);
50 StringBuilder sb = new StringBuilder(LoadTestDataGeneratorWithTags.class.getName());
51 sb.append(COLON);
52 sb.append(minTagsPerKey);
53 sb.append(COLON);
54 sb.append(maxTagsPerKey);
55 sb.append(COLON);
56 sb.append(minTagLength);
57 sb.append(COLON);
58 sb.append(maxTagLength);
59 tmp.add(sb.toString());
60 return tmp.toArray(new String[tmp.size()]);
61 }
62 }