mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-10-31 08:39:34 +08:00
60 lines
1.6 KiB
C++
60 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) 2016 The ZLToolKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLToolKit(https://github.com/ZLMediaKit/ZLToolKit).
|
|
*
|
|
* Use of this source code is governed by MIT license that can be found in the
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef ZLMEDIAKIT_EPOLL_H
|
|
#define ZLMEDIAKIT_EPOLL_H
|
|
#include "wepoll.h"
|
|
#include <map>
|
|
#include <mutex>
|
|
|
|
// 屏蔽 EPOLLET
|
|
#define EPOLLET 0
|
|
|
|
namespace toolkit {
|
|
// 索引handle
|
|
extern std::map<int, HANDLE> s_wepollHandleMap;
|
|
extern int s_handleIndex;
|
|
extern std::mutex s_handleMtx;
|
|
// 屏蔽epoll_create epoll_ctl epoll_wait参数差异
|
|
inline int epoll_create(int size) {
|
|
HANDLE handle = ::epoll_create(size);
|
|
if (!handle) {
|
|
return -1;
|
|
}
|
|
{
|
|
std::lock_guard<std::mutex> lck(s_handleMtx);
|
|
int idx = ++s_handleIndex;
|
|
s_wepollHandleMap[idx] = handle;
|
|
return idx;
|
|
}
|
|
}
|
|
|
|
inline int epoll_ctl(int ephnd, int op, SOCKET sock, struct epoll_event *ev) {
|
|
HANDLE handle;
|
|
{
|
|
std::lock_guard<std::mutex> lck(s_handleMtx);
|
|
handle = s_wepollHandleMap[ephnd];
|
|
}
|
|
return ::epoll_ctl(handle, op, sock, ev);
|
|
}
|
|
|
|
inline int epoll_wait(int ephnd, struct epoll_event *events, int maxevents, int timeout) {
|
|
HANDLE handle;
|
|
{
|
|
std::lock_guard<std::mutex> lck(s_handleMtx);
|
|
handle = s_wepollHandleMap[ephnd];
|
|
}
|
|
return ::epoll_wait(handle, events, maxevents, timeout);
|
|
}
|
|
|
|
} // namespace toolkit
|
|
|
|
#endif // ZLMEDIAKIT_EPOLL_H
|