lib/cpp/src/thrift/windows/TWinsockSingleton.cpp

/*
 * 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 <thrift/windows/TWinsockSingleton.h>

// boost
#include <boost/assert.hpp>
#include <stdexcept>

namespace apache { namespace thrift { namespace transport {

TWinsockSingleton::instance_ptr TWinsockSingleton::instance_ptr_(NULL);
#if USE_BOOST_THREAD
boost::once_flag                TWinsockSingleton::flags_ = BOOST_ONCE_INIT;
#elif USE_STD_THREAD
std::once_flag                  TWinsockSingleton::flags_;
#else
#error For windows you must choose USE_BOOST_THREAD or USE_STD_THREAD
#endif

//------------------------------------------------------------------------------
TWinsockSingleton::TWinsockSingleton(void)
{
    WORD    version(MAKEWORD(2, 2));
    WSAData data = {0};

    int error(WSAStartup(version, &data));
    if (error != 0)
    {
        BOOST_ASSERT(false);
        throw std::runtime_error("Failed to initialise Winsock.");
    }
}

//------------------------------------------------------------------------------
TWinsockSingleton::~TWinsockSingleton(void)
{
    WSACleanup();
}

//------------------------------------------------------------------------------
void TWinsockSingleton::create(void)
{
#if USE_BOOST_THREAD
    boost::call_once(init, flags_);
#elif USE_STD_THREAD
    std::call_once(flags_, init);
#endif
}

//------------------------------------------------------------------------------
void TWinsockSingleton::init(void)
{
    instance_ptr_.reset(new TWinsockSingleton);
}

}}} // apache::thrift::transport