Fuzzy query

Author: changsanjiang

SJDBMap/SJDatabaseMap+Server.h

@@ -57,6 +57,11 @@ extern char *_sjmystrcat(char *dst, const char *src);
  */
 - (NSArray<id<SJDBMapUseProtocol>> *)sjQueryConversionMolding:(Class)cls dict:(NSDictionary *)dict;
 
+/*!
+ *  根据条件模糊查询
+ */
+- (NSArray<id<SJDBMapUseProtocol>> *)sjFuzzyQueryConversionMolding:(Class)cls match:(SJDatabaseMapFuzzyMatch)match dict:(NSDictionary *)dict;
+
 /*!
  *  查询数据库原始存储数据
  */

SJDBMap/SJDatabaseMap+Server.m

@@ -175,7 +175,10 @@ - (void)sjAutoCreateOrAlterRelevanceTabWithClass:(Class)cls {
     NSMutableString *fieldsSqlM = [NSMutableString new];
     [fieldsSqlM appendFormat:@"select * from %s where ", tabName];
     [dict enumerateKeysAndObjectsUsingBlock:^(NSString * _Nonnull key, id  _Nonnull obj, BOOL * _Nonnull stop) {
-        [fieldsSqlM appendFormat:@"%@ = '%@'", key, obj];
+        if ( [obj isKindOfClass:[NSString class]] && [(NSString *)obj containsString:@"'"] )
+            [fieldsSqlM appendFormat:@"%@ = \"%@\"", key, obj];
+        else
+            [fieldsSqlM appendFormat:@"%@ = '%@'", key, obj];
         [fieldsSqlM appendString:@" and "];
     }];
     [fieldsSqlM deleteCharactersInRange:NSMakeRange(fieldsSqlM.length - 5, 5)];
@@ -189,6 +192,66 @@ - (void)sjAutoCreateOrAlterRelevanceTabWithClass:(Class)cls {
     return [self _sjConversionMolding:cls rawStorageData:incompleteData];
 }
 
+/*!
+ *  根据条件模糊查询
+ */
+- (NSArray<id<SJDBMapUseProtocol>> *)sjFuzzyQueryConversionMolding:(Class)cls match:(SJDatabaseMapFuzzyMatch)match dict:(NSDictionary *)dict {
+    SJDBMapUnderstandingModel *uM = [self sjGetUnderstandingWithClass:cls];
+    if ( !uM.primaryKey && !uM.autoincrementPrimaryKey ) return nil;
+    NSAssert(uM.primaryKey || uM.autoincrementPrimaryKey, @"[%@] 该类没有设置主键", cls);
+    
+    const char *tabName = [self sjGetTabName:cls];
+    
+    NSMutableString *fieldsSqlM = [NSMutableString new];
+    [fieldsSqlM appendFormat:@"select * from %s where ", tabName];
+    [dict enumerateKeysAndObjectsUsingBlock:^(NSString * _Nonnull key, id  _Nonnull obj, BOOL * _Nonnull stop) {
+        
+        switch (match) {
+                //      *  ...A...
+            case SJDatabaseMapFuzzyMatchAll:
+            {
+                if ( [obj isKindOfClass:[NSString class]] && [(NSString *)obj containsString:@"'"] )
+                    [fieldsSqlM appendFormat:@"%@ like \"%%%@%%\"", key, obj];
+                else
+                    [fieldsSqlM appendFormat:@"%@ like '%%%@%%'", key, obj];
+            }
+                break;
+                //      *  ABC.....
+            case SJDatabaseMapFuzzyMatchFront:
+            {
+                if ( [obj isKindOfClass:[NSString class]] && [(NSString *)obj containsString:@"'"] )
+                    [fieldsSqlM appendFormat:@"%@ like \"%@%%\"", key, obj];
+                else
+                    [fieldsSqlM appendFormat:@"%@ like '%@%%'", key, obj];
+            }
+                break;
+                //     *  ...DEF
+            case SJDatabaseMapFuzzyMatchLater:
+            {
+                if ( [obj isKindOfClass:[NSString class]] && [(NSString *)obj containsString:@"'"] )
+                    [fieldsSqlM appendFormat:@"%@ like \"%%%@\"", key, obj];
+                else
+                    [fieldsSqlM appendFormat:@"%@ like '%%%@'", key, obj];
+            }
+                break;
+            default:
+                break;
+        }
+        
+        [fieldsSqlM appendString:@" and "];
+    }];
+    [fieldsSqlM deleteCharactersInRange:NSMakeRange(fieldsSqlM.length - 5, 5)];
+    [fieldsSqlM appendString:@";"];
+    
+    NSMutableArray<NSMutableDictionary *> *incompleteData = [NSMutableArray new];
+    [[self sjQueryWithSQLStr:fieldsSqlM] enumerateObjectsUsingBlock:^(NSDictionary * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
+        [incompleteData addObject:obj.mutableCopy];
+    }];
+    
+    return [self _sjConversionMolding:cls rawStorageData:incompleteData];
+}
+
+
 - (NSArray<id> *)_sjConversionMolding:(Class)cls rawStorageData:(NSArray<NSDictionary *> *)rawStorageData {
     NSMutableArray<id> *allDataModel = [NSMutableArray new];
     NSArray<SJDBMapCorrespondingKeyModel *>*cKr = [self sjGetCorrespondingKeys:cls];

SJDBMap/SJDatabaseMap.h

@@ -103,6 +103,7 @@ NS_ASSUME_NONNULL_BEGIN
 
 // MARK: Query
 
+
 @interface SJDatabaseMap (Query)
 
 /*!
@@ -122,6 +123,37 @@ NS_ASSUME_NONNULL_BEGIN
  */
 - (void)queryDataWithClass:(Class)cls queryDict:(NSDictionary *)dict completeCallBlock:(void (^ __nullable)(NSArray<id<SJDBMapUseProtocol>> * _Nullable data))block;
 
+
+/*!
+ *  模糊查询
+ *  default SJDatabaseMapFuzzyMatchAll
+ */
+- (void)fuzzyQueryDataWithClass:(Class)cls queryDict:(NSDictionary *)dict completeCallBlock:(void (^ __nullable)(NSArray<id<SJDBMapUseProtocol>> * _Nullable data))block;
+
+
+typedef NS_ENUM(NSUInteger, SJDatabaseMapFuzzyMatch) {
+    /*!
+     *  匹配左右两边
+     *  ...A...
+     */
+    SJDatabaseMapFuzzyMatchAll = 0,
+    /*!
+     *  匹配以什么开头
+     *  ABC.....
+     */
+    SJDatabaseMapFuzzyMatchFront,
+    /*!
+     *  匹配以什么结尾
+     *  ...DEF
+     */
+    SJDatabaseMapFuzzyMatchLater,
+};
+
+/*!
+ *  模糊查询
+ */
+- (void)fuzzyQueryDataWithClass:(Class)cls queryDict:(NSDictionary *)dict match:(SJDatabaseMapFuzzyMatch)match completeCallBlock:(void (^ __nullable)(NSArray<id<SJDBMapUseProtocol>> * _Nullable data))block;
+
 @end
 
 

SJDBMap/SJDatabaseMap.m

@@ -307,6 +307,25 @@ - (void)queryDataWithClass:(Class)cls queryDict:(NSDictionary *)dict completeCal
     }];
 }
 
+/*!
+ *  模糊查询
+ */
+- (void)fuzzyQueryDataWithClass:(Class)cls queryDict:(NSDictionary *)dict completeCallBlock:(void (^ __nullable)(NSArray<id<SJDBMapUseProtocol>> * _Nullable data))block {
+    [self fuzzyQueryDataWithClass:cls queryDict:dict match:SJDatabaseMapFuzzyMatchAll completeCallBlock:block];
+}
+
+/*!
+ *  模糊查询
+ */
+- (void)fuzzyQueryDataWithClass:(Class)cls queryDict:(NSDictionary *)dict match:(SJDatabaseMapFuzzyMatch)match completeCallBlock:(void (^ __nullable)(NSArray<id<SJDBMapUseProtocol>> * _Nullable data))block {
+    [self.operationQueue addOperationWithBlock:^{
+        NSArray *models = [self sjFuzzyQueryConversionMolding:cls match:match dict:dict];
+        dispatch_async(dispatch_get_main_queue(), ^{
+            if ( block ) block(models);
+        });
+    }];
+}
+
 @end
 
 

SJDBMapProject/Model/Person.h

@@ -31,4 +31,6 @@
 
 @property (nonatomic, strong) NSArray<Goods *> *goods;
 
+@property (nonatomic, assign) NSInteger age;
+
 @end

SJDBMapProject/ViewController.m

@@ -50,8 +50,11 @@ - (void)viewDidLoad {
 
 
     [self insertOrUpdate];
-//    [self query];
-
+    
+    [[SJDatabaseMap sharedServer] fuzzyQueryDataWithClass:[Person class] queryDict:@{@"name":@"j"} match:SJDatabaseMapFuzzyMatchLater completeCallBlock:^(NSArray<id<SJDBMapUseProtocol>> * _Nullable data) {
+        NSLog(@"count = %zd", data.count);
+        NSLog(@"%@", data);
+    }];
     // Do any additional setup after loading the view, typically from a nib.
 }
 
@@ -74,6 +77,7 @@ - (void)insertOrUpdate {
                 [PersonTag tagWithID:4 des:@"E"],];
 
     sj.aBook = [Book bookWithID:123 name:@"How Are You?"];
+    sj.age = 20;
 
     Goods *g = [Goods new];
     g.name = @"G1";
@@ -82,7 +86,21 @@ - (void)insertOrUpdate {
 
     sj.goods = @[g, g2];
 
-    [[SJDatabaseMap sharedServer] insertOrUpdateDataWithModels:@[sj] callBlock:^(BOOL r) {
+    
+    Person *sj1 = [Person new];
+    sj1.personID = 1;
+    sj1.name = @"'sj'";
+    sj1.tags = @[[PersonTag tagWithID:0 des:@"A"],
+                 [PersonTag tagWithID:1 des:@"B"],
+                 [PersonTag tagWithID:2 des:@"C"],
+                 [PersonTag tagWithID:3 des:@"'D'"],
+                 [PersonTag tagWithID:4 des:@"E"],];
+    
+    sj1.aBook = [Book bookWithID:123 name:@"How Are You?"];
+    sj1.age = 20;
+    sj1.goods = @[g, g2];
+    
+    [[SJDatabaseMap sharedServer] insertOrUpdateDataWithModels:@[sj, sj1] callBlock:^(BOOL r) {
 
     }];
 }