每个线程使用唯一ID并释放它以供重用

时间:2012-08-18 17:49:24

标签: java multithreading synchronization

下面是run method中的代码,我总是试图通过制作unique id from the availableExistingIds同时获取releasinglinked list order,但是在某些情况下我发现,我得到的是NoSuchElementException而id是zero few times我认为不应该是这种情况。

class IdPool {
    private final LinkedList<Integer> availableExistingIds = new LinkedList<Integer>();

    public IdPool() {
        for (int i = 1; i <= 1000; i++) {
            availableExistingIds.add(i);
        }
    }

    public synchronized Integer getExistingId() {
        return availableExistingIds.removeFirst();
    }

    public synchronized void releaseExistingId(Integer id) {
        availableExistingIds.add(id);
    }
}


class ThreadNewTask implements Runnable {
    private IdPool idPool;
    private int id;

    public ThreadNewTask(IdPool idPool) {
        this.idPool = idPool;
    }

    public void run() {
        try {
            id = idPool.getExistingId();
            //Anything wrong here?  
                    if(id==0) {
                        System.out.println("Found Zero");
                    }
            someMethod(id);
        } catch (Exception e) {
            System.out.println(e);
        } finally {
            idPool.releaseExistingId(id);
        }
    }

    // This method needs to be synchronized or not?
            private synchronized void someMethod(Integer id) {
                System.out.println("Task: " +id);
                // and do other calcuations whatever you need to do in your program
            }
}

问题陈述: -

如何在代码中避免使用此zero id case?我可以获得id = 0的一种情况是id池耗尽(空)。当发生这种情况时,行:

id = idPool.getExistingId();

将失败并显示NoSuchElementException。在这种情况下,finally块将运行:

idPool.releaseExistingId(id);

但是,由于第一行失败,id仍然会有default value of 0。所以我最终“释放”0并将其添加回id池,即使它从未在池中开始。然后,后来的任务可以合法地取0。这就是我不需要的东西。任何人都可以建议我如何在我的代码中克服这种情况?我一直希望id应该在1 to 1000范围内。

1 个答案:

答案 0 :(得分:5)

为什么不修改你的代码,以便在没有可用的id时不会崩溃,而是等待一个可用的?

否则,每次有太多线程同时工作时,池将会耗尽,并且您将不得不处理许多失败的线程。同步工作也会自动为您完成。

编辑:这是修改后的代码

class ThreadNewTask implements Runnable {
  private BlockingQueue<Integer> pool;
  private int id;

  public ThreadNewTask(BlockingQueue<Integer> pool) {
    this.pool = pool;
  }

  public void run() {
    try {
        id = pool.take();
        someMethod(id);
    } catch (Exception e) {
        System.out.println(e);
    } finally {
        pool.offer(id);
    }
  }

  private void someMethod(Integer id) {
    System.out.println("Task: " +id);
            // and do other calcuations whatever you need to do in your program
  }
}  

然后用这样的东西初始化池:

LinkedList<Integer> availableExistingIds = new LinkedList<Integer>();
for (int i = 1; i <= 1000; i++) {
  availableExistingIds.add(i);
}
BlockingQueue<Integer> pool = new ArrayBlockingQueue<Integer>(1000, false, availableExistingIds);