java怎么實(shí)現(xiàn)生產(chǎn)者消費(fèi)者模型

小億
82
2024-09-26 01:32:01
欄目: 編程語言

在Java中,生產(chǎn)者消費(fèi)者模型可以通過使用線程同步機(jī)制(如synchronized關(guān)鍵字、Lock接口、Semaphore信號(hào)量等)和wait()、notifyAll()等待通知機(jī)制實(shí)現(xiàn)。這里給出一個(gè)簡單的示例,使用synchronized關(guān)鍵字和wait()、notifyAll()方法實(shí)現(xiàn)生產(chǎn)者消費(fèi)者模型。

首先,創(chuàng)建一個(gè)共享資源類(共享隊(duì)列):

import java.util.LinkedList;
import java.util.Queue;

public class SharedQueue {
    private Queue<Integer> queue = new LinkedList<>();

    public synchronized void add(int item) {
        while (queue.size() == 10) {
            try {
                wait(); // 當(dāng)前線程等待,釋放鎖
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        queue.add(item);
        notifyAll(); // 當(dāng)前線程通知其他線程
    }

    public synchronized int remove() {
        while (queue.isEmpty()) {
            try {
                wait(); // 當(dāng)前線程等待,釋放鎖
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        int item = queue.poll();
        notifyAll(); // 當(dāng)前線程通知其他線程
        return item;
    }
}

接下來,創(chuàng)建生產(chǎn)者和消費(fèi)者線程類:

public class Producer implements Runnable {
    private SharedQueue sharedQueue;

    public Producer(SharedQueue sharedQueue) {
        this.sharedQueue = sharedQueue;
    }

    @Override
    public void run() {
        for (int i = 0; i < 20; i++) {
            sharedQueue.add(i);
            System.out.println("生產(chǎn)者生產(chǎn)了: " + i);
            try {
                Thread.sleep(100); // 模擬生產(chǎn)耗時(shí)
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

public class Consumer implements Runnable {
    private SharedQueue sharedQueue;

    public Consumer(SharedQueue sharedQueue) {
        this.sharedQueue = sharedQueue;
    }

    @Override
    public void run() {
        for (int i = 0; i < 20; i++) {
            int item = sharedQueue.remove();
            System.out.println("消費(fèi)者消費(fèi)了: " + item);
            try {
                Thread.sleep(200); // 模擬消費(fèi)耗時(shí)
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

最后,在主類中創(chuàng)建生產(chǎn)者和消費(fèi)者線程并啟動(dòng):

public class Main {
    public static void main(String[] args) {
        SharedQueue sharedQueue = new SharedQueue();
        Producer producer = new Producer(sharedQueue);
        Consumer consumer = new Consumer(sharedQueue);

        Thread producerThread = new Thread(producer);
        Thread consumerThread = new Thread(consumer);

        producerThread.start();
        consumerThread.start();
    }
}

運(yùn)行上述代碼,可以看到生產(chǎn)者和消費(fèi)者線程交替執(zhí)行,生產(chǎn)者生產(chǎn)商品放入共享隊(duì)列,消費(fèi)者從共享隊(duì)列取出商品。這樣就實(shí)現(xiàn)了一個(gè)簡單的生產(chǎn)者消費(fèi)者模型。

0