对可变对象的 AtomicReference 和可见性

Question

假设我有一个对象列表的 AtomicReference：

AtomicReference<List<?>> batch = new AtomicReference<List<Object>>(new ArrayList<Object>());

线程 A 将元素添加到此列表：batch.get().add(o);

随后，线程 B 获取该列表，例如，将其存储在数据库中： insertBatch(batch.get());

当写入（线程 A）和读取（线程 B）确保线程 B 看到的列表与 A 离开时的方式相同，还是由 AtomicReference 处理？

换句话说：如果我有一个可变对象的 AtomicReference，并且一个线程更改了该对象，其他线程是否会立即看到此更改？

编辑：

也许一些示例代码是有序的：

public void process(Reader in) throws IOException {
    List<Future<AtomicReference<List<Object>>>> tasks = new ArrayList<Future<AtomicReference<List<Object>>>>();
    ExecutorService exec = Executors.newFixedThreadPool(4);

    for (int i = 0; i < 4; ++i) {
        tasks.add(exec.submit(new Callable<AtomicReference<List<Object>>>() {
            @Override public AtomicReference<List<Object>> call() throws IOException {

                final AtomicReference<List<Object>> batch = new AtomicReference<List<Object>>(new ArrayList<Object>(batchSize));

                Processor.this.parser.parse(in, new Parser.Handler() {
                    @Override public void onNewObject(Object event) {
                            batch.get().add(event);

                            if (batch.get().size() >= batchSize) {
                                dao.insertBatch(batch.getAndSet(new ArrayList<Object>(batchSize)));
                            }
                    }
                });

                return batch;
            }
        }));
    }

    List<Object> remainingBatches = new ArrayList<Object>();

    for (Future<AtomicReference<List<Object>>> task : tasks) {
        try {
            AtomicReference<List<Object>> remainingBatch = task.get();
            remainingBatches.addAll(remainingBatch.get());
        } catch (ExecutionException e) {
            Throwable cause = e.getCause();

            if (cause instanceof IOException) {
                throw (IOException)cause;
            }

            throw (RuntimeException)cause;
        }
    }

    // these haven't been flushed yet by the worker threads
    if (!remainingBatches.isEmpty()) {
        dao.insertBatch(remainingBatches);
    }
}

这里发生的情况是，我创建了四个工作线程来解析一些文本（这是 process() 的 Reader in 参数方法）。每个工作进程都会将已解析的行保存在一个批次中，并在批次满时刷新该批次 (dao.insertBatch(batch.getAndSet(new ArrayList

由于文本中的行数不是批处理大小的倍数，因此最后一个对象最终会出现在未刷新的批处理中，因为它未满。因此，这些剩余批次由主线程插入。

我使用 AtomicReference.getAndSet() 将完整批次替换为空批次。该程序在线程方面正确吗？

Answer 1

嗯...实际上不是这样的。 AtomicReference 保证引用本身在线程中可见，即如果您为其分配与原始引用不同的引用，则更新将可见。它不保证引用所指向的对象的实际内容。

因此，对列表内容的读/写操作需要单独同步。

编辑：因此，从您更新的代码和您发布的评论来看，将本地引用设置为易失性足以确保可见性。

Answer 2

我认为，忘记这里的所有代码，你确切的问题是这样的：

在写入（线程 A）和
读取（线程 B）以确保线程 B 按 A 留下的方式查看列表，
或者这是由 AtomicReference 处理的？

因此，对此的确切响应是：是，原子处理可见性。这不是我的观点，而是 JDK文档一：

原子的访问和更新的内存效应通常遵循易失性的规则，如《Java语言规范，第三版（17.4内存模型）》中所述。

我希望这有帮助。

Answer 3

添加到Tudor的答案：你将必须制作ArrayList 本身是线程安全的，或者 - 根据您的要求 - 甚至更大的代码块。

如果您可以使用线程安全的ArrayList，您可以像这样“装饰”它：

batch = java.util.Collections.synchronizedList(new ArrayList<Object>());

但请记住：即使像这样的“简单”构造也不是线程安全的：

Object o = batch.get(batch.size()-1);

Answer 4

AtomicReference 只会帮助您引用列表，它不会对列表本身执行任何操作。更具体地说，在您的场景中，当系统处于负载状态时，您几乎肯定会遇到问题，其中消费者已获取列表，而生产者正在向列表添加项目。

在我看来，您应该使用 BlockingQueue。如果生产者比消费者更快，那么您可以限制内存占用，并让队列处理所有争用。

类似于：

ArrayBlockingQueue<Object> queue = new ArrayBlockingQueue<Object> (50);

// ... Producer
queue.put(o);

// ... Consumer
List<Object> queueContents = new ArrayList<Object> ();
// Grab everything waiting in the queue in one chunk. Should never be more than 50 items.
queue.drainTo(queueContents);

添加

感谢@Tudor指出您正在使用的架构。 ……我不得不承认这很奇怪。据我所知，你根本不需要 AtomicReference 。每个线程都拥有自己的 ArrayList，直到它被传递到 dao，此时它被替换，因此任何地方都不会发生争用。

我有点担心您在一个 Reader 上创建四个解析器。我希望您有某种方法可以确保每个解析器不会影响其他解析器。

我个人会使用某种形式的生产者-消费者模式，正如我在上面的代码中所描述的那样。也许是这样的。

static final int PROCESSES = 4;
static final int batchSize = 10;

public void process(Reader in) throws IOException, InterruptedException {

  final List<Future<Void>> tasks = new ArrayList<Future<Void>>();
  ExecutorService exec = Executors.newFixedThreadPool(PROCESSES);
  // Queue of objects.
  final ArrayBlockingQueue<Object> queue = new ArrayBlockingQueue<Object> (batchSize * 2);
  // The final object to post.
  final Object FINISHED = new Object();

  // Start the producers.
  for (int i = 0; i < PROCESSES; i++) {
    tasks.add(exec.submit(new Callable<Void>() {
      @Override
      public Void call() throws IOException {

        Processor.this.parser.parse(in, new Parser.Handler() {
          @Override
          public void onNewObject(Object event) {
            queue.add(event);
          }
        });
        // Post a finished down the queue.
        queue.add(FINISHED);
        return null;
      }
    }));
  }

  // Start the consumer.
  tasks.add(exec.submit(new Callable<Void>() {
    @Override
    public Void call() throws IOException {
      List<Object> batch = new ArrayList<Object>(batchSize);
      int finishedCount = 0;
      // Until all threads finished.
      while ( finishedCount < PROCESSES ) {
        Object o = queue.take();
        if ( o != FINISHED ) {
          // Batch them up.
          batch.add(o);
          if ( batch.size() >= batchSize ) {
            dao.insertBatch(batch);
            // If insertBatch takes a copy we could merely clear it.
            batch = new ArrayList<Object>(batchSize);
          }
        } else {
          // Count the finishes.
          finishedCount += 1;
        }
      }
      // Finished! Post any incopmplete batch.
      if ( batch.size() > 0 ) {
        dao.insertBatch(batch);
      }
      return null;
    }
  }));

  // Wait for everything to finish.
  exec.shutdown();
  // Wait until all is done.
  boolean finished = false;
  do {
    try {
      // Wait up to 1 second for termination.
      finished = exec.awaitTermination(1, TimeUnit.SECONDS);
    } catch (InterruptedException ex) {
    }
  } while (!finished);
}