0

一旦生产者完成对 HDFS 中文件的写入，我们希望保证消费者进程读取生产者创建的数据。以下是我们正在努力改进的应用程序中使用的一种方法。

制片人：

private void produce(String file, int sleepSeconds) throws Exception {
        Configuration conf = new Configuration();
        conf.addResource(new Path(
                "C:\\dev\\software\\hadoop-0.22.0-src\\conf\\core-site.xml"));
        conf.set("fs.defaultFS", "hdfs://XXX:9000");
        FileSystem fileSystem = FileSystem.get(conf);

        Path path = new Path(file);
        if (fileSystem.exists(path)) {
            fileSystem.delete(path, false);
        }
        System.out.println("Creating file");
        FSDataOutputStream out = fileSystem.create(path);
        System.out.println("Writing data");
        out.writeUTF("--data--");
        System.out.println("Sleeping");
        Thread.sleep(sleepSeconds * 1000L);
        System.out.println("Writing data");
        out.writeUTF("--data--");
        System.out.println("Flushing");
        out.flush();
        out.close();
        fileSystem.close();
        System.out.println("Releasing lock on file");
    }

消费者：

private void consume(String file) throws Exception {
        Configuration conf = new Configuration();
        conf.addResource(new Path(
                "C:\\dev\\software\\hadoop-0.22.0-src\\conf\\core-site.xml"));
        conf.set("fs.defaultFS", "hdfs://XXX:9000");
        FileSystem fileSystem = FileSystem.get(conf);

        Path path = new Path(file);
        if (fileSystem.exists(path)) {
            System.out.println("File exists");
        } else {
            System.out.println("File doesn't exist");
            return;
        }
        FSDataOutputStream fsOut = null;
        while (fsOut == null) {
            try {
                fsOut = fileSystem.append(path);
            } catch (IOException e) {
                Thread.sleep(1000);
            }
        }
        FSDataInputStream in = fileSystem.open(path);
        OutputStream out = new BufferedOutputStream(System.out);
        byte[] b = new byte[1024];
        int numBytes = 0;
        while ((numBytes = in.read(b)) > 0) {
            out.write(b, 0, numBytes);
        }
        in.close();
        out.close();
        if (fsOut != null)
            fsOut.close();
        fileSystem.close();
        System.out.println("Releasing lock on file");
    }

流程应如何运行的要求如下：

  1. 生产者进程（不是线程）已启动。thread.sleep 模拟了一堆数据库调用和业务逻辑

  2. 消费者进程（不是线程）在另一台机器上启动，该机器会阻塞直到生产者释放它的锁。消费者读取时，其他进程不应修改数据文件

关于我们如何改进此代码/设计的任何建议，同时使用 HDFS java API 保证阅读器不会丢失数据？

4

1 回答 1

1

一种解决方案是使用临时后缀 / 前缀写入文件，并在写入完成后重命名文件：

例如输出到文件 file1.txt：

  • 写入名为.file1.txt或的文件file1.txt.tmp
  • 完成后关闭文件
  • file1.txt.tmp重命名 .file1.txt 或file1.txt
  • 与此同时，消费者正在等待file1.txt可用
于 2012-05-30T16:27:40.307 回答