Files
scylladb/tests/exchanger.hh
Avi Kivity 7f8d88371a Add LICENSE, NOTICE, and copyright headers to all source files.
The two files imported from the OSv project retain their original licenses.
2015-02-19 16:52:34 +02:00

51 lines
1.5 KiB
C++

/*
* This file is open source software, licensed to you under the terms
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. 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.
*/
/*
* Copyright (C) 2015 Cloudius Systems, Ltd.
*/
#pragma once
#include <mutex>
#include <condition_variable>
#include <experimental/optional>
// Single-element blocking queue
template <typename T>
class exchanger {
private:
std::mutex _mutex;
std::condition_variable _cv;
std::experimental::optional<T> _element;
public:
void give(T value) {
std::unique_lock<std::mutex> lock(_mutex);
_cv.wait(lock, [this] { return !_element; });
_element = value;
_cv.notify_one();
}
T take() {
std::unique_lock<std::mutex> lock(_mutex);
_cv.wait(lock, [this] { return bool(_element); });
auto v = *_element;
_element = {};
_cv.notify_one();
return v;
}
};