diff options
| author | Mike Buland <eichlan@xagasoft.com> | 2010-05-10 07:15:05 +0000 |
|---|---|---|
| committer | Mike Buland <eichlan@xagasoft.com> | 2010-05-10 07:15:05 +0000 |
| commit | 9731dc86fa9b12adc064b99910dddb58932c71cf (patch) | |
| tree | 00a358ff397e2b7b0fc3377564288f5c86b850c5 /src/cachestorefiles.h | |
| parent | 8baf7e1e75a185c742dc6d5b27e50058635e5522 (diff) | |
| download | libbu++-9731dc86fa9b12adc064b99910dddb58932c71cf.tar.gz libbu++-9731dc86fa9b12adc064b99910dddb58932c71cf.tar.bz2 libbu++-9731dc86fa9b12adc064b99910dddb58932c71cf.tar.xz libbu++-9731dc86fa9b12adc064b99910dddb58932c71cf.zip | |
Added the new Bu::CacheStoreFiles, it's an uber-simple cache storage system that
maybe would be better to call an example than a fully fledged storage strategy.
It just names files based on your keys. It's very slow, and very wasteful, and
shouldn't be used long-term in most normal cache systems.
Diffstat (limited to 'src/cachestorefiles.h')
| -rw-r--r-- | src/cachestorefiles.h | 192 |
1 files changed, 192 insertions, 0 deletions
diff --git a/src/cachestorefiles.h b/src/cachestorefiles.h new file mode 100644 index 0000000..6b37d0a --- /dev/null +++ b/src/cachestorefiles.h | |||
| @@ -0,0 +1,192 @@ | |||
| 1 | /* | ||
| 2 | * Copyright (C) 2007-2010 Xagasoft, All rights reserved. | ||
| 3 | * | ||
| 4 | * This file is part of the libbu++ library and is released under the | ||
| 5 | * terms of the license contained in the file LICENSE. | ||
| 6 | */ | ||
| 7 | |||
| 8 | #ifndef BU_CACHE_STORE_FILES_H | ||
| 9 | #define BU_CACHE_STORE_FILES_H | ||
| 10 | |||
| 11 | #include "bu/fstring.h" | ||
| 12 | #include "bu/file.h" | ||
| 13 | #include "bu/cachestore.h" | ||
| 14 | #include "bu/archive.h" | ||
| 15 | #include "bu/membuf.h" | ||
| 16 | #include "bu/formatter.h" | ||
| 17 | #include "bu/sio.h" | ||
| 18 | |||
| 19 | #include <sys/types.h> | ||
| 20 | #include <sys/stat.h> | ||
| 21 | #include <dirent.h> | ||
| 22 | #include <unistd.h> | ||
| 23 | |||
| 24 | namespace Bu | ||
| 25 | { | ||
| 26 | template<class keytype, class obtype> | ||
| 27 | keytype __cacheGetKey( const obtype *pObj ); | ||
| 28 | |||
| 29 | template<class keytype, class obtype> | ||
| 30 | obtype *__cacheStoreFilesAlloc( const keytype &key ) | ||
| 31 | { | ||
| 32 | return new obtype(); | ||
| 33 | } | ||
| 34 | |||
| 35 | template<class keytype, class obtype> | ||
| 36 | void __cacheStoreFilesStore( Bu::Stream &s, obtype &rObj, | ||
| 37 | const keytype & ) | ||
| 38 | { | ||
| 39 | Bu::Archive ar( s, Bu::Archive::save ); | ||
| 40 | ar << rObj; | ||
| 41 | } | ||
| 42 | |||
| 43 | template<class keytype, class obtype> | ||
| 44 | obtype *__cacheStoreFilesLoad( Bu::Stream &s, const keytype &key ) | ||
| 45 | { | ||
| 46 | obtype *pObj = __cacheStoreFilesAlloc<keytype, obtype>( key ); | ||
| 47 | Bu::Archive ar( s, Bu::Archive::load ); | ||
| 48 | ar >> (*pObj); | ||
| 49 | return pObj; | ||
| 50 | } | ||
| 51 | |||
| 52 | template<class keytype, class obtype> | ||
| 53 | class CacheStoreFiles : public CacheStore<keytype, obtype> | ||
| 54 | { | ||
| 55 | public: | ||
| 56 | CacheStoreFiles( const Bu::FString &sPrefix ) : | ||
| 57 | sPrefix( sPrefix ) | ||
| 58 | { | ||
| 59 | if( access( sPrefix.getStr(), W_OK|R_OK|X_OK ) ) | ||
| 60 | { | ||
| 61 | mkdir( sPrefix.getStr(), 0755 ); | ||
| 62 | } | ||
| 63 | } | ||
| 64 | |||
| 65 | virtual ~CacheStoreFiles() | ||
| 66 | { | ||
| 67 | } | ||
| 68 | |||
| 69 | virtual obtype *load( const keytype &key ) | ||
| 70 | { | ||
| 71 | Bu::MemBuf mb; | ||
| 72 | Bu::Formatter f( mb ); | ||
| 73 | f << sPrefix << "/" << key; | ||
| 74 | Bu::File fIn( mb.getString(), Bu::File::Read ); | ||
| 75 | obtype *pOb = __cacheStoreFilesLoad<keytype, obtype>( fIn, key ); | ||
| 76 | return pOb; | ||
| 77 | } | ||
| 78 | |||
| 79 | virtual void unload( obtype *pObj, const keytype &key ) | ||
| 80 | { | ||
| 81 | delete pObj; | ||
| 82 | } | ||
| 83 | |||
| 84 | virtual keytype create( obtype *pSrc ) | ||
| 85 | { | ||
| 86 | keytype key = __cacheGetKey<keytype, obtype>( pSrc ); | ||
| 87 | Bu::MemBuf mb; | ||
| 88 | Bu::Formatter f( mb ); | ||
| 89 | f << sPrefix << "/" << key; | ||
| 90 | |||
| 91 | Bu::File fTouch( mb.getString(), Bu::File::WriteNew ); | ||
| 92 | |||
| 93 | return key; | ||
| 94 | } | ||
| 95 | |||
| 96 | virtual void sync() | ||
| 97 | { | ||
| 98 | } | ||
| 99 | |||
| 100 | virtual void sync( obtype *pSrc, const keytype &key ) | ||
| 101 | { | ||
| 102 | Bu::MemBuf mb; | ||
| 103 | Bu::Formatter f( mb ); | ||
| 104 | f << sPrefix << "/" << key; | ||
| 105 | |||
| 106 | Bu::File fOut( mb.getString(), Bu::File::WriteNew ); | ||
| 107 | __cacheStoreFilesStore<keytype, obtype>( fOut, *pSrc, key ); | ||
| 108 | } | ||
| 109 | |||
| 110 | virtual void destroy( obtype *pObj, const keytype &key ) | ||
| 111 | { | ||
| 112 | Bu::MemBuf mb; | ||
| 113 | Bu::Formatter f( mb ); | ||
| 114 | f << sPrefix << "/" << key; | ||
| 115 | |||
| 116 | unlink( mb.getString().getStr() ); | ||
| 117 | delete pObj; | ||
| 118 | } | ||
| 119 | |||
| 120 | virtual void destroy( const keytype &key ) | ||
| 121 | { | ||
| 122 | Bu::MemBuf mb; | ||
| 123 | Bu::Formatter f( mb ); | ||
| 124 | f << sPrefix << "/" << key; | ||
| 125 | |||
| 126 | unlink( mb.getString().getStr() ); | ||
| 127 | } | ||
| 128 | |||
| 129 | virtual bool has( const keytype &key ) | ||
| 130 | { | ||
| 131 | Bu::MemBuf mb; | ||
| 132 | Bu::Formatter f( mb ); | ||
| 133 | f << sPrefix << "/" << key; | ||
| 134 | |||
| 135 | return access( mb.getString().getStr(), F_OK ) == 0; | ||
| 136 | } | ||
| 137 | |||
| 138 | virtual Bu::List<keytype> getKeys() | ||
| 139 | { | ||
| 140 | DIR *dir = opendir( sPrefix.getStr() ); | ||
| 141 | struct dirent *de; | ||
| 142 | Bu::List<keytype> lKeys; | ||
| 143 | |||
| 144 | while( (de = readdir( dir ) ) ) | ||
| 145 | { | ||
| 146 | if( de->d_type != DT_REG ) | ||
| 147 | continue; | ||
| 148 | |||
| 149 | keytype tmp; | ||
| 150 | Bu::MemBuf mb( de->d_name ); | ||
| 151 | Bu::Formatter f( mb ); | ||
| 152 | try | ||
| 153 | { | ||
| 154 | f >> tmp; | ||
| 155 | } | ||
| 156 | catch( Bu::ExceptionBase &e ) | ||
| 157 | { | ||
| 158 | Bu::sio << "Parse error in dir-scan: " << e.what() | ||
| 159 | << Bu::sio.nl; | ||
| 160 | } | ||
| 161 | lKeys.append( tmp ); | ||
| 162 | } | ||
| 163 | closedir( dir ); | ||
| 164 | |||
| 165 | return lKeys; | ||
| 166 | } | ||
| 167 | |||
| 168 | virtual int getSize() | ||
| 169 | { | ||
| 170 | DIR *dir = opendir( sPrefix.getStr() ); | ||
| 171 | struct dirent *de; | ||
| 172 | int iCount = 0; | ||
| 173 | |||
| 174 | while( (de = readdir( dir ) ) ) | ||
| 175 | { | ||
| 176 | if( de->d_type != DT_REG ) | ||
| 177 | continue; | ||
| 178 | |||
| 179 | iCount++; | ||
| 180 | } | ||
| 181 | closedir( dir ); | ||
| 182 | |||
| 183 | return iCount; | ||
| 184 | } | ||
| 185 | |||
| 186 | private: | ||
| 187 | Bu::FString sPrefix; | ||
| 188 | }; | ||
| 189 | |||
| 190 | }; | ||
| 191 | |||
| 192 | #endif | ||
