protected void setup(Context context) throws IOException, InterruptedException {
super.setup(context);
Path pt = new Path("/user/yao/query/query");
FileSystem fs = FileSystem.get(new Configuration());
BufferedReader br = new BufferedReader(new InputStreamReader(fs.open(pt)));
String line = br.readLine();
String[] keywords = line.split(",");
k0 = keywords[0];
k1 = keywords[1];
k2 = keywords[2];
br.close();
}
public static List<String> readConfig(Path path, Configuration conf) throws IOException {
final FileSystem fs = FileSystem.get(conf);
final BufferedReader reader = new BufferedReader(new InputStreamReader(fs.open(path)));
try {
final ArrayList<String> addrs = new ArrayList<String>();
String line;
while ((line = reader.readLine()) != null) {
addrs.add(line);
}
return addrs;
} finally {
reader.close();
}
}
/* called once at the beginning of the task */
public void setup(Context context) throws IOException, InterruptedException {
BufferedReader br = null;
Path[] files = DistributedCache.getLocalCacheFiles(context.getConfiguration());
if (files != null && files.length > 0)
br = new BufferedReader(new FileReader(files[0].toString()));
String line = null;
/* reads the cached file into a hashmap */
try {
while ((line = br.readLine()) != null) {
String input[] = line.split(",");
vector.put(Long.valueOf(input[0]), Double.valueOf(input[1]));
}
} finally {
br.close();
}
}
public static void main(String[] args) throws Exception {
Configuration conf = new Configuration();
FileSystem fs = FileSystem.get(conf);
Path matFile = new Path(args[0]);
FSDataInputStream matData = fs.open(matFile);
BufferedReader br = new BufferedReader(new InputStreamReader(matData));
int i = 0;
String line;
while ((line = br.readLine()) != null) {
StringTokenizer tokenizer = new StringTokenizer(line);
String iRow = tokenizer.nextToken();
String iCol = tokenizer.nextToken();
if (Integer.parseInt(iRow) == Integer.parseInt(iCol)) {
i++;
}
}
br.close();
int dimention = i;
conf.setInt("DIMENTION", dimention);
Path xFile = new Path("preX/Result");
FSDataOutputStream xData = fs.create(xFile);
BufferedWriter iniX = new BufferedWriter(new OutputStreamWriter(xData));
for (int j = 0; j < dimention; j++) {
iniX.write(String.valueOf(j) + " 0");
iniX.newLine();
}
iniX.close();
URI matVec = new URI(args[0]);
DistributedCache.addCacheFile(matVec, conf);
int iteration = 0;
do {
ToolRunner.run(conf, new Jacobi(), args);
} while (iteration++ < max_iter && (!stopIteration(conf)));
}
public synchronized void close() throws IOException {
if (m_Input != null) {
m_Input.close();
}
}