summaryrefslogtreecommitdiffstats
path: root/src/common/event_socket.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/common/event_socket.h')
-rw-r--r--src/common/event_socket.h83
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 00000000..9224f768
--- /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