Sirikata
|
00001 /* Sirikata Transfer 00002 * DiskManager.hpp 00003 * 00004 * Copyright (c) 2010, Jeff Terrace 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_DiskManager_HPP__ 00034 #define SIRIKATA_DiskManager_HPP__ 00035 00036 #include <sirikata/core/queue/ThreadSafeQueue.hpp> 00037 #include <sirikata/core/util/Thread.hpp> 00038 #include <sirikata/core/util/Singleton.hpp> 00039 #include <sirikata/core/transfer/TransferData.hpp> 00040 #include <boost/filesystem.hpp> 00041 00042 namespace Sirikata { 00043 namespace Transfer { 00044 00045 namespace fs = boost::filesystem; 00046 00047 namespace Filesystem { 00048 typedef boost::filesystem::path Path; 00049 typedef boost::filesystem::file_status FileStatus; 00050 typedef boost::filesystem::file_type FileType; 00051 namespace boost_fs = boost::filesystem; 00052 class PathInfo { 00053 public: 00054 Path mPath; 00055 FileStatus mFileStatus; 00056 PathInfo(Path p, FileStatus f) : mPath(p), mFileStatus(f) {} 00057 }; 00058 } 00059 00060 class SIRIKATA_EXPORT DiskManager 00061 : public AutoSingleton<DiskManager> { 00062 00063 public: 00064 00065 class DiskRequest { 00066 protected: 00067 virtual void execute() = 0; 00068 00069 virtual ~DiskRequest() {} 00070 friend class DiskManager; 00071 }; 00072 00073 class SIRIKATA_EXPORT ScanRequest : public DiskRequest { 00074 public: 00075 typedef std::vector<Filesystem::PathInfo> DirectoryListing; 00076 typedef std::tr1::function<void( 00077 std::tr1::shared_ptr<DirectoryListing> dirListing 00078 )> ScanRequestCallback; 00079 00080 ScanRequest(Filesystem::Path path, ScanRequestCallback cb); 00081 private: 00082 ScanRequestCallback mCb; 00083 Filesystem::Path mPath; 00084 protected: 00085 void execute(); 00086 }; 00087 00088 class SIRIKATA_EXPORT ReadRequest : public DiskRequest { 00089 public: 00090 typedef std::tr1::function<void( 00091 std::tr1::shared_ptr<DenseData> fileContents 00092 )> ReadRequestCallback; 00093 00094 ReadRequest(Filesystem::Path path, ReadRequestCallback cb); 00095 private: 00096 ReadRequestCallback mCb; 00097 Filesystem::Path mPath; 00098 protected: 00099 void execute(); 00100 }; 00101 00102 class SIRIKATA_EXPORT WriteRequest : public DiskRequest { 00103 public: 00104 typedef std::tr1::function<void( 00105 bool status 00106 )> WriteRequestCallback; 00107 00108 WriteRequest(Filesystem::Path path, std::tr1::shared_ptr<DenseData> fileContents, WriteRequestCallback cb); 00109 private: 00110 WriteRequestCallback mCb; 00111 Filesystem::Path mPath; 00112 std::tr1::shared_ptr<DenseData> mFileContents; 00113 protected: 00114 void execute(); 00115 }; 00116 00117 DiskManager(); 00118 ~DiskManager(); 00119 00120 void addRequest(std::tr1::shared_ptr<DiskRequest> req); 00121 00122 static DiskManager& getSingleton(); 00123 static void destroy(); 00124 00125 private: 00126 ThreadSafeQueue<std::tr1::shared_ptr<DiskRequest> > mRequestQueue; 00127 Thread *mWorkerThread; 00128 boost::mutex destroyLock; 00129 boost::condition_variable destroyCV; 00130 00131 void workerThread(); 00132 00133 }; 00134 00135 } 00136 } 00137 00138 #endif /* SIRIKATA_DiskManager_HPP__ */