diff options
Diffstat (limited to '')
-rw-r--r-- | src/common/event_socket.h | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/src/common/event_socket.h b/src/common/event_socket.h new file mode 100644 index 000000000..9224f7683 --- /dev/null +++ b/src/common/event_socket.h @@ -0,0 +1,83 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2015 XSky <haomai@xsky.com> + * + * Author: Haomai Wang <haomaiwang@gmail.com> + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ + +#ifndef CEPH_COMMON_EVENT_SOCKET_H +#define CEPH_COMMON_EVENT_SOCKET_H + +#include <unistd.h> +#if defined(__FreeBSD__) || defined(__APPLE__) +#include <errno.h> +#endif +#include "include/event_type.h" + +class EventSocket { + int socket; + int type; + + public: + EventSocket(): socket(-1), type(EVENT_SOCKET_TYPE_NONE) {} + bool is_valid() const { return socket != -1; } + int init(int fd, int t) { + switch (t) { + case EVENT_SOCKET_TYPE_PIPE: +#ifdef HAVE_EVENTFD + case EVENT_SOCKET_TYPE_EVENTFD: +#endif + { + socket = fd; + type = t; + return 0; + } + } + return -EINVAL; + } + int notify() { + int ret; + switch (type) { + case EVENT_SOCKET_TYPE_PIPE: + { + char buf[1]; + buf[0] = 'i'; + ret = write(socket, buf, 1); + if (ret < 0) + ret = -errno; + else + ret = 0; + break; + } +#ifdef HAVE_EVENTFD + case EVENT_SOCKET_TYPE_EVENTFD: + { + uint64_t value = 1; + ret = write(socket, &value, sizeof (value)); + if (ret < 0) + ret = -errno; + else + ret = 0; + break; + } +#endif + default: + { + ret = -1; + break; + } + } + return ret; + } +}; + +#endif |