]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | /* | |
4 | * Ceph - scalable distributed file system | |
5 | * | |
6 | * Copyright (C) 2014 CohortFS, LLC | |
7 | * | |
8 | * This is free software; you can redistribute it and/or | |
9 | * modify it under the terms of the GNU Lesser General Public | |
10 | * License version 2.1, as published by the Free Software | |
11 | * Foundation. See file COPYING. | |
12 | * | |
13 | */ | |
14 | ||
15 | ||
16 | #ifndef QUEUE_STRATEGY_H | |
17 | #define QUEUE_STRATEGY_H | |
18 | ||
224ce89b WB |
19 | #include <vector> |
20 | #include <memory> | |
7c673cae FG |
21 | #include <boost/intrusive/list.hpp> |
22 | #include "DispatchStrategy.h" | |
23 | #include "msg/Messenger.h" | |
24 | ||
25 | namespace bi = boost::intrusive; | |
26 | ||
27 | class QueueStrategy : public DispatchStrategy { | |
28 | Mutex lock; | |
224ce89b | 29 | const int n_threads; |
7c673cae FG |
30 | bool stop; |
31 | ||
32 | Message::Queue mqueue; | |
33 | ||
34 | class QSThread : public Thread { | |
35 | public: | |
36 | bi::list_member_hook<> thread_q; | |
37 | QueueStrategy *dq; | |
38 | Cond cond; | |
39 | explicit QSThread(QueueStrategy *dq) : thread_q(), dq(dq), cond() {} | |
40 | void* entry() { | |
41 | dq->entry(this); | |
7c673cae FG |
42 | return NULL; |
43 | } | |
44 | ||
45 | typedef bi::list< QSThread, | |
46 | bi::member_hook< QSThread, | |
47 | bi::list_member_hook<>, | |
48 | &QSThread::thread_q > > Queue; | |
49 | }; | |
50 | ||
224ce89b WB |
51 | std::vector<std::unique_ptr<QSThread>> threads; //< all threads |
52 | QSThread::Queue disp_threads; //< waiting threads | |
7c673cae FG |
53 | |
54 | public: | |
55 | explicit QueueStrategy(int n_threads); | |
56 | void ds_dispatch(Message *m) override; | |
57 | void shutdown() override; | |
58 | void start() override; | |
59 | void wait() override; | |
60 | void entry(QSThread *thrd); | |
61 | virtual ~QueueStrategy() {} | |
62 | }; | |
63 | #endif /* QUEUE_STRATEGY_H */ |