-
Notifications
You must be signed in to change notification settings - Fork 228
/
SQLResult.m
123 lines (92 loc) · 1.89 KB
/
SQLResult.m
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
//
// SQLResult.m
// SQLite Test
//
// Created by Dustin Mierau on Tue Apr 02 2002.
// Copyright (c) 2002 Blackhole Media, Inc. All rights reserved.
//
#import "SQLDatabase.h"
#import "SQLDatabasePrivate.h"
@implementation SQLResult
static SEL sRowAtIndexSelector;
static SEL sRowCountSelector;
+ (void)initialize
{
if( self != [SQLResult class])
return;
sRowAtIndexSelector = @selector( rowAtIndex: );
sRowCountSelector = @selector( rowCount );
}
#pragma mark -
-(id)initWithTable:(char**)inTable rows:(int)inRows columns:(int)inColumns
{
if( ![super init])
return nil;
mTable = inTable;
mRows = inRows;
mColumns = inColumns;
return self;
}
-(id)init
{
if( ![super init])
return nil;
mTable = NULL;
mRows = 0;
mColumns = 0;
return self;
}
-(void)dealloc
{
if( mTable )
{
sqlite3_free_table( mTable );
mTable = NULL;
}
[super dealloc];
}
#pragma mark -
-(int)rowCount
{
return mRows;
}
-(int)columnCount
{
return mColumns;
}
#pragma mark -
-(SQLRow*)rowAtIndex:(int)inIndex
{
if( inIndex >= mRows )
return nil;
return [[[SQLRow alloc] initWithColumns:mTable rowData:( mTable + ( ( inIndex + 1 ) * mColumns ) ) columns:mColumns] autorelease];
}
-(NSEnumerator*)rowEnumerator
{
return [[[SQLRowEnumerator allocWithZone:[self zone]] initWithResult:self] autorelease];
}
@end
#pragma mark -
@implementation SQLRowEnumerator
-(id)initWithResult:(SQLResult*)inResult
{
if( ![super init])
return nil;
mResult = [inResult retain];
mPosition = 0;
mRowAtIndexMethod = [mResult methodForSelector:sRowAtIndexSelector];
mRowCountMethod = (int (*)(SQLResult*, SEL))[mResult methodForSelector:sRowCountSelector];
return self;
}
-(void)dealloc
{
[mResult release];
[super dealloc];
}
-(id)nextObject
{
if( mPosition >= (*mRowCountMethod)(mResult, sRowCountSelector) )
return nil;
return (*mRowAtIndexMethod)(mResult, sRowAtIndexSelector, mPosition++);
}
@end