60 lines
1.2 KiB
C++
60 lines
1.2 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2017
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
#pragma once
|
|
|
|
#include "td/utils/port/config.h"
|
|
|
|
#ifdef TD_POLL_SELECT
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/port/Fd.h"
|
|
#include "td/utils/port/PollBase.h"
|
|
|
|
#include <sys/select.h>
|
|
|
|
namespace td {
|
|
namespace detail {
|
|
|
|
class Select final : public PollBase {
|
|
public:
|
|
Select() = default;
|
|
Select(const Select &) = delete;
|
|
Select &operator=(const Select &) = delete;
|
|
Select(Select &&) = delete;
|
|
Select &operator=(Select &&) = delete;
|
|
~Select() override = default;
|
|
|
|
void init() override;
|
|
|
|
void clear() override;
|
|
|
|
void subscribe(const Fd &fd, Fd::Flags flags) override;
|
|
|
|
void unsubscribe(const Fd &fd) override;
|
|
|
|
void unsubscribe_before_close(const Fd &fd) override;
|
|
|
|
void run(int timeout_ms) override;
|
|
|
|
private:
|
|
struct FdInfo {
|
|
Fd fd_ref;
|
|
Fd::Flags flags;
|
|
};
|
|
vector<FdInfo> fds_;
|
|
fd_set all_fd_;
|
|
fd_set read_fd_;
|
|
fd_set write_fd_;
|
|
fd_set except_fd_;
|
|
int max_fd_;
|
|
};
|
|
|
|
} // namespace detail
|
|
} // namespace td
|
|
|
|
#endif
|