CloseableThreadLocal.h

Go to the documentation of this file.

// Copyright (c) 2009-2014 Alan Wright. All rights reserved.
// Distributable under the terms of either the Apache License (Version 2.0)
// or the GNU Lesser General Public License.
 
#ifndef CLOSEABLETHREADLOCAL_H
#define CLOSEABLETHREADLOCAL_H
 
#include "LuceneThread.h"
 
namespace Lucene {

template <typename TYPE>
class CloseableThreadLocal : public LuceneObject {
public:
    typedef boost::shared_ptr<TYPE> localDataPtr;
    typedef Map<int64_t, localDataPtr> MapLocalData;
 
    CloseableThreadLocal() {
        localData = MapLocalData::newInstance();
    }
 
public:
    localDataPtr get() {
        SyncLock syncLock(this);
        typename MapLocalData::iterator local = localData.find(LuceneThread::currentId());
        if (local != localData.end()) {
            return local->second;
        }
        localDataPtr initial(initialValue());
        if (initial) {
            localData.put(LuceneThread::currentId(), initial);
        }
        return initial;
    }
 
    void set(const localDataPtr& data) {
        SyncLock syncLock(this);
        localData.put(LuceneThread::currentId(), data);
    }
 
    void close() {
        SyncLock syncLock(this);
        localData.remove(LuceneThread::currentId());
    }
 
protected:
    MapLocalData localData;
 
    virtual localDataPtr initialValue() {
        return localDataPtr(); // override
    }
};
 
}
 
#endif