Jonathan Schleifer
1fb20d5791
FossilOrigin-Name: d80ebb2ce13b075e0e67dfb391cd120f2bf99f6ad5492c50d4e71e59dcbd1ac1
110 lines
2.8 KiB
Objective-C
110 lines
2.8 KiB
Objective-C
/*
|
|
* Copyright (c) 2020, 2024 Jonathan Schleifer <js@nil.im>
|
|
*
|
|
* https://fl.nil.im/objsqlite3
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
|
|
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
* AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
|
|
* OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
* PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
#import "SL3Connection.h"
|
|
#import "SL3PreparedStatement.h"
|
|
#import "SL3PreparedStatement+Private.h"
|
|
|
|
#import "SL3ExecuteStatementFailedException.h"
|
|
#import "SL3OpenFailedException.h"
|
|
|
|
@implementation SL3Connection
|
|
+ (instancetype)connectionWithIRI: (OFIRI *)IRI
|
|
{
|
|
return objc_autoreleaseReturnValue([[self alloc] initWithIRI: IRI]);
|
|
}
|
|
|
|
+ (instancetype)connectionWithIRI: (OFIRI *)IRI flags: (int)flags
|
|
{
|
|
return objc_autoreleaseReturnValue([[self alloc] initWithIRI: IRI
|
|
flags: flags]);
|
|
}
|
|
|
|
- (instancetype)initWithIRI: (OFIRI *)IRI
|
|
{
|
|
return [self initWithIRI: IRI
|
|
flags: SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE];
|
|
}
|
|
|
|
- (instancetype)initWithIRI: (OFIRI *)IRI flags: (int)flags
|
|
{
|
|
self = [super init];
|
|
|
|
@try {
|
|
int code = sqlite3_open_v2(
|
|
IRI.fileSystemRepresentation.UTF8String, &_conn, flags,
|
|
NULL);
|
|
|
|
if (code != SQLITE_OK)
|
|
@throw [SL3OpenFailedException exceptionWithIRI: IRI
|
|
flags: flags
|
|
errorCode: code];
|
|
} @catch (id e) {
|
|
objc_release(self);
|
|
@throw e;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc
|
|
{
|
|
sqlite3_close(_conn);
|
|
|
|
[super dealloc];
|
|
}
|
|
|
|
- (SL3PreparedStatement *)prepareStatement: (OFConstantString *)SQLStatement
|
|
{
|
|
return objc_autoreleaseReturnValue([[SL3PreparedStatement alloc]
|
|
sl3_initWithConnection: self
|
|
SQLStatement: SQLStatement]);
|
|
}
|
|
|
|
- (void)executeStatement: (OFConstantString *)SQLStatement
|
|
{
|
|
int code =
|
|
sqlite3_exec(_conn, SQLStatement.UTF8String, NULL, NULL, NULL);
|
|
|
|
if (code != SQLITE_OK)
|
|
@throw [SL3ExecuteStatementFailedException
|
|
exceptionWithConnection: self
|
|
errorCode: code];
|
|
}
|
|
|
|
#ifdef OF_HAVE_BLOCKS
|
|
- (void)transactionWithBlock: (bool (^)(void))block
|
|
{
|
|
bool commit;
|
|
|
|
[self executeStatement: @"BEGIN TRANSACTION"];
|
|
|
|
@try {
|
|
commit = block();
|
|
} @catch (id e) {
|
|
[self executeStatement: @"ROLLBACK TRANSACTION"];
|
|
@throw e;
|
|
}
|
|
|
|
if (commit)
|
|
[self executeStatement: @"COMMIT TRANSACTION"];
|
|
else
|
|
[self executeStatement: @"ROLLBACK TRANSACTION"];
|
|
}
|
|
#endif
|
|
@end
|