Sirikata
|
00001 /* Sirikata 00002 * FairServerMessageQueue.hpp 00003 * 00004 * Copyright (c) 2010, Daniel Reiter Horn 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions are 00009 * met: 00010 * * Redistributions of source code must retain the above copyright 00011 * notice, this list of conditions and the following disclaimer. 00012 * * Redistributions in binary form must reproduce the above copyright 00013 * notice, this list of conditions and the following disclaimer in 00014 * the documentation and/or other materials provided with the 00015 * distribution. 00016 * * Neither the name of Sirikata nor the names of its contributors may 00017 * be used to endorse or promote products derived from this software 00018 * without specific prior written permission. 00019 * 00020 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 00021 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 00022 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 00023 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 00024 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 00025 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 00026 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00027 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00028 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00029 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00030 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00031 */ 00032 00033 #ifndef _SIRIKATA_FAIRSENDQUEUE_HPP 00034 #define _SIRIKATA_FAIRSENDQUEUE_HPP 00035 00036 #include <sirikata/core/queue/FairQueue.hpp> 00037 #include "ServerMessageQueue.hpp" 00038 00039 namespace Sirikata { 00040 class FairServerMessageQueue:public ServerMessageQueue { 00041 protected: 00042 struct SenderAdapterQueue { 00043 SenderAdapterQueue(Sender* sender, ServerID sid); 00044 Message* front(); 00045 Message* pop(); 00046 bool empty(); 00047 private: 00048 Sender* mSender; 00049 ServerID mDestServer; 00050 Message* mFront; 00051 }; 00052 00053 typedef FairQueue<Message, ServerID, SenderAdapterQueue> FairSendQueue; 00054 FairSendQueue mServerQueues; 00055 00056 Sirikata::AtomicValue<bool> mServiceScheduled; 00057 00058 uint64 mStoppedBlocked; 00059 uint64 mStoppedUnderflow; 00060 00061 // These must be recursive because networkReadyToSend callbacks may or may 00062 // not be triggered by this code. 00063 mutable boost::recursive_mutex mMutex; 00064 typedef boost::lock_guard<boost::recursive_mutex> MutexLock; 00065 public: 00066 FairServerMessageQueue(SpaceContext* ctx, SpaceNetwork* net, Sender* sender); 00067 ~FairServerMessageQueue(); 00068 00069 protected: 00070 // Must be thread safe: 00071 00072 // Public ServerMessageQueue interface 00073 virtual void messageReady(ServerID sid); 00074 // SpaceNetwork::SendListener Interface 00075 virtual void networkReadyToSend(const ServerID& from); 00076 00077 // Should always be happening inside ServerMessageQueue thread 00078 00079 // ServerMessageReceiver Protected (Implementation) Interface 00080 virtual void handleUpdateReceiverStats(ServerID sid, double total_weight, double used_weight); 00081 00082 // Internal methods 00083 void addInputQueue(ServerID sid, float weight); 00084 void removeInputQueue(ServerID sid); 00085 00086 // Get average weight over all queues. Used when we don't have a weight for 00087 // a new input queue yet. 00088 float getAverageServerWeight() const; 00089 00090 void enableDownstream(ServerID sid); 00091 void disableDownstream(ServerID sid); 00092 00093 // Schedules servicing to occur, but only if it isn't already currently scheduled. 00094 void scheduleServicing(); 00095 // Services the queue, will be called in response to network ready events 00096 // and message ready events since one of those conditions must have changed 00097 // in order to make any additional progress. 00098 void service(); 00099 }; 00100 00101 } 00102 00103 #endif