Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Licensed to the Apache Software Foundation (ASF) under one |
| 3 | * or more contributor license agreements. See the NOTICE file |
| 4 | * distributed with this work for additional information |
| 5 | * regarding copyright ownership. The ASF licenses this file |
| 6 | * to you under the Apache License, Version 2.0 (the |
| 7 | * "License"); you may not use this file except in compliance |
| 8 | * with the License. You may obtain a copy of the License at |
| 9 | * |
| 10 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 11 | * |
| 12 | * Unless required by applicable law or agreed to in writing, |
| 13 | * software distributed under the License is distributed on an |
| 14 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| 15 | * KIND, either express or implied. See the License for the |
| 16 | * specific language governing permissions and limitations |
| 17 | * under the License. |
| 18 | */ |
| 19 | |
| 20 | #ifndef _THRIFT_TRANSPORT_WINDOWS_TWINSOCKSINGLETON_H_ |
| 21 | #define _THRIFT_TRANSPORT_WINDOWS_TWINSOCKSINGLETON_H_ 1 |
| 22 | |
| 23 | #if defined(_MSC_VER) && (_MSC_VER > 1200) |
| 24 | #pragma once |
| 25 | #endif // _MSC_VER |
| 26 | |
| 27 | #ifndef _WIN32 |
| 28 | #error This is a MSVC header only. |
| 29 | #endif |
| 30 | |
Roger Meier | ff77d07 | 2013-06-28 22:26:43 +0200 | [diff] [blame] | 31 | #include <thrift/thrift-config.h> |
Carl Yeksigian | 7cb7fc8 | 2013-06-07 07:33:01 -0400 | [diff] [blame] | 32 | |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 33 | // boost |
| 34 | #include <boost/noncopyable.hpp> |
| 35 | #include <boost/scoped_ptr.hpp> |
Carl Yeksigian | 7cb7fc8 | 2013-06-07 07:33:01 -0400 | [diff] [blame] | 36 | |
| 37 | #if USE_BOOST_THREAD |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 38 | #include <boost/thread/once.hpp> |
Carl Yeksigian | 7cb7fc8 | 2013-06-07 07:33:01 -0400 | [diff] [blame] | 39 | #elif USE_STD_THREAD |
| 40 | #include <mutex> |
| 41 | #else |
| 42 | #error For windows you must choose USE_BOOST_THREAD or USE_STD_THREAD |
| 43 | #endif |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 44 | |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 45 | namespace apache { |
| 46 | namespace thrift { |
| 47 | namespace transport { |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 48 | |
| 49 | /** |
| 50 | * Winsock2 must be intialised once only in order to create sockets. This class |
| 51 | * performs a one time initialisation when create is called. |
| 52 | */ |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 53 | class TWinsockSingleton : private boost::noncopyable { |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 54 | |
| 55 | public: |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 56 | typedef boost::scoped_ptr<TWinsockSingleton> instance_ptr; |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 57 | |
| 58 | private: |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 59 | TWinsockSingleton(void); |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 60 | |
| 61 | public: |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 62 | ~TWinsockSingleton(void); |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 63 | |
| 64 | public: |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 65 | static void create(void); |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 66 | |
| 67 | private: |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 68 | static void init(void); |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 69 | |
| 70 | private: |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 71 | static instance_ptr instance_ptr_; |
Carl Yeksigian | 7cb7fc8 | 2013-06-07 07:33:01 -0400 | [diff] [blame] | 72 | #if USE_BOOST_THREAD |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 73 | static boost::once_flag flags_; |
Carl Yeksigian | 7cb7fc8 | 2013-06-07 07:33:01 -0400 | [diff] [blame] | 74 | #elif USE_STD_THREAD |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 75 | static std::once_flag flags_; |
Carl Yeksigian | 7cb7fc8 | 2013-06-07 07:33:01 -0400 | [diff] [blame] | 76 | #else |
| 77 | #error Need a non-Boost non-C++11 way to track single initialization here. |
| 78 | #endif |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 79 | }; |
Konrad Grochowski | 16a23a6 | 2014-11-13 15:33:38 +0100 | [diff] [blame] | 80 | } |
| 81 | } |
| 82 | } // apache::thrift::transport |
Roger Meier | 84e4a3c | 2011-09-16 20:58:44 +0000 | [diff] [blame] | 83 | |
| 84 | #endif // _THRIFT_TRANSPORT_WINDOWS_TWINSOCKSINGLETON_H_ |