Merge pull request #29 from zhenian/master

support drop table
kanyun-inc · Aug 15, 2016 · 7ac052d · 7ac052d
2 parents be23a3c + e89c525
commit 7ac052d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/YTKKeyValueStore/YTKKeyValueStore.h b/YTKKeyValueStore/YTKKeyValueStore.h
@@ -29,6 +29,8 @@
 
 - (void)clearTable:(NSString *)tableName;
 
+- (void)dropTable:(NSString *)tableName;
+
 - (void)close;
 
 ///************************ Put&Get methods *****************************************

diff --git a/YTKKeyValueStore/YTKKeyValueStore.m b/YTKKeyValueStore/YTKKeyValueStore.m
@@ -65,6 +65,8 @@ PRIMARY KEY(id)) \
 
 static NSString *const DELETE_ITEMS_WITH_PREFIX_SQL = @"DELETE from %@ where id like ? ";
 
+static NSString *const DROP_TABLE_SQL = @" DROP TABLE '%@' ";
+
 + (BOOL)checkTableName:(NSString *)tableName {
     if (tableName == nil || tableName.length == 0 || [tableName rangeOfString:@" "].location != NSNotFound) {
         debugLog(@"ERROR, table name: %@ format error.", tableName);
@@ -144,6 +146,20 @@ - (void)clearTable:(NSString *)tableName {
     }
 }
 
+- (void)dropTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSString * sql = [NSString stringWithFormat:DROP_TABLE_SQL, tableName];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to drop table: %@", tableName);
+    }
+}
+
 - (void)putObject:(id)object withId:(NSString *)objectId intoTable:(NSString *)tableName {
     if ([YTKKeyValueStore checkTableName:tableName] == NO) {
         return;