← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/pbms/src/cloud_ms.h

  • Committer: Mark Atwood
  • Date: 2011-12-20 02:32:53 UTC
  • mfrom: (2469.1.1 drizzle-build)
  • Revision ID: me@mark.atwood.name-20111220023253-bvu0kr14kwsdvz7g
merge lp:~brianaker/drizzle/deprecate-pbms

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/pbms/src/cloud_ms.h
1
 
#pragma once
2
 
#ifndef __CLOUD_H__
3
 
#define __CLOUD_H__
4
 
/* Copyright (C) 2009 PrimeBase Technologies GmbH, Germany
5
 
 *
6
 
 * PrimeBase Media Stream for MySQL
7
 
 *
8
 
 * This program is free software; you can redistribute it and/or modify
9
 
 * it under the terms of the GNU General Public License as published by
10
 
 * the Free Software Foundation; either version 2 of the License, or
11
 
 * (at your option) any later version.
12
 
 *
13
 
 * This program is distributed in the hope that it will be useful,
14
 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16
 
 * GNU General Public License for more details.
17
 
 *
18
 
 * You should have received a copy of the GNU General Public License
19
 
 * along with this program; if not, write to the Free Software
20
 
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
21
 
 *
22
 
 *  Created by Barry Leslie on 3/20/09.
23
 
 *
24
 
 */
25
 
#include "cslib/CSMd5.h"
26
 
#include <inttypes.h>
27
 
 
28
 
 
29
 
/* NOTES:
30
 
 *
31
 
 * - TODO: If cl_deleteData() fails then the BLOB deletion must fail and be rescheduled to try again
32
 
 *                      later.
33
 
 * - TODO: Copying of BLOBs from one database to another needs to be handled. Look for copyBlob() and 
34
 
 *                      resetBlobHead(). There are 3 cases to handle depending on if the databases involved use
35
 
 *                      cload storage.
36
 
 */
37
 
 
38
 
//===============================
39
 
class CSS3Protocol;     
40
 
class MSCloudInfo : public CSRefObject {
41
 
        private:
42
 
        static uint32_t gMaxInfoRef;
43
 
        static CSSyncSparseArray *gCloudInfo;
44
 
 
45
 
        friend class MSCloudTable;
46
 
        friend class CloudDB;
47
 
        
48
 
        private:        
49
 
        uint32_t                        cloudRefId;
50
 
        CSString                *bucket;
51
 
        CSS3Protocol    *s3Prot;
52
 
        
53
 
public:
54
 
 
55
 
        static void startUp()
56
 
        {
57
 
                new_(gCloudInfo, CSSyncSparseArray(5));
58
 
                gMaxInfoRef = 0;
59
 
        }
60
 
        
61
 
        static void shutDown()
62
 
        {
63
 
                if (gCloudInfo) {
64
 
                        gCloudInfo->clear();
65
 
                        gCloudInfo->release();
66
 
                        gCloudInfo = NULL;
67
 
                }       
68
 
        }
69
 
 
70
 
 
71
 
        static MSCloudInfo *getCloudInfo(uint32_t in_cloudRefId)
72
 
        {
73
 
                MSCloudInfo *info;
74
 
                enter_();
75
 
                
76
 
                lock_(gCloudInfo);
77
 
                
78
 
                info = (MSCloudInfo *) gCloudInfo->get(in_cloudRefId);
79
 
                if (!info) {
80
 
                        char msg[80];
81
 
                        snprintf(msg, 80, "Cloud info with reference ID %"PRIu32" not found", in_cloudRefId);
82
 
                        CSException::throwException(CS_CONTEXT, CS_ERR_GENERIC_ERROR, msg);
83
 
                }
84
 
                info->retain();
85
 
                unlock_(gCloudInfo);
86
 
                return_(info);
87
 
        }
88
 
        
89
 
        MSCloudInfo(uint32_t id, const char *server, const char *bucket, const char *publicKey, const char *privateKey );
90
 
        ~MSCloudInfo();
91
 
        
92
 
        uint32_t getCloudRefId() { return cloudRefId;}
93
 
        
94
 
        const char *getServer();
95
 
        
96
 
        const char *getBucket();
97
 
        
98
 
        const char *getPublicKey();
99
 
        
100
 
        const char *getPrivateKey();
101
 
 
102
 
        CSString *getSignature(const char *key, const char *content_type, uint32_t *s3AuthorizationTime);
103
 
        
104
 
        CSString *getDataURL(const char *key, int keep_alive);
105
 
 
106
 
        void send(CSInputStream *input, const char *key, off64_t size);
107
 
        
108
 
        void receive(CSOutputStream *output, const char *key);
109
 
        
110
 
        void copy(MSCloudInfo *dst_cloud, const char *dst_key, const char *src_key);
111
 
 
112
 
        void cDelete(const char *key);
113
 
        
114
 
        CSVector *list(const char *key_prefix, uint32_t max = 0);
115
 
};
116
 
 
117
 
typedef struct CloudKey {
118
 
        uint32_t creation_time;
119
 
        uint32_t ref_index;             // Just a sequence counter in case 2 blobs have the same creation time.
120
 
        uint32_t cloud_ref;             // A reference into the pbms.pbms_cloud table.
121
 
} CloudKeyRec, *CloudKeyPtr;
122
 
 
123
 
 
124
 
//===============================
125
 
class CloudObjectKey : public CSStringBuffer
126
 
{
127
 
        uint32_t default_db_id;
128
 
        
129
 
        public:
130
 
        CloudObjectKey(uint32_t id): CSStringBuffer(), default_db_id(id){ }
131
 
        ~CloudObjectKey(){}
132
 
        
133
 
        static const uint32_t base_key_size = 64; // enough space for <db_id>/<backup_id>/<creation_time>/<ref_index>
134
 
 
135
 
        void setObjectKey(const char *object_key)
136
 
        {
137
 
                setLength(base_key_size + strlen(object_key) +1);
138
 
                
139
 
                snprintf(getBuffer(0), length(), "%"PRIu32"/0/%s",default_db_id,  object_key);
140
 
        }
141
 
        
142
 
        void setObjectKey(CloudKeyPtr key = NULL, uint32_t backup_id = 0, uint32_t db_id = 0)
143
 
        {
144
 
                if (!db_id) db_id = default_db_id;
145
 
                setLength(base_key_size);
146
 
                
147
 
                if (key)
148
 
                        snprintf(getBuffer(0), length(), "%"PRIu32"/%"PRIu32"/%"PRIu32".%"PRIu32".%"PRIu32"", db_id, backup_id, key->cloud_ref, key->creation_time, key->ref_index);
149
 
                else 
150
 
                        snprintf(getBuffer(0), length(), "%"PRIu32"/%"PRIu32"s/", db_id, backup_id);
151
 
                        
152
 
        }
153
 
        
154
 
        static void parseObjectKey(const char *object_key, CloudKeyPtr key, uint32_t *backup_id = NULL, uint32_t *db_id = NULL)
155
 
        {
156
 
                uint32_t v1;
157
 
                
158
 
                if (!backup_id) backup_id = &v1;
159
 
                if (!db_id) db_id = &v1;
160
 
                
161
 
                sscanf(object_key, "%"PRIu32"/%"PRIu32"/%"PRIu32".%"PRIu32".%"PRIu32"", db_id, backup_id, &(key->cloud_ref), &(key->creation_time), &(key->ref_index));
162
 
        }
163
 
};
164
 
 
165
 
//===============================
166
 
class MSBackupInfo;
167
 
class CloudDB: public CSRefObject {
168
 
        
169
 
private:
170
 
        static uint32_t gKeyIndex;
171
 
        static CSMutex  gCloudKeyLock;
172
 
        
173
 
        uint32_t        dfltCloudRefId;
174
 
        
175
 
        uint32_t        keep_alive; // The length of time a redirect URL will remain valid. In seconds.
176
 
        uint32_t        blob_recovery_no; // This is the backup number from which the recovery should be done.
177
 
        uint32_t        blob_db_id;
178
 
        
179
 
        bool isBackup;
180
 
        MSBackupInfo *backupInfo;
181
 
        MSCloudInfo     *backupCloud;
182
 
        
183
 
        static const uint32_t base_key_size = 64; // enough space for <db_id>/<backup_id>/<creation_time>/<ref_index>
184
 
 
185
 
public:
186
 
        CSStringBuffer          *clObjectKey;
187
 
        
188
 
        CloudDB(uint32_t db_id);
189
 
        ~CloudDB();
190
 
        
191
 
        void cl_setDefaultCloudRef(uint32_t dflt) { dfltCloudRefId = dflt;}
192
 
        uint32_t cl_getDefaultCloudRef() { return dfltCloudRefId;}
193
 
 
194
 
        MSCloudInfo *cl_getCloudInfo(uint32_t cloudRefId = 0)
195
 
        {
196
 
                return MSCloudInfo::getCloudInfo((cloudRefId)?cloudRefId:dfltCloudRefId);
197
 
        }
198
 
        
199
 
        void cl_getNewKey(CloudKeyPtr key)
200
 
        {
201
 
                enter_();
202
 
                lock_(&gCloudKeyLock);  
203
 
                
204
 
                key->creation_time = time(NULL);
205
 
                key->ref_index = gKeyIndex++;
206
 
                key->cloud_ref = dfltCloudRefId;
207
 
                
208
 
                unlock_(&gCloudKeyLock);        
209
 
                exit_();
210
 
        }
211
 
 
212
 
        bool cl_mustRecoverBlobs() { return (blob_recovery_no != 0);}
213
 
        
214
 
        void cl_setRecoveryNumber(const char *number)
215
 
        {
216
 
                blob_recovery_no = atol(number);
217
 
        }
218
 
 
219
 
        const char *cl_getRecoveryNumber()
220
 
        {
221
 
                static char number[20];
222
 
                
223
 
                snprintf(number, 20, "%"PRIu32"", blob_recovery_no);
224
 
                return number;
225
 
        }
226
 
 
227
 
        CSString *cl_getObjectKey(CloudKeyPtr key)
228
 
        {
229
 
                CloudObjectKey *objectKey;
230
 
                enter_();
231
 
                
232
 
                new_(objectKey, CloudObjectKey(blob_db_id));
233
 
                push_(objectKey);
234
 
                
235
 
                objectKey->setObjectKey(key);
236
 
                
237
 
                CSString *str = CSString::newString(objectKey->getCString());
238
 
                release_(objectKey);
239
 
                
240
 
                return_(str);
241
 
        }
242
 
        
243
 
        void cl_setKeepAlive(uint32_t keep_alive_arg) {keep_alive = keep_alive_arg;}
244
 
        
245
 
        void cl_createDB();
246
 
        void cl_dropDB();
247
 
        void cl_restoreDB();
248
 
        uint32_t cl_getNextBackupNumber(uint32_t cloud_ref = 0);
249
 
        bool cl_dbExists();
250
 
        
251
 
        // setting backup_blob_no to -1 ensures that if the database is dropped no BLOBs will be deleted.
252
 
        void cl_setCloudIsBackup(){ isBackup = true;}
253
 
        void cl_setBackupInfo(MSBackupInfo *info){ backupInfo = info;}
254
 
        MSBackupInfo *cl_getBackupInfo();
255
 
        void cl_clearBackupInfo();
256
 
        
257
 
        void cl_backupBLOB(CloudKeyPtr key);
258
 
        void cl_restoreBLOB(CloudKeyPtr key, uint32_t backup_db_id);
259
 
 
260
 
        void cl_putData( CloudKeyPtr key, CSInputStream *stream, off64_t size);
261
 
        off64_t cl_getData(CloudKeyPtr key, char *data, off64_t size);
262
 
        CSString *cl_getDataURL(CloudKeyPtr key);
263
 
        void cl_deleteData(CloudKeyPtr key);
264
 
        CSString *cl_getSignature(CloudKeyPtr key, CSString *content_type, uint32_t *s3AuthorizationTime);
265
 
        
266
 
};
267
 
 
268
 
#endif // __CLOUD_H__