代码线程安全设计、REST控制器访问ConcurrentMap及方法线程安全问询

一、如何在代码中实现线程安全设计？ #

常见的实现方式包括：

• 使用线程安全的集合类：比如ConcurrentHashMap、CopyOnWriteArrayList，这类集合内部通过分段锁、CAS等机制保证多线程下的操作安全。
• 使用原子类：如AtomicLong、AtomicInteger，基于CAS（比较并交换）操作实现无锁的原子更新，避免锁竞争。
• 同步机制：通过synchronized关键字（方法或代码块）或者Lock接口（如ReentrantLock）对共享资源的访问进行互斥控制。
• 避免共享可变状态：将可变状态封装在局部变量中，或者使用线程局部变量ThreadLocal，让每个线程持有独立的变量副本。
• 使用不可变对象：不可变对象的状态一旦创建就无法修改，天然线程安全，比如String、Integer等。

二、针对代码的疑问解答 #

1. 如何从REST控制器端访问类级别的ConcurrentMap对象？ #

当前ConcurrentMap是私有成员，不能直接访问，建议通过以下方式实现：

• 给该类添加一个公共的封装方法，用于获取指定文本的计数，避免直接暴露map引用：

  public Long getTextCount(String text) {
      AtomicLong count = map.get(text);
      return count != null ? count.get() : 0L;
  }
  

• 如果该类是Spring管理的Bean（比如标注了@Component），在REST控制器中通过依赖注入获取该类的实例，然后调用上述方法即可：

  @RestController
  public class TextCountController {
      private final TextCounter textCounter;
  
      // 构造注入
      public TextCountController(TextCounter textCounter) {
          this.textCounter = textCounter;
      }
  
      @GetMapping("/count/{text}")
      public Long getCount(@PathVariable String text) {
          return textCounter.getTextCount(text);
      }
  }
  

  注：TextCounter是包含ConcurrentMap和addText方法的类。

2. 在多线程场景下，addText方法是否仍需使用Lock或CountDownLatch来保障线程安全？ #

不需要额外使用Lock或CountDownLatch，当前代码的逻辑已经是线程安全的：

• ConcurrentHashMap的get和putIfAbsent方法都是原子操作，保证多线程下不会出现数据不一致。
• AtomicLong的incrementAndGet是基于CAS的原子操作，能安全地完成计数递增。

不过可以简化当前的addText逻辑，使用ConcurrentHashMap的computeIfAbsent方法，代码更简洁且同样线程安全：

public void addText(String text){
    System.out.println("current thread name "+Thread.currentThread().getName());
    map.computeIfAbsent(text, k -> new AtomicLong(0)).incrementAndGet();
}

内容的提问来源于stack exchange，提问作者chaitu