Initial commit

2018-08-22 11:15:52 +08:00
parent d98e20881f
commit 249bf6864e
491 changed files with 68665 additions and 7 deletions
@@ -0,0 +1,57 @@
+//
+//  YTKKeyValueStore.h
+//  Ape
+//
+//  Created by TangQiao on 12-11-6.
+//  Copyright (c) 2012年 TangQiao. All rights reserved.
+//
+
+#import <Foundation/Foundation.h>
+
+@interface YTKKeyValueItem : NSObject
+
+@property (strong, nonatomic) NSString *itemId;
+@property (strong, nonatomic) id itemObject;
+@property (strong, nonatomic) NSDate *createdTime;
+
+@end
+
+
+@interface YTKKeyValueStore : NSObject
+
+- (id)initDBWithName:(NSString *)dbName;
+
+- (id)initWithDBWithPath:(NSString *)dbPath;
+
+- (void)createTableWithName:(NSString *)tableName;
+
+- (void)clearTable:(NSString *)tableName;
+
+- (void)close;
+
+///************************ Put&Get methods *****************************************
+
+- (void)putObject:(id)object withId:(NSString *)objectId intoTable:(NSString *)tableName;
+
+- (id)getObjectById:(NSString *)objectId fromTable:(NSString *)tableName;
+
+- (YTKKeyValueItem *)getYTKKeyValueItemById:(NSString *)objectId fromTable:(NSString *)tableName;
+
+- (void)putString:(NSString *)string withId:(NSString *)stringId intoTable:(NSString *)tableName;
+
+- (NSString *)getStringById:(NSString *)stringId fromTable:(NSString *)tableName;
+
+- (void)putNumber:(NSNumber *)number withId:(NSString *)numberId intoTable:(NSString *)tableName;
+
+- (NSNumber *)getNumberById:(NSString *)numberId fromTable:(NSString *)tableName;
+
+- (NSArray *)getAllItemsFromTable:(NSString *)tableName;
+
+- (void)deleteObjectById:(NSString *)objectId fromTable:(NSString *)tableName;
+
+- (void)deleteObjectsByIdArray:(NSArray *)objectIdArray fromTable:(NSString *)tableName;
+
+- (void)deleteObjectsByIdPrefix:(NSString *)objectIdPrefix fromTable:(NSString *)tableName;
+
+
+@end
@@ -0,0 +1,316 @@
+//
+//  YTKKeyValueStore.m
+//  Ape
+//
+//  Created by TangQiao on 12-11-6.
+//  Copyright (c) 2012年 TangQiao. All rights reserved.
+//
+
+#import "YTKKeyValueStore.h"
+#import "FMDatabase.h"
+#import "FMDatabaseQueue.h"
+
+#ifdef DEBUG
+#define debugLog(...)    NSLog(__VA_ARGS__)
+#define debugMethod()    NSLog(@"%s", __func__)
+#define debugError()     NSLog(@"Error at %s Line:%d", __func__, __LINE__)
+#else
+#define debugLog(...)
+#define debugMethod()
+#define debugError()
+#endif
+
+#define PATH_OF_DOCUMENT    [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0]
+
+@implementation YTKKeyValueItem
+
+- (NSString *)description {
+    return [NSString stringWithFormat:@"id=%@, value=%@, timeStamp=%@", _itemId, _itemObject, _createdTime];
+}
+
+@end
+
+@interface YTKKeyValueStore()
+
+@property (strong, nonatomic) FMDatabaseQueue * dbQueue;
+
+@end
+
+@implementation YTKKeyValueStore
+
+static NSString *const DEFAULT_DB_NAME = @"database.sqlite";
+
+static NSString *const CREATE_TABLE_SQL =
+@"CREATE TABLE IF NOT EXISTS %@ ( \
+id TEXT NOT NULL, \
+json TEXT NOT NULL, \
+createdTime TEXT NOT NULL, \
+PRIMARY KEY(id)) \
+";
+
+static NSString *const UPDATE_ITEM_SQL = @"REPLACE INTO %@ (id, json, createdTime) values (?, ?, ?)";
+
+static NSString *const QUERY_ITEM_SQL = @"SELECT json, createdTime from %@ where id = ? Limit 1";
+
+static NSString *const SELECT_ALL_SQL = @"SELECT * from %@";
+
+static NSString *const CLEAR_ALL_SQL = @"DELETE from %@";
+
+static NSString *const DELETE_ITEM_SQL = @"DELETE from %@ where id = ?";
+
+static NSString *const DELETE_ITEMS_SQL = @"DELETE from %@ where id in ( %@ )";
+
+static NSString *const DELETE_ITEMS_WITH_PREFIX_SQL = @"DELETE from %@ where id like ? ";
+
+ (BOOL)checkTableName:(NSString *)tableName {
+    if (tableName == nil || tableName.length == 0 || [tableName rangeOfString:@" "].location != NSNotFound) {
+        debugLog(@"ERROR, table name: %@ format error.", tableName);
+        return NO;
+    }
+    return YES;
+}
+
+- (id)init {
+    return [self initDBWithName:DEFAULT_DB_NAME];
+}
+
+- (id)initDBWithName:(NSString *)dbName {
+    self = [super init];
+    if (self) {
+        NSString * dbPath = [PATH_OF_DOCUMENT stringByAppendingPathComponent:dbName];
+        debugLog(@"dbPath = %@", dbPath);
+        if (_dbQueue) {
+            [self close];
+        }
+        _dbQueue = [FMDatabaseQueue databaseQueueWithPath:dbPath];
+    }
+    return self;
+}
+
+- (id)initWithDBWithPath:(NSString *)dbPath {
+    self = [super init];
+    if (self) {
+        debugLog(@"dbPath = %@", dbPath);
+        if (_dbQueue) {
+            [self close];
+        }
+        _dbQueue = [FMDatabaseQueue databaseQueueWithPath:dbPath];
+    }
+    return self;
+}
+
+- (void)createTableWithName:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSString * sql = [NSString stringWithFormat:CREATE_TABLE_SQL, tableName];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to create table: %@", tableName);
+    }
+}
+
+- (void)clearTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSString * sql = [NSString stringWithFormat:CLEAR_ALL_SQL, tableName];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to clear table: %@", tableName);
+    }
+}
+
+- (void)putObject:(id)object withId:(NSString *)objectId intoTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSError * error;
+    NSData * data = [NSJSONSerialization dataWithJSONObject:object options:0 error:&error];
+    if (error) {
+        debugLog(@"ERROR, faild to get json data");
+        return;
+    }
+    NSString * jsonString = [[NSString alloc] initWithData:data encoding:(NSUTF8StringEncoding)];
+    NSDate * createdTime = [NSDate date];
+    NSString * sql = [NSString stringWithFormat:UPDATE_ITEM_SQL, tableName];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql, objectId, jsonString, createdTime];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to insert/replace into table: %@", tableName);
+    }
+}
+
+- (id)getObjectById:(NSString *)objectId fromTable:(NSString *)tableName {
+    YTKKeyValueItem * item = [self getYTKKeyValueItemById:objectId fromTable:tableName];
+    if (item) {
+        return item.itemObject;
+    } else {
+        return nil;
+    }
+}
+
+- (YTKKeyValueItem *)getYTKKeyValueItemById:(NSString *)objectId fromTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return nil;
+    }
+    NSString * sql = [NSString stringWithFormat:QUERY_ITEM_SQL, tableName];
+    __block NSString * json = nil;
+    __block NSDate * createdTime = nil;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        FMResultSet * rs = [db executeQuery:sql, objectId];
+        if ([rs next]) {
+            json = [rs stringForColumn:@"json"];
+            createdTime = [rs dateForColumn:@"createdTime"];
+        }
+        [rs close];
+    }];
+    if (json) {
+        NSError * error;
+        id result = [NSJSONSerialization JSONObjectWithData:[json dataUsingEncoding:NSUTF8StringEncoding]
+                                                    options:(NSJSONReadingAllowFragments) error:&error];
+        if (error) {
+            debugLog(@"ERROR, faild to prase to json");
+            return nil;
+        }
+        YTKKeyValueItem * item = [[YTKKeyValueItem alloc] init];
+        item.itemId = objectId;
+        item.itemObject = result;
+        item.createdTime = createdTime;
+        return item;
+    } else {
+        return nil;
+    }
+}
+
+- (void)putString:(NSString *)string withId:(NSString *)stringId intoTable:(NSString *)tableName {
+    if (string == nil) {
+        debugLog(@"error, string is nil");
+        return;
+    }
+    [self putObject:@[string] withId:stringId intoTable:tableName];
+}
+
+- (NSString *)getStringById:(NSString *)stringId fromTable:(NSString *)tableName {
+    NSArray * array = [self getObjectById:stringId fromTable:tableName];
+    if (array && [array isKindOfClass:[NSArray class]]) {
+        return array[0];
+    }
+    return nil;
+}
+
+- (void)putNumber:(NSNumber *)number withId:(NSString *)numberId intoTable:(NSString *)tableName {
+    if (number == nil) {
+        debugLog(@"error, number is nil");
+        return;
+    }
+    [self putObject:@[number] withId:numberId intoTable:tableName];
+}
+
+- (NSNumber *)getNumberById:(NSString *)numberId fromTable:(NSString *)tableName {
+    NSArray * array = [self getObjectById:numberId fromTable:tableName];
+    if (array && [array isKindOfClass:[NSArray class]]) {
+        return array[0];
+    }
+    return nil;
+}
+
+- (NSArray *)getAllItemsFromTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return nil;
+    }
+    NSString * sql = [NSString stringWithFormat:SELECT_ALL_SQL, tableName];
+    __block NSMutableArray * result = [NSMutableArray array];
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        FMResultSet * rs = [db executeQuery:sql];
+        while ([rs next]) {
+            YTKKeyValueItem * item = [[YTKKeyValueItem alloc] init];
+            item.itemId = [rs stringForColumn:@"id"];
+            item.itemObject = [rs stringForColumn:@"json"];
+            item.createdTime = [rs dateForColumn:@"createdTime"];
+            [result addObject:item];
+        }
+        [rs close];
+    }];
+    // parse json string to object
+    NSError * error;
+    for (YTKKeyValueItem * item in result) {
+        error = nil;
+        id object = [NSJSONSerialization JSONObjectWithData:[item.itemObject dataUsingEncoding:NSUTF8StringEncoding]
+                                                    options:(NSJSONReadingAllowFragments) error:&error];
+        if (error) {
+            debugLog(@"ERROR, faild to prase to json.");
+        } else {
+            item.itemObject = object;
+        }
+    }
+    return result;
+}
+
+- (void)deleteObjectById:(NSString *)objectId fromTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSString * sql = [NSString stringWithFormat:DELETE_ITEM_SQL, tableName];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql, objectId];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to delete item from table: %@", tableName);
+    }
+}
+
+- (void)deleteObjectsByIdArray:(NSArray *)objectIdArray fromTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSMutableString *stringBuilder = [NSMutableString string];
+    for (id objectId in objectIdArray) {
+        NSString *item = [NSString stringWithFormat:@" '%@' ", objectId];
+        if (stringBuilder.length == 0) {
+            [stringBuilder appendString:item];
+        } else {
+            [stringBuilder appendString:@","];
+            [stringBuilder appendString:item];
+        }
+    }
+    NSString *sql = [NSString stringWithFormat:DELETE_ITEMS_SQL, tableName, stringBuilder];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to delete items by ids from table: %@", tableName);
+    }
+}
+
+- (void)deleteObjectsByIdPrefix:(NSString *)objectIdPrefix fromTable:(NSString *)tableName {
+    if ([YTKKeyValueStore checkTableName:tableName] == NO) {
+        return;
+    }
+    NSString *sql = [NSString stringWithFormat:DELETE_ITEMS_WITH_PREFIX_SQL, tableName];
+    NSString *prefixArgument = [NSString stringWithFormat:@"%@%%", objectIdPrefix];
+    __block BOOL result;
+    [_dbQueue inDatabase:^(FMDatabase *db) {
+        result = [db executeUpdate:sql, prefixArgument];
+    }];
+    if (!result) {
+        debugLog(@"ERROR, failed to delete items by id prefix from table: %@", tableName);
+    }
+}
+
+- (void)close {
+    [_dbQueue close];
+    _dbQueue = nil;
+}
+
+@end