Support transaction.

a20251313 · Mar 28, 2013 · c4ef632 · c4ef632
1 parent 2b0c522
commit c4ef632
Showing 1 changed file with 23 additions and 3 deletions.
diff --git a/SQLiteDemo/SQLiteDemo/KSViewController.m b/SQLiteDemo/SQLiteDemo/KSViewController.m
@@ -20,7 +20,8 @@ - (void)openDatabase;
 - (void)closeDatabase;
 - (void)createTable;
 
-- (void)excuteSQL:(NSString *)sqlCmd;
+- (BOOL)excuteSQL:(NSString *)sqlCmd;
+- (BOOL)excuteSQLWithCString:(const char *)sqlCmd;
 - (void)insertCustomer:(KSCustomer *)customer;
 - (void)deleteCustomer:(KSCustomer *)customer;
 - (void)updateCustomer:(KSCustomer *)oldValue newValue:(KSCustomer *)newValue;
@@ -93,7 +94,7 @@ - (void)closeDatabase
     }
 }
 
-- (void)excuteSQLWithCString:(const char *)sqlCmd
+- (BOOL)excuteSQLWithCString:(const char *)sqlCmd
 {
     char * errorMsg;
     int state = sqlite3_exec(database, sqlCmd, NULL, NULL, &errorMsg);
@@ -108,9 +109,11 @@ - (void)excuteSQLWithCString:(const char *)sqlCmd
 
         sqlite3_free(errorMsg);
     }
+
+    return (state == SQLITE_OK);
 }
 
-- (void)excuteSQL:(NSString *)sqlCmd
+- (BOOL)excuteSQL:(NSString *)sqlCmd
 {
     char * errorMsg;
     const char * sql = [sqlCmd cStringUsingEncoding:NSUTF8StringEncoding];
@@ -125,6 +128,8 @@ - (void)excuteSQL:(NSString *)sqlCmd
 
         sqlite3_free(errorMsg);
     }
+
+    return (state == SQLITE_OK);
 }
 
 - (void)createTable
@@ -231,6 +236,21 @@ - (NSArray *)queryAllCustomers
     return array;
 }
 
+- (BOOL)beginTransaction
+{
+    return [self excuteSQLWithCString:"BEGIN EXCLUSIVE TRANSACTION;"];
+}
+
+- (BOOL)commit
+{
+    return [self excuteSQLWithCString:"COMMIT TRANSACTION;"];	
+}
+
+- (BOOL)rollback
+{
+    return [self excuteSQLWithCString:"ROLLBACK TRANSACTION;"];
+}
+
 - (IBAction)addButtonPressed:(id)sender
 {
     NSString * name = @"飘飘白云";