如何在ThreadFactory中初始化ThreadLocal变量？

Question

I have n resources and n threads, each thread will use a resource for its tasks.
When thread is created I want it to receive a resources and store it local thread variable.
When thread is done, I need to release the resource

ExecutorService pool = Executors.newFixedThreadPool(10, new ThreadFactory() {
        public Thread newThread(final Runnable r) {
            Thread thread = new Thread(new Runnable() {
                public void run() {
                         //set ThreadLocal with a resource
                         //run   
                         //close the resource 
                }
            });
                return thread;
        }
});

Answer 1

Rewrite getResource() and releaseResource() according to your needs.

class MyThreadFactory implements ThreadFactory {
      // String is the type of resource; change it if nesessary
  static final ThreadLocal<String> currentResourceKey = new ThreadLocal<String>();

  int n = 0;

  String getResource() {
    n++;
    System.out.println("aquired:" + n);
    return Integer.valueOf(n).toString();
  }

  void releaseResource(String res) {
    System.out.println("released:" + res);
  }

  @Override
  public Thread newThread(Runnable r) {
    return new Thread(r) {
        public void run() {
            currentResourceKey.set(getResource());
            try {
                super.run();
            } finally {
                releaseResource(currentResourceKey.get());
            }
        }
    };
  }
}

Test code:

    ExecutorService pool = Executors.newFixedThreadPool(2,new MyThreadFactory());
    for (int k = 0; k < 5; k++) {
        pool.submit(new Runnable() {
            public void run() {
                String res = currentResourceKey.get();
                try {
                    Thread.sleep(50);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("  executed:"+res+" on "+Thread.currentThread().getName());
            }
        });
    }
    pool.shutdown();