ThreadLocal.h

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef _DECAF_LANG_THREADLOCAL_H_
#define _DECAF_LANG_THREADLOCAL_H_

#include <decaf/util/Config.h>

#include <decaf/internal/util/concurrent/ThreadLocalImpl.h>

namespace decaf {
namespace lang {

    template <typename E>
    class ThreadLocal : protected decaf::internal::util::concurrent::ThreadLocalImpl {
    private:

        ThreadLocal(const ThreadLocal&);
        ThreadLocal& operator= (const ThreadLocal&);

    public:

        ThreadLocal() : ThreadLocalImpl() {}

        virtual ~ThreadLocal() {
            try {
                removeAll();
            } catch(...) {}
        }

        E& get() {
            void* bytes = getRawValue();
            if (bytes == NULL) {
                E* value = new E();
                *value = initialValue();
                setRawValue((void*)value);
                bytes = value;
            }
            return *((E*)bytes);
        }

        void set(const E& value) {
            setRawValue((void*)(new E(value)));
        }

        void remove() {
            this->setRawValue(NULL);
        }

    protected:

        virtual E initialValue() const {
            return E();
        }

    protected:

        virtual void doDelete(void* value) {
            if (value != 0) {
                delete static_cast<E*>(value);
            }
        }

    };

}}

#endif /* _DECAF_LANG_THREADLOCAL_H_ */