Blame view

trunk/src/app/srs_app_thread.hpp 7.7 KB
1 2 3
/*
The MIT License (MIT)
4
Copyright (c) 2013-2015 SRS(simple-rtmp-server)
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23

Permission is hereby granted, free of charge, to any person obtaining a copy of
this software and associated documentation files (the "Software"), to deal in
the Software without restriction, including without limitation the rights to
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
the Software, and to permit persons to whom the Software is furnished to do so,
subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
24 25
#ifndef SRS_APP_THREAD_HPP
#define SRS_APP_THREAD_HPP
26 27

/*
28
#include <srs_app_thread.hpp>
29 30 31
*/
#include <srs_core.hpp>
32
#include <srs_app_st.hpp>
winlin authored
33
34 35 36
/**
* the handler for the thread, callback interface.
* the thread model defines as:
37
*     handler->on_thread_start()
38 39 40 41 42 43
*     while loop:
*        handler->on_before_cycle()
*        handler->cycle()
*        handler->on_end_cycle()
*        if !loop then break for user stop thread.
*        sleep(CycleIntervalMilliseconds)
44
*     handler->on_thread_stop()
45 46 47
* when stop, the thread will interrupt the st_thread,
* which will cause the socket to return error and 
* terminate the cycle thread.
48
*
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
* Usage 1: stop by other thread.
*       user can create thread and stop then start again and again,
*       generally must provides a start and stop method, @see SrsIngester.
*       the step to create a thread stop by other thread:
*       1. create SrsThread field, with joinable true.
*       2. must use stop to stop and join the thread.
*       for example:
*           class SrsIngester : public ISrsThreadHandler {
*               public: SrsIngester() { pthread = new SrsThread("ingest", this, SRS_AUTO_INGESTER_SLEEP_US, true); }
*               public: virtual int start() { return pthread->start(); }
*               public: virtual void stop() { pthread->stop(); }
*               public: virtual int cycle() { 
*                   // check status, start ffmpeg when stopped. 
*               }
*           };
64
*
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120
* Usage 2: stop by thread itself.
*       user can create thread which stop itself,
*       generally only need to provides a start method, 
*       the object will destroy itself then terminate the thread, @see SrsConnection
*       1. create SrsThread field, with joinable false.
*       2. owner stop thread loop, destroy itself when thread stop.
*       for example:
*           class SrsConnection : public ISrsThreadHandler {
*               public: SrsConnection() { pthread = new SrsThread("conn", this, 0, false); }
*               public: virtual int start() { return pthread->start(); }
*               public: virtual int cycle() { 
*                   // serve client.
*                   // set loop to stop to quit, stop thread itself.
*                   pthread->stop_loop();
*               }
*               public: virtual int on_thread_stop() { 
*                   // remove the connection in thread itself.
*                   server->remove(this); 
*               }
*           };
* 
* Usage 3: loop in the cycle method.
*       user can use loop code in the cycle method, @see SrsForwarder
*       1. create SrsThread field, with or without joinable is ok.
*       2. loop code in cycle method, check the can_loop() for thread to quit.
*       for example:
*           class SrsForwarder : public ISrsThreadHandler {
*               public: virtual int cycle() {
*                   while (pthread->can_loop()) {
*                       // read msgs from queue and forward to server.
*                   }
*               }
*           };
*
* @remark why should check can_loop() in cycle method?
*       when thread interrupt, the socket maybe not got EINT,
*       espectially on st_usleep(), so the cycle must check the loop,
*       when handler->cycle() has loop itself, for example:
*               while (true):
*                   if (read_from_socket(skt) < 0) break;
*       if thread stop when read_from_socket, it's ok, the loop will break,
*       but when thread stop interrupt the s_usleep(0), then the loop is
*       death loop.
*       in a word, the handler->cycle() must:
*               while (pthread->can_loop()):
*                   if (read_from_socket(skt) < 0) break;
*       check the loop, then it works.
*
* @remark why should use stop_loop() to terminate thread in itself?
*       in the thread itself, that is the cycle method,
*       if itself want to terminate the thread, should never use stop(),
*       but use stop_loop() to set the loop to false and terminate normally.
* 
* @remark when should set the interval_us, and when not?
*       the cycle will invoke util cannot loop, eventhough the return code of cycle is error,
*       so the interval_us used to sleep for each cycle. 
121 122 123 124
*/
class ISrsThreadHandler
{
public:
125 126
    ISrsThreadHandler();
    virtual ~ISrsThreadHandler();
127
public:
128
    virtual void on_thread_start();
129 130 131
    virtual int on_before_cycle();
    virtual int cycle() = 0;
    virtual int on_end_cycle();
132
    virtual void on_thread_stop();
133 134 135 136 137 138 139 140 141
};

/**
* provides servies from st_thread_t,
* for common thread usage.
*/
class SrsThread
{
private:
142
    st_thread_t tid;
143
    int _cid;
144
    bool loop;
145
    bool can_run;
146
    bool really_terminated;
147
    bool _joinable;
148
    const char* _name;
149
private:
150 151
    ISrsThreadHandler* handler;
    int64_t cycle_interval_us;
152
public:
153 154
    /**
    * initialize the thread.
155
    * @param name, human readable name for st debug.
156 157
    * @param thread_handler, the cycle handler for the thread.
    * @param interval_us, the sleep interval when cycle finished.
158 159
    * @param joinable, if joinable, other thread must stop the thread.
    * @remark if joinable, thread never quit itself, or memory leak. 
160
    * @see: https://github.com/simple-rtmp-server/srs/issues/78
161
    * @remark about st debug, see st-1.9/README, _st_iterate_threads_flag
162
    */
163 164
    /**
    * TODO: FIXME: maybe all thread must be reap by others threads, 
165
    * @see: https://github.com/simple-rtmp-server/srs/issues/77
166
    */
167
    SrsThread(const char* name, ISrsThreadHandler* thread_handler, int64_t interval_us, bool joinable);
168
    virtual ~SrsThread();
169
public:
170
    /**
171 172 173 174 175 176
    * get the context id. @see: ISrsThreadContext.get_id().
    * used for parent thread to get the id.
    * @remark when start thread, parent thread will block and wait for this id ready.
    */
    virtual int cid();
    /**
177 178 179 180
    * start the thread, invoke the cycle of handler util
    * user stop the thread.
    * @remark ignore any error of cycle of handler.
    * @remark user can start multiple times, ignore if already started.
181
    * @remark wait for the cid is set by thread pfn.
182 183 184 185 186 187 188
    */
    virtual int start();
    /**
    * stop the thread, wait for the thread to terminate.
    * @remark user can stop multiple times, ignore if already stopped.
    */
    virtual void stop();
189
public:
190 191 192 193 194 195
    /**
    * whether the thread should loop,
    * used for handler->cycle() which has a loop method,
    * to check this method, break if false.
    */
    virtual bool can_loop();
196 197 198 199 200 201
    /**
    * for the loop thread to stop the loop.
    * other thread can directly use stop() to stop loop and wait for quit.
    * this stop loop method only set loop to false.
    */
    virtual void stop_loop();
202
private:
203 204
    virtual void thread_cycle();
    static void* thread_fun(void* arg);
205 206 207
};

#endif
208