工作二总结——objective-C中sqlite3数据库的处理（其二）

最新推荐文章于 2024-09-20 15:37:36 发布

lcpk0411

最新推荐文章于 2024-09-20 15:37:36 发布

阅读量629

点赞数

分类专栏：工作总结—IOS和objC 文章标签： sqlite3 objective-c

本文链接：https://blog.csdn.net/lcpk0411/article/details/40681531

版权

工作总结—IOS和objC 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

这篇是其一中工作的最终实现，使用了FMDB，更具有objective-C的风格。

    FMDatabase *db = [[FMDatabase alloc] initWithPath:[NSString stringWithFormat:@"%s", databasePath]];
    if ([db open] == NO) {
        NSLog(@"Failed to open database!");
        return 1;
    }

首先打开路径下的数据库，顺便有个小技巧，如何把C的字符串转换为NSString？我就是用stringWithFormat:@"%s"的方式。

[db open]进行的操作是打开数据库，并返回BOOL值表示是否打开成功。如果没有数据库，会自行创建。

依然如同（其一），先删去数据库相应的表再添加新表：

    // 为了防止表格式有问题，因此先删去同名表，再创建相应表
    [db executeUpdateWithFormat:@"drop table entity;"];
    BOOL ret = [db executeUpdateWithFormat:@"create table entity(id integer not null, name TEXT not null unique, score integer, popularity integer);"];
    if (ret != YES) {
        NSLog(@"Creating table entity failed!");
        exit(1);
    }

数据库对象的实例方法executeUpdate和executeUpdateWithFormat用于更改数据库内容。DROP、CREATE、INSERT、DELETE都在其列。

两者的区别是对格式占位符不同，WithFormat会用%@、%s之类的占位符，而前者全部占位符都是?。

本文还是那三个表，entity表和tag表的name字段必须唯一。entity非常好说，一行就一个，不会有重复的情况，而tag呢？

上一篇中完全没有进行判断，直接依靠数据库的unique字段来搞定。

然而实际工作中出现的情况是，往往有上万个entity，而tag只有十几个。为了这十几个不断重复的tag我们要进行entity数量*4次数据库插入，要知道数据库的数据交换比内存中直接交换要慢的多，这十万次数据库处理的时间开销实在不忍直视。

因此我用内存中处理的方式判断tag是否重复，也就是一个NSMutableDictionary。

NSMutableDictionary *allTags = [NSMutableDictionary dictionary];/

key当然是tag名称，而value则是tag的id，如此，如果出现

allTags[tagObject] == nil;

的情况，就可以判断没有这个tag，可以加入到数据库中。

如果不是这样，那么肯定可以得到tag的id，（其一）使用select语句来查找tag的id，而在这里，NSMutableDictionary就能解决这个问题，又减少了一次数据库查询开销，真是一举两得！

dictionary的实质是个hash table，其查询效率还是很高的。

插入数据的话，使用

        NSString *SQL = [NSString stringWithFormat:@"insert into entity(id,name,score,popularity) values(%d,'%s',%s, %d);", entityCount, entityName, entityScore, 0];
        ret = [db executeUpdate:SQL];

我不知道为什么，直接用FMDB的Format老出现问题，因此我直接调整好格式输入就行。

直接贴全文好了，我注释做的很认真，应该可以看懂。

本文没有用到select语句，其实FMDBselect语句极其简单，它会将查询结果储存在一个resultSet里面，之后根据resultSet的方法获取需要的数据即可。

我会在（其三）中讲解select语句用法。

下篇文章就不是这个电脑上的工作了，下个工作是把生成的数据库的这三个表加载到iPhone上，制成iphone上的4个NSDictionary。

第一个是tag对应其映射的entity列表

第二个是entity对应其映射的tag列表

第三个是tag对应其分数

第四个是entity对应其分数

根据最初乱码文本得到的这4个NSDictionary在之后的工作中会有很大作用。

#import <Foundation/Foundation.h>
#import "FMDB.h"

const char *databasePath;
const char *txtPath;

int entityCount, tagCount, mapCount;

int main(int argc, const char * argv[]) {
    if (argc != 3) {
        NSLog(@"Wrong arg number! Please type in txt path at first, and then the db path!");
        return 1;
    }
    txtPath = argv[1];
    FILE *txtFile = fopen(txtPath, "r");
    if (!txtFile) {
        NSLog(@"File loading is failed!");
        return 1;
    }
    databasePath = argv[2];
    FMDatabase *db = [[FMDatabase alloc] initWithPath:[NSString stringWithFormat:@"%s", databasePath]];
    if ([db open] == NO) {
        NSLog(@"Failed to open database!");
        return 1;
    }
    
    // 为了防止表格式有问题，因此先删去同名表，再创建相应表
    [db executeUpdateWithFormat:@"drop table entity;"];
    BOOL ret = [db executeUpdateWithFormat:@"create table entity(id integer not null, name TEXT not null unique, score integer, popularity integer);"];
    if (ret != YES) {
        NSLog(@"Creating table entity failed!");
        exit(1);
    }
    [db executeUpdateWithFormat:@"drop table tag;"];
    ret = [db executeUpdateWithFormat:@"create table tag(id integer not null, name TEXT not null unique, score integer, popularity integer);"];
    if (ret != YES) {
        NSLog(@"Creating table tag failed!");
        exit(1);
    }
    [db executeUpdateWithFormat:@"drop table map;"];
    ret = [db executeUpdateWithFormat:@"create table map(id integer not null, entity_id not null, tag_id integer);"];
    if (ret != YES) {
        NSLog(@"Creating table map failed!");
        exit(1);
    }
    
    // 用C的方式操作文本文件更加方便
    char stringFromTxt[500], last[500]; //开始读取txt中数据，stringFromTxt代表txt一行，last表示上次处理的一行
    fgets(stringFromTxt, 500, txtFile);
    memset(last, 0, sizeof(last));
    NSMutableDictionary *allTags = [NSMutableDictionary dictionary];//内存判断tag是否重复比让数据库判断unique值更快
    while (1) {
        // 每趟循环处理文本的一行，循环结束条件为“到达最后一行”，经测试，下面两种情况覆盖了结束条件
        if (strcmp(last, stringFromTxt) == 0 || stringFromTxt == NULL){
            break;
        }
        // 插入entity和score的数据
        char *entityName = strtok(stringFromTxt," \t");
        char *entityScore = strtok(NULL, " \t");
        NSString *SQL = [NSString stringWithFormat:@"insert into entity(id,name,score,popularity) values(%d,'%s',%s, %d);", entityCount, entityName, entityScore, 0];
        ret = [db executeUpdate:SQL];
        if (ret != YES) NSLog(@"插入entity失败,第%d次", entityCount);
        
        // 开始处理tag，txt文件中出现过一个entity对应多个相同tag的情况，在map表构造中应该极力避免这种情况，用mutableSet可以保证高效
        NSMutableSet *tagSet = [NSMutableSet set];
        char *tag = strtok(NULL, " \n\t");
        while (tag != NULL) {
            // entity对应了两个相同的tag，直接跳过，处理下一个tag
            NSString *tagObject = [NSString stringWithFormat:@"%s", tag];//会多次使用，因此先存下来
            if ([tagSet containsObject:tagObject] == NO) {
                [tagSet addObject:tagObject];
            } else {
                tag = strtok(NULL, " \t\n");
                continue;
            }
            // 如果是新出现的tag，就放入tag总集，查询开销为O(log(tag数))
            NSNumber *tagId = allTags[tagObject];
            if (tagId == nil) {
                tagId = [NSNumber numberWithInt:tagCount];
                [allTags setObject:tagId forKey:tagObject];
                
                // 随机得到tagScore，并插入tag表，只有新出现的tag才有必要加入tag表
                int tagScore = arc4random() % 101;
                SQL = [NSString stringWithFormat:@"insert into tag(id,name,score,popularity) values(%d, '%s', %d, %d);", tagCount ++, tag, tagScore, 0];
                ret = [db executeUpdate:SQL];
            }
            // 需要将所有tag映射关系添加至map表
            int intTagId = [tagId intValue];
            SQL = [NSString stringWithFormat:@"insert into map(id,entity_id,tag_id) values(%d, %d, %d);", mapCount++, entityCount, intTagId];
            [db executeUpdate: SQL];
            
            tag = strtok(NULL, " \t\n");
        }
        
        strcpy(last, stringFromTxt);
        fgets(stringFromTxt, 500, txtFile);
        entityCount ++;
    }
    [db close];
    return 0;
}