diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-21 11:54:28 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-21 11:54:28 +0000 |
commit | e6918187568dbd01842d8d1d2c808ce16a894239 (patch) | |
tree | 64f88b554b444a49f656b6c656111a145cbbaa28 /src/test/direct_messenger/QueueStrategy.h | |
parent | Initial commit. (diff) | |
download | ceph-upstream/18.2.2.tar.xz ceph-upstream/18.2.2.zip |
Adding upstream version 18.2.2.upstream/18.2.2
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/test/direct_messenger/QueueStrategy.h')
-rw-r--r-- | src/test/direct_messenger/QueueStrategy.h | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/src/test/direct_messenger/QueueStrategy.h b/src/test/direct_messenger/QueueStrategy.h new file mode 100644 index 000000000..b7f6df85d --- /dev/null +++ b/src/test/direct_messenger/QueueStrategy.h @@ -0,0 +1,63 @@ +// -*- 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) 2014 CohortFS, LLC + * + * 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 QUEUE_STRATEGY_H +#define QUEUE_STRATEGY_H + +#include <vector> +#include <memory> +#include <boost/intrusive/list.hpp> +#include "DispatchStrategy.h" +#include "msg/Messenger.h" + +namespace bi = boost::intrusive; + +class QueueStrategy : public DispatchStrategy { + ceph::mutex lock = ceph::make_mutex("QueueStrategy::lock"); + const int n_threads; + bool stop; + + Message::Queue mqueue; + + class QSThread : public Thread { + public: + bi::list_member_hook<> thread_q; + QueueStrategy *dq; + ceph::condition_variable cond; + explicit QSThread(QueueStrategy *dq) : thread_q(), dq(dq) {} + void* entry() { + dq->entry(this); + return NULL; + } + + typedef bi::list< QSThread, + bi::member_hook< QSThread, + bi::list_member_hook<>, + &QSThread::thread_q > > Queue; + }; + + std::vector<std::unique_ptr<QSThread>> threads; //< all threads + QSThread::Queue disp_threads; //< waiting threads + +public: + explicit QueueStrategy(int n_threads); + void ds_dispatch(Message *m) override; + void shutdown() override; + void start() override; + void wait() override; + void entry(QSThread *thrd); + virtual ~QueueStrategy() {} +}; +#endif /* QUEUE_STRATEGY_H */ |