SQLite的基本使用-示例代码

SQLite的基本使用

//
//  IWViewController.m
//  01-SQLite的基本使用
//
//  Created by apple on 14-5-22.
//  Copyright (c) 2014年 itcast. All rights reserved.
//

#import "IWViewController.h"
#import <sqlite3.h>

@interface IWViewController ()
{
// 成员变量默认=NULL
sqlite3 *_db; // db代表着整个数据库，db是数据库实例
}
- (IBAction)insert;
- (IBAction)update;
- (IBAction)delete;
- (IBAction)query;
@end

@implementation IWViewController

- (void)viewDidLoad
{
[super viewDidLoad];

// 0.获得沙盒中的数据库文件名
NSString *filename = [[NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:@"student.sqlite"];

NSLog(@"filename=====%@",filename);

// 1.创建(打开)数据库（如果数据库文件不存在，会自动创建数据库main）
int result = sqlite3_open(filename.UTF8String, &_db);
if (result == SQLITE_OK) {
NSLog(@"成功打开数据库");

// 2.创表
const char *sql = "create table if not exists t_student (id integer primary key autoincrement, name text, age integer);";

// 为了严谨起见，凡是传递地址的，一般最初都要先清空。首先保证该变量是空指针，而不是野指针。
char *errorMesg = NULL;

// sqlite3_exec 执行SQL语句
// 参数解释：第一个数据库实例，第二个SQL语句，第三个sqlite3_exec语句执行成功后的回调，第四个回调要传递的东西，第五个错误信息
// C语言的空是大写的NULL
// sqlite3_exec(<#sqlite3 *#>, <#const char *sql#>, <#int (*callback)(void *, int, char **, char **)#>, <#void *#>, <#char **errmsg#>)
int result = sqlite3_exec(_db, sql, NULL, NULL, &errorMesg);
if (result == SQLITE_OK) {
NSLog(@"成功创建t_student表");
} else {
NSLog(@"创建t_student表失败:%s", errorMesg);// %s用来打印C语言的字符串
}
} else {
NSLog(@"打开数据库失败");
}
}

- (IBAction)insert
{
for (int i = 0; i<30; i++) {
NSString *name = [NSString stringWithFormat:@"Jack-%d", arc4random()%100];
int age = arc4random()%100;
NSString *sql = [NSString stringWithFormat:@"insert into t_student (name, age) values('%@', %d);", name, age];

char *errorMesg = NULL;
int result = sqlite3_exec(_db, sql.UTF8String, NULL, NULL, &errorMesg);
if (result == SQLITE_OK) {
NSLog(@"成功添加数据");
} else {
NSLog(@"添加数据失败:%s", errorMesg);
}
}
}

- (IBAction)query
{
// SQL注入漏洞

/**
登录功能

1.用户输入账号和密码
* 账号：123' or 1 = 1 or '' = '
* 密码：456654679

2.拿到用户输入的账号和密码去数据库查询（查询有没有这个用户名和密码）
select * from t_user where username = '123' and password = '456';

select * from t_user where username = '123' and password = '456';
*/

// 1.定义sql语句
const char *sql = "select id, name, age from t_student where name = ?;";

// 2.定义一个stmt来存放结果集
sqlite3_stmt *stmt = NULL;

// 3.检测SQL语句的合法性 第三个参数：SQL语句的长度，这里写-1会自动帮我们算SQL字符串的长度。最后一个参数：跟SQL语句尾部有关，此处先不写。
int result = sqlite3_prepare_v2(_db, sql, -1, &stmt, NULL);
if (result == SQLITE_OK) {
NSLog(@"查询语句是合法的");

/**
sqlite3_bind_text()：大部分绑定函数都只有3个参数
第1个参数是sqlite3_stmt *类型
第2个参数指占位符的位置，第一个占位符的位置是1，不是0，代表第几个问号
第3个参数指占位符要绑定的值
第4个参数指在第3个参数中所传递数据的长度，对于C字符串，传递-1会自动计算字符串的长度
第5个参数是一个可选的函数回调，一般用于在语句执行后完成内存清理工作
sqlite_step()：执行SQL语句，返回SQLITE_DONE代表成功执行完毕
sqlite_finalize()：销毁sqlite3_stmt *对象
*/
// 设置占位符的内容，用占位符的方式，会自动检测SQL注入情况，可有效防止SQL注入问题。
sqlite3_bind_text(stmt, 1, "jack", -1, NULL);

// 4.执行SQL语句，从结果集中取出数据
//        int stepResult = sqlite3_step(stmt);
while (sqlite3_step(stmt) == SQLITE_ROW) { // 查询到一行数据
// 获得这行对应的数据

// 获得第0列的id
int sid = sqlite3_column_int(stmt, 0);

// 获得第1列的name
const unsigned char *sname = sqlite3_column_text(stmt, 1);

// 获得第2列的age
int sage = sqlite3_column_int(stmt, 2);

NSLog(@"%d %s %d", sid, sname, sage);
}
} else {
NSLog(@"查询语句非合法");
}
}
@end