blob: ac766df7a81d777cc26639be5acc134239c1e4db (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/** @file gridfs.h */
#pragma once
#include "dbclient.h"
namespace mongo {
typedef unsigned long long gridfs_offset;
class GridFS;
class GridFile;
class Chunk {
public:
Chunk( BSONObj data );
Chunk( BSONObj fileId , int chunkNumber , const char * data , int len );
int len(){
int len;
const char * data = _data["data"].binData( len );
int * foo = (int*)data;
assert( len - 4 == foo[0] );
return len - 4;
}
const char * data( int & len ){
const char * data = _data["data"].binData( len );
int * foo = (int*)data;
assert( len - 4 == foo[0] );
len = len - 4;
return data + 4;
}
private:
BSONObj _data;
friend class GridFS;
};
/**
this is the main entry point into the mongo grid fs
*/
class GridFS{
public:
/**
* @param client - db connection
* @param dbName - root database name
* @param prefix - if you want your data somewhere besides <dbname>.fs
*/
GridFS( DBClientBase& client , const string& dbName , const string& prefix="fs" );
~GridFS();
/**
* puts the file reference by fileName into the db
* @param fileName relative to process
* @return the file object
*/
BSONObj storeFile( const string& fileName );
/**
* removes file referenced by fileName from the db
* @param fileName relative to process
* @return the file object
*/
void removeFile( const string& fileName );
/**
* returns a file object matching the query
*/
GridFile findFile( BSONObj query );
/**
* equiv to findFile( { filename : filename } )
*/
GridFile findFile( const string& fileName );
/**
* convenience method to get all the files
*/
auto_ptr<DBClientCursor> list();
/**
* convenience method to get all the files with a filter
*/
auto_ptr<DBClientCursor> list( BSONObj query );
private:
DBClientBase& _client;
string _dbName;
string _prefix;
string _filesNS;
string _chunksNS;
friend class GridFile;
};
/**
wrapper for a file stored in the Mongo database
*/
class GridFile {
public:
/**
* @return whether or not this file exists
* findFile will always return a GriFile, so need to check this
*/
bool exists(){
return ! _obj.isEmpty();
}
string getFilename(){
return _obj["filename"].str();
}
int getChunkSize(){
return (int)(_obj["chunkSize"].number());
}
gridfs_offset getContentLength(){
return (gridfs_offset)(_obj["length"].number());
}
int getNumChunks(){
return (int) ceil( (double)getContentLength() / (double)getChunkSize() );
}
Chunk getChunk( int n );
/**
write the file to the output stream
*/
gridfs_offset write( ostream & out );
/**
write the file to this filename
*/
gridfs_offset write( const string& where );
private:
GridFile( GridFS * grid , BSONObj obj );
void _exists();
GridFS * _grid;
BSONObj _obj;
friend class GridFS;
};
}
|
