| /* |
| * 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. |
| */ |
| |
| #include <log4cxx/logstring.h> |
| #include <log4cxx/helpers/threadspecificdata.h> |
| #include <log4cxx/helpers/exception.h> |
| #include <apr_thread_proc.h> |
| #if !defined(LOG4CXX) |
| #define LOG4CXX 1 |
| #endif |
| #include <log4cxx/helpers/aprinitializer.h> |
| |
| using namespace log4cxx; |
| using namespace log4cxx::helpers; |
| |
| |
| ThreadSpecificData::ThreadSpecificData() |
| : ndcStack(), mdcMap() |
| { |
| } |
| |
| ThreadSpecificData::~ThreadSpecificData() |
| { |
| } |
| |
| |
| log4cxx::NDC::Stack& ThreadSpecificData::getStack() |
| { |
| return ndcStack; |
| } |
| |
| log4cxx::MDC::Map& ThreadSpecificData::getMap() |
| { |
| return mdcMap; |
| } |
| |
| ThreadSpecificData& ThreadSpecificData::getDataNoThreads() |
| { |
| static ThreadSpecificData noThreadData; |
| return noThreadData; |
| } |
| |
| ThreadSpecificData* ThreadSpecificData::getCurrentData() |
| { |
| #if APR_HAS_THREADS |
| void* pData = NULL; |
| apr_threadkey_private_get(&pData, APRInitializer::getTlsKey()); |
| return (ThreadSpecificData*) pData; |
| #else |
| return &getDataNoThreads(); |
| #endif |
| } |
| |
| void ThreadSpecificData::recycle() |
| { |
| #if APR_HAS_THREADS |
| |
| if (ndcStack.empty() && mdcMap.empty()) |
| { |
| void* pData = NULL; |
| apr_status_t stat = apr_threadkey_private_get(&pData, APRInitializer::getTlsKey()); |
| |
| if (stat == APR_SUCCESS && pData == this) |
| { |
| stat = apr_threadkey_private_set(0, APRInitializer::getTlsKey()); |
| |
| if (stat == APR_SUCCESS) |
| { |
| delete this; |
| } |
| } |
| } |
| |
| #endif |
| } |
| |
| void ThreadSpecificData::put(const LogString& key, const LogString& val) |
| { |
| ThreadSpecificData* data = getCurrentData(); |
| |
| if (data == 0) |
| { |
| data = createCurrentData(); |
| } |
| |
| if (data != 0) |
| { |
| data->getMap()[key] = val; |
| } |
| } |
| |
| |
| |
| |
| void ThreadSpecificData::push(const LogString& val) |
| { |
| ThreadSpecificData* data = getCurrentData(); |
| |
| if (data == 0) |
| { |
| data = createCurrentData(); |
| } |
| |
| if (data != 0) |
| { |
| NDC::Stack& stack = data->getStack(); |
| |
| if (stack.empty()) |
| { |
| stack.push(NDC::DiagnosticContext(val, val)); |
| } |
| else |
| { |
| LogString fullMessage(stack.top().second); |
| fullMessage.append(1, (logchar) 0x20); |
| fullMessage.append(val); |
| stack.push(NDC::DiagnosticContext(val, fullMessage)); |
| } |
| } |
| } |
| |
| void ThreadSpecificData::inherit(const NDC::Stack& src) |
| { |
| ThreadSpecificData* data = getCurrentData(); |
| |
| if (data == 0) |
| { |
| data = createCurrentData(); |
| } |
| |
| if (data != 0) |
| { |
| data->getStack() = src; |
| } |
| } |
| |
| |
| |
| ThreadSpecificData* ThreadSpecificData::createCurrentData() |
| { |
| #if APR_HAS_THREADS |
| ThreadSpecificData* newData = new ThreadSpecificData(); |
| apr_status_t stat = apr_threadkey_private_set(newData, APRInitializer::getTlsKey()); |
| |
| if (stat != APR_SUCCESS) |
| { |
| delete newData; |
| newData = NULL; |
| } |
| |
| return newData; |
| #else |
| return 0; |
| #endif |
| } |