多线程使用FMDB操作sqlite数据库

痛定思痛。 2022-05-30 08:22 262阅读 0赞

#### 1.创建model类 ####

#import <Foundation/Foundation.h>
    
    @interface LoginModel : NSObject
    @property NSString *userName;
    @property NSString *pwd;
    @end

#import "LoginModel.h"
    
    @implementation LoginModel
    
    @end

#### 2.创建工具类 ####

//
    //  LoginUtilTool.h
    //  iOSDataDemo
    //
    //  Created by liuyinghui on 2018/3/8.
    //  Copyright © 2018年 liuyinghui. All rights reserved.
    //
    
    #import <Foundation/Foundation.h>
    #import "LoginModel.h"
    @interface LoginUtilTool : NSObject
    +(instancetype)shareLoginUtilTool;
    -(BOOL)saveUserLoginInfo:(LoginModel*) loginModel;
    -(NSArray *)getAllLoginModel;
    @end

//
    //  LoginUtilTool.m
    //  iOSDataDemo
    //
    //  Created by liuyinghui on 2018/3/8.
    //  Copyright © 2018年 liuyinghui. All rights reserved.
    //
    
    #import "LoginUtilTool.h"
    #import "FMDB.h"
    
    @interface LoginUtilTool()<NSCopying,NSMutableCopying>
    
    @end
    
    static FMDatabaseQueue *databaseQueue=nil;
    static LoginUtilTool *loginUtil=nil;
    @implementation LoginUtilTool
    +(instancetype)shareLoginUtilTool{
    
        @synchronized (self)
        {
            if (loginUtil==nil) {
                loginUtil = [[self alloc] init];
            }
        }
        return loginUtil;
    }
    
    
    - (id)copyWithZone:(nullable NSZone *)zone{
        return self;
    }
    
    
    - (id)mutableCopyWithZone:(nullable NSZone *)zone{
        return self;
    }
    
    + (id) allocWithZone:(NSZone *)zone
    {
        @synchronized (self) {
            
            
            if (loginUtil == nil) {
                loginUtil = [super allocWithZone:zone];
                [self getFMDatabaseQueue];
                
                
                return loginUtil;
            }
            else return loginUtil;
        }
        return nil;
    }
    
    
    +(FMDatabaseQueue *)getFMDatabaseQueue{
        
        if(databaseQueue==nil){
            NSString *docPath = [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0];
            NSString *dbPath = [docPath stringByAppendingPathComponent:@"loginInfo.db"];
            databaseQueue = [FMDatabaseQueue databaseQueueWithPath:dbPath];
            [databaseQueue inDatabase:^(FMDatabase *database){
                
                if (database!=nil) {
                    BOOL result = NO;
                    //查询数据库中表名为loginStatus的个数
                    FMResultSet *rs = [database executeQuery:@"select count(*) as 'count' from sqlite_master where type = 'table' and name = 'loginStatus'"];
                    while ([rs next]) {
                        NSInteger count = [rs intForColumn:@"count"];
                        if (0 ==  count) {
                            result=  [database executeUpdate:@"create table loginStatus(userIndex integer PRIMARY KEY autoincrement,userName text,pwd text)"];
                        }else{
                            //检查表中的字段
                            NSArray *array = [[NSArray alloc]initWithObjects:@"userName", @"pwd",nil ];
                            NSString *tableName = @"loginStatus";
                            for (int i = 0; i < array.count; i++) {
                                NSString *columnName = [array objectAtIndex:i];
                                if (![database columnExists:columnName inTableWithName:tableName]) {
                                    result = [database executeUpdate:[NSString stringWithFormat:@"alter table %@ add column %@ text",tableName,columnName]];
                                }
                            }
                        }
                    }
                }
            }];
        }
        return databaseQueue;
    }
    
    -(BOOL)saveUserLoginInfo:(LoginModel*) loginModel{
        __block BOOL result=NO;
        [databaseQueue inDatabase:^(FMDatabase *database){
            
            if (database!=nil) {
                FMResultSet *resultSet = [database executeQuery:@"select * from loginStatus where userName = ?",loginModel.userName];
                BOOL haveRes=NO;
                while ([resultSet next]) {
                    haveRes=YES;
                    break;
                }
                if (haveRes) {
                    result= [database executeUpdateWithFormat:@"update loginStatus set userName = %@,pwd = %@",loginModel.userName,loginModel.pwd];
                }
                else{
                    
                    result= [database executeUpdateWithFormat:@"insert into loginStatus  (userName,pwd)  values (%@,%@)",loginModel.userName,loginModel.pwd];
                }
            }
        }];
        return result;
    }
    -(NSArray *)getAllLoginModel{
        
        __block NSArray *result=nil;
        [databaseQueue inDatabase:^(FMDatabase *db) {
            
            if (db!=nil){
                NSMutableArray *resultTmpArray=[[NSMutableArray alloc] init];
                FMResultSet *resultSet = [db executeQuery:@"select * from loginStatus"];
                while ([resultSet next]) {
                    LoginModel *loginmodel=[[LoginModel alloc] init];
                    loginmodel.userName=[resultSet stringForColumn:@"userName"];
                    loginmodel.pwd=[resultSet stringForColumn:@"pwd"];
                    [resultTmpArray addObject:loginmodel];
                }
                result=[resultTmpArray copy];
            }
        }];
        return result;
    }
    
    
    @end

#### 3 调用 ####

-(void)useFMDBSqlite{
        dispatch_queue_t dispatchQueue=dispatch_queue_create("com.langnuo.www", DISPATCH_QUEUE_CONCURRENT);
        for (int i=0; i<100; i++) {
            dispatch_async(dispatchQueue, ^{
                LoginModel *model=[[LoginModel alloc] init];
                model.userName=[NSString stringWithFormat:@"userName%d",i];
                model.pwd=@"123456";
                LoginUtilTool *tool=[LoginUtilTool shareLoginUtilTool];
                [tool saveUserLoginInfo:model];
                NSLog(@"当前存入的账号是%@",model.userName);
                NSArray * array=[tool getAllLoginModel];
                for (LoginModel *model in array) {
                    NSLog(@"第%d个线程创建的  %@ ",i,model.userName);
                }
            });
        }
        
        
    }

[对应源码][Link 1]

[Link 1]: https://download.csdn.net/download/liuyinghui523/10275603