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
|
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
|
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
|
#import <ObjSQLite3/ObjSQLite3.h>
#import "MTXSQLite3Storage.h"
@implementation MTXSQLite3Storage
{
SL3Connection *_conn;
SL3PreparedStatement *_nextBatchSetStatement, *_nextBatchGetStatement;
}
+ (instancetype)storageWithPath: (OFString *)path
{
return [[[self alloc] initWithPath: path] autorelease];
}
- (instancetype)initWithPath: (OFString *)path
{
self = [super init];
@try {
void *pool = objc_autoreleasePoolPush();
_conn = [[SL3Connection alloc] initWithPath: path];
[self createTables];
_nextBatchSetStatement = [[_conn prepareStatement:
@"INSERT OR REPLACE INTO next_batch (\n"
@" device_id, next_batch\n"
@") VALUES (\n"
@" $device_id, $next_batch\n"
@")"] retain];
_nextBatchGetStatement = [[_conn prepareStatement:
@"SELECT next_batch FROM next_batch\n"
@"WHERE device_id=$device_id"] retain];
objc_autoreleasePoolPop(pool);
} @catch (id e) {
[self release];
@throw e;
}
return self;
}
- (void)dealloc
{
[_nextBatchSetStatement release];
[_nextBatchGetStatement release];
[_conn release];
[super dealloc];
}
- (void)createTables
{
[_conn executeStatement: @"CREATE TABLE IF NOT EXISTS next_batch (\n"
@" device_id TEXT PRIMARY KEY,\n"
@" next_batch TEXT\n"
@")"];
}
- (void)setNextBatch: (OFString *)nextBatch
forDeviceID: (OFString *)deviceID
{
void *pool = objc_autoreleasePoolPush();
[_nextBatchSetStatement reset];
[_nextBatchSetStatement bindWithDictionary: @{
@"$device_id": deviceID,
@"$next_batch": nextBatch
}];
[_nextBatchSetStatement step];
objc_autoreleasePoolPop(pool);
}
- (OFString *)nextBatchForDeviceID: (OFString *)deviceID
{
void *pool = objc_autoreleasePoolPush();
[_nextBatchGetStatement reset];
[_nextBatchGetStatement bindWithDictionary: @{
@"$device_id": deviceID
}];
if (![_nextBatchGetStatement step])
return nil;
OFString *nextBatch =
[[_nextBatchGetStatement rowDictionary][@"next_batch"] retain];
objc_autoreleasePoolPop(pool);
return [nextBatch autorelease];
}
@end
|