//
|
// LHModelStateMent.m
|
// LHDBDemo
|
//
|
// Created by 3wchina01 on 16/1/25.
|
// Copyright © 2016年 李浩. All rights reserved.
|
//
|
|
#import "LHModelStateMent.h"
|
#import "LHPredicate.h"
|
#import "NSObject+LHModel.h"
|
#import <objc/runtime.h>
|
|
#define CREATE_TABLENAME_HEADER @"CREATE TABLE IF NOT EXISTS "
|
#define INSERT_HEADER @"INSERT INTO "
|
#define UPDATE_HEADER @"UPDATE "
|
#define DELETE_HEADER @"DELETE FROM "
|
#define SELECT_HEADER @"SELECT * FROM "
|
|
|
@implementation LHModelStateMent
|
|
static NSString* tableNameValueString(NSString* type,NSString* name)
|
{
|
NSString* finalStr = @",";
|
NSString* typeStr = (NSString*)type;
|
if ([typeStr isEqualToString:@"i"]) {
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"INT",finalStr];
|
}else if ([typeStr isEqualToString:@"f"]) {
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"FLOAT",finalStr];
|
}else if ([typeStr isEqualToString:@"B"]) {
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"BOOL",finalStr];
|
}else if ([typeStr isEqualToString:@"d"]) {
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"DOUBLE",finalStr];
|
}else if ([typeStr isEqualToString:@"q"]) {
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"LONG",finalStr];
|
}else if ([typeStr isEqualToString:@"NSData"]||[typeStr isEqualToString:@"UIImage"]) {
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"BLOB",finalStr];
|
}else if ([typeStr isEqualToString:@"NSNumber"]){
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"INT",finalStr];
|
} else
|
return [NSString stringWithFormat:@"%@ %@%@",name,@"TEXT",finalStr];
|
}
|
|
static NSDictionary* insertValueString(id value,NSString* name,NSString* type)
|
{
|
if ([value isKindOfClass:[NSNumber class]]) {
|
return @{name:@"?"};
|
}else if ([type isEqualToString:@"UIImage"]) {
|
return @{name:@"?"};
|
}else
|
return @{name:@"?"};
|
}
|
|
static NSString* updateSQL(NSArray* value)
|
{
|
NSMutableString* string = [NSMutableString string];
|
for (NSString* propertyName in value) {
|
[string appendFormat:@" %@ = ?,",propertyName];
|
}
|
if (string.length>0) {
|
[string deleteCharactersInRange:NSMakeRange(string.length-1, 1)];
|
}
|
return string;
|
}
|
|
NSString* createTableString(Class modelClass)
|
{
|
NSMutableString* sqlString = [NSMutableString stringWithString:CREATE_TABLENAME_HEADER];
|
NSDictionary* stateMentDic = [modelClass getAllPropertyNameAndType];
|
[sqlString appendString:NSStringFromClass(modelClass)];
|
NSMutableString* valueStr = [NSMutableString string];
|
[stateMentDic enumerateKeysAndObjectsUsingBlock:^(NSString* key, NSString* obj, BOOL* stop) {
|
[valueStr appendString:tableNameValueString(obj, key)];
|
}];
|
if (valueStr.length>0) {
|
[valueStr deleteCharactersInRange:NSMakeRange(valueStr.length-1, 1)];
|
}
|
[sqlString appendFormat:@"(%@)",valueStr];
|
return sqlString;
|
}
|
|
NSString* addColum(Class modelClass,NSString* propertyName)
|
{
|
NSString* sqlString = [NSString stringWithFormat:@"alter table %@ add %@ %@",NSStringFromClass(modelClass),propertyName,[modelClass getTypeNameWith:propertyName]];
|
return sqlString;
|
}
|
|
NSString* insertString(id model)
|
{
|
NSMutableString* sqlString = [NSMutableString stringWithString:INSERT_HEADER];
|
[sqlString appendString:NSStringFromClass([model class])];
|
NSDictionary* valueDic = [model lh_ModelToDictionary];
|
NSDictionary* typeDic = [object_getClass(model) getAllPropertyNameAndType];
|
NSMutableString* keyStr = [NSMutableString string];
|
NSMutableString* valueStr = [NSMutableString string];
|
for (int i=0; i<valueDic.allKeys.count; i++) {
|
NSDictionary* dic = insertValueString(valueDic[valueDic.allKeys[i]], valueDic.allKeys[i], typeDic[valueDic.allKeys[i]]);
|
[keyStr appendFormat:@"%@,",dic.allKeys[0]];
|
[valueStr appendFormat:@"%@,",dic[dic.allKeys[0]]];
|
}
|
[sqlString appendFormat:@"(%@) VALUES (%@)",[keyStr substringToIndex:keyStr.length-1],[valueStr substringToIndex:valueStr.length-1]];
|
return sqlString;
|
}
|
|
NSString* insertStringWithDic(Class cls, NSDictionary* dic)
|
{
|
NSMutableString* sqlString = [NSMutableString stringWithString:INSERT_HEADER];
|
[sqlString appendString:NSStringFromClass(cls)];
|
NSDictionary* typeDic = [cls getAllPropertyNameAndType];
|
NSMutableString* keyStr = [NSMutableString string];
|
NSMutableString* valueStr = [NSMutableString string];
|
for (int i=0; i<dic.allKeys.count; i++) {
|
NSDictionary* valueDic = insertValueString(dic[dic.allKeys[i]], dic.allKeys[i], typeDic[dic.allKeys[i]]);
|
[keyStr appendFormat:@"%@,",valueDic.allKeys[0]];
|
[valueStr appendFormat:@"%@,",valueDic[valueDic.allKeys[0]]];
|
}
|
[sqlString appendFormat:@"(%@) VALUES (%@)",[keyStr substringToIndex:keyStr.length-1],[valueStr substringToIndex:valueStr.length-1]];
|
return sqlString;
|
}
|
|
NSString* updateString(id model,LHPredicate* predicate)
|
{
|
NSMutableString* updateSql = [NSMutableString stringWithString:UPDATE_HEADER];
|
[updateSql appendFormat:@"%@ set",NSStringFromClass([model class])];
|
[updateSql appendFormat:@"%@",updateSQL([model lh_ModelToDictionary].allKeys)];
|
[updateSql appendFormat:@" WHERE %@",predicate.predicateFormat];
|
return updateSql;
|
}
|
|
NSString* updateStringWithDic(Class cls,LHPredicate* predicate,NSDictionary* valueDic)
|
{
|
NSMutableString* updateSql = [NSMutableString stringWithString:UPDATE_HEADER];
|
[updateSql appendFormat:@"%@ set",NSStringFromClass(cls)];
|
[updateSql appendFormat:@"%@",updateSQL(valueDic.allKeys)];
|
[updateSql appendFormat:@" WHERE %@",predicate.predicateFormat];
|
return updateSql;
|
}
|
|
|
NSString* deleteString(Class modelClass,LHPredicate* predicate)
|
{
|
NSMutableString* deleteStr = [NSMutableString stringWithString:DELETE_HEADER];
|
|
[deleteStr appendString:NSStringFromClass(modelClass)];
|
if (predicate.predicateFormat) {
|
[deleteStr appendFormat:@" WHERE %@",predicate.predicateFormat];
|
}
|
return deleteStr;
|
}
|
|
NSString* selectString(Class modelClass,LHPredicate* predicate)
|
{
|
NSMutableString* selectStr = [NSMutableString stringWithString:SELECT_HEADER];
|
[selectStr appendString:NSStringFromClass(modelClass)];
|
if (predicate.predicateFormat) {
|
[selectStr appendFormat:@" WHERE %@",predicate.predicateFormat];
|
}
|
if (predicate.sortString) {
|
[selectStr appendFormat:@" ORDER BY %@",predicate.sortString];
|
}
|
return selectStr;
|
}
|
|
@end
|
