队列缓存区-db写入

近期需要写入大量操作记录到db中，在不考虑高并发的情况下做一个简单缓存区，使用阻塞队列，定时将队列数据刷进db中。

代码如下：

1.建立监听器

/**
 * 监听器
 *
 * @author johnson
 * @date 2020/12/31 11:03
 */
@Component
public class MyListener implements ApplicationRunner {

    @Autowired
    private AutoThreadConfig autoThreadConfig;

    ScheduledExecutorService schedule = Executors.newScheduledThreadPool(1);

    @Override
    public void run(ApplicationArguments args) throws Exception {
        System.out.println("启动监听器....");
        //设置延迟1s 每个10s执行依次数据写入
        schedule.scheduleAtFixedRate(new MyThread(),1L, 10L,TimeUnit.SECONDS);
    }

   class MyThread implements Runnable{
       @Override
       public void run() {
           SimpleDateFormat sdf = new SimpleDateFormat("HH:mm:ss:sss");
            String time = sdf.format(new Date());
           System.out.println("执行写入，时间："+time);
           autoThreadConfig.push();
       }
   }
}

2.建立单例队列

public class LogQueue {

    private static volatile BlockingQueue<String> LogDBEntityQueue;

    public LogQueue(){
    }

    public static BlockingQueue<String> getOrderLogQueue(){
        if(LogDBEntityQueue == null){
            synchronized (LogQueue.class){
                if(LogDBEntityQueue == null){
                    LogDBEntityQueue = new LinkedBlockingDeque<>();
                }
            }
        }
        return LogDBEntityQueue;
    }
}

3.完成数据写入方法

/**
 * TODO
 *
 * @author johnson
 * @date 2020/12/31 11:01
 */
@Component
public class AutoThreadConfig {


    private BlockingQueue<String> blockingQueue = LogQueue.getOrderLogQueue();


    public void push(){
        if(CollectionUtils.isEmpty(blockingQueue)){
            return;
        }
        List<String> list = new ArrayList<>();
        try {
            Queues.drain(blockingQueue,list,blockingQueue.size(),60, TimeUnit.SECONDS);
            System.out.println("本次写入数据量："+ list.size());
            list.stream().forEach(str ->{
                System.out.println(str);
            });

        }catch (Exception e){
            e.printStackTrace();
        }

        return;
    }


    public void save(String string){
        blockingQueue.offer(string);
    }

}

4.测试类

/**
 * 传入数据
 *
 * @author johnson
 * @date 2020/12/31 13:38
 */
@RestController
public class SaveDataController {



    @Autowired
    private AutoThreadConfig autoThreadConfig;


    @GetMapping("/save")
    public Boolean save(@RequestParam("str") String str){
        autoThreadConfig.save(str);
        return true;
    }
}

在调用save方法后，数据写入到队列中，队列在服务启动后，每隔10s自动将队列数据打印出来