java并发编程 日志类demo

public class LogWriter {

    private long count = 0;

    private volatile boolean isShutDown;

    private final LogThread logThread;

    private final BlockingQueue<String> queue;

    public LogWriter(PrintWriter writer) {
        this.logThread = new LogThread(writer);
        this.queue = new LinkedBlockingQueue<>();
    }

    public void start() {
        logThread.start();
    }

    public void end() {
        // 中断take阻塞
        logThread.interrupt();
        // 判断是否停止，和interrupt区别，需要把日志打印完，在停止线程
        isShutDown = true;
    }

    public void log(String message) {
        try {
            synchronized (this) {
                // 如果停止了，就不能在添加日志
                if (isShutDown) {
                    throw new IllegalStateException("已关闭");
                }
                // 记录日志数量，在关闭的时候，打印完成
                count++;
            }
            queue.put(message);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

    }

    private class LogThread extends Thread {

        private PrintWriter writer;

        public LogThread(PrintWriter writer) {
            this.writer = writer;
        }

        @Override
        public void run() {
            try {
                while (true) {
                    try {
                        // 限判断，在打印，判断需要加锁
                        synchronized (LogWriter.this) {
                            if (isShutDown && count == 0) {
                                break;
                            }
                        }
                        writer.println(queue.take());
                        // 64位机.long-- 操作并非原子，加锁保证原子性和可见性
                        synchronized (LogWriter.this) {
                            count--;
                        }
                    } catch (InterruptedException e) {
                        // 继续执行
                        e.printStackTrace();
                    }
                }
            } finally {
                writer.close();
            }
        }
    }

    public static void main(String[] args) {
        PrintWriter printWriter = new PrintWriter(System.out, true);
        LogWriter logWriter = new LogWriter(printWriter);
        logWriter.start();
        logWriter.log("test");
        logWriter.end();
        System.out.println("main end");
    }
}