iOS学习4.存储聊天记录

1 #import <Foundation/Foundation.h>
2 
3 @interface Text : NSObject
4 //聊天内容
5 @property(nonatomic,copy)NSString *userText;
6 //聊天内容发送的时间
7 @property(nonatomic,copy)NSString *currentTime;
8 @end

 1 #import <Foundation/Foundation.h>
 2 #import <sqlite3.h>
 3 @class Text;
 4 @interface TextModel : NSObject
 5 //建表
 6 -(BOOL)createList:(sqlite3 *)db;
 7 //插入
 8 -(BOOL)insertList:(Text *)insertList;
 9 //获取数据
10 -(NSMutableArray *)getList;
11 @end

//
//  TextModel.m
//  保存聊天记录
//
//  Created by 736376103@qq.com on 16/4/4.
//  Copyright © 2016年 736376103@qq.com. All rights reserved.
//

#import "TextModel.h"
#import "Text.h"
@implementation TextModel
//定义一个变量
static sqlite3 *_database;

-(NSString *)filename{
    NSString *path = [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES)lastObject];
    //这里打印，是方便用MesaSQLite打开sqlite3文件,可以增删改查，非常方便,主要是免费~~~
    NSLog(@"%@",path);
    return [path stringByAppendingPathComponent:@"LIKE.sqlite"];
}

-(BOOL)openDB{
    //获取路径
    NSString *path = [self filename];
    NSFileManager *fileManager = [NSFileManager defaultManager];
    //判断数据库是否存在
    BOOL find = [fileManager fileExistsAtPath:path];
    //如果为真，就打开数据库，不存在，自动创建
    if (find) {
        NSLog(@"database存在");
        
        if (sqlite3_open([path UTF8String], &_database)!=SQLITE_OK) {
            //failed关闭，据说这个习惯好，不明觉厉
            sqlite3_close(_database);
            NSLog(@"打开database失败");
            return NO;
        }
        //建表
        [self createList:_database];
        return YES;
    }
    //同上
    if (sqlite3_open(path.UTF8String, &_database) == SQLITE_OK) {
        [self createList:_database];
        return YES;
    }else{
        sqlite3_close(_database);
        NSLog(@"打开database失败");
        return NO;
    }
        return NO;
    
}
#pragma mark -- 建表
-(BOOL)createList:(sqlite3 *)db{
    //我这里缺少一个主键，自己加上即可--ID INTEGER PRIMARY KEY AUTOINCREMENT
    NSString *sql = [NSString stringWithFormat:@"CREATE TABLE IF NOT EXISTS DRINK(userText TEXT,currentTime TEXT)"];
    sqlite3_stmt *stmt;
    //sqlite3_prepare_v2 接口把一条SQL语句解析到statement结构里去. 使用该接口访问数据库是当前比较好的的一种方法
    NSInteger sqlReturn = sqlite3_prepare_v2(_database, sql.UTF8String, -1, &stmt, NULL);
    //-1是sql语句的长度，<0会自动计算
    if (sqlReturn != SQLITE_OK) {
        NSLog(@"创建表失败");
        return NO;
    }
    int success = sqlite3_step(stmt);
    //释放stmt
    sqlite3_finalize(stmt);
    if (success != SQLITE_DONE) {
        NSLog(@"创建表失败");
        return NO;
    }
    NSLog(@"创建表成功");
    return YES;
}
#pragma mark -- 插入
-(BOOL)insertList:(Text *)insertList{

    if ([self openDB])
    {
        sqlite3_stmt *stmt;
        //?表示待会儿插入
        NSString *sql = [NSString stringWithFormat:@"INSERT INTO DRINK(userText,currentTime)VALUES(?,?)"];
        //int success = sqlite3_exec(_database, sql.UTF8String, NULL, NULL, &error);
        int success = sqlite3_prepare_v2(_database, sql.UTF8String, -1, &stmt, NULL);
        if (success != SQLITE_OK) {
            NSLog(@"insert failed");
            sqlite3_close(_database);
            return NO;
        }
        sqlite3_bind_text(stmt, 1, [insertList.userText UTF8String], -1, SQLITE_TRANSIENT);
        sqlite3_bind_text(stmt, 2, [insertList.currentTime UTF8String], -1, SQLITE_TRANSIENT);
        //执行插入语句
        success = sqlite3_step(stmt);
        //释放stmt
        sqlite3_finalize(stmt);
        NSLog(@"%@",insertList.userText);
        NSLog(@"%@",insertList.currentTime);
        //如果failed
        if (success == SQLITE_ERROR) {
            NSLog(@"failed insert into database");
            sqlite3_close(_database);
            return NO;
        }
        sqlite3_close(_database);
        return YES;
    }
    return NO;
}
#pragma mark -- 获取
-(NSMutableArray *)getList{
    NSMutableArray *array = nil;
    //判断是否打开,这里可以用 dispatch_once 只执行一次
    if ([self openDB]) {
        sqlite3_stmt *stmt;
        NSString *sql = [NSString stringWithFormat:@"SELECT userText,currentTime FROM DRINK"];
        if (sqlite3_prepare_v2(_database, sql.UTF8String, -1, &stmt, NULL) != SQLITE_OK) {
            NSLog(@"failed to get list");
        }else{
            array = [NSMutableArray array];
            //遍历记录，这里是从0开始，别写错了
            while (sqlite3_step(stmt) == SQLITE_ROW) {
                Text *p = [[Text alloc]init];
                char *strText = (char *)sqlite3_column_text(stmt, 0);
                //做个判断，如果记录为nil，不执行，这里自己要想一下，为什么要判断
                if (strText != NULL) {
                    p.userText = [NSString stringWithUTF8String:strText];
                }
                char *strTime = (char *)sqlite3_column_text(stmt, 1);
                //时间为Null，不执行
                if (strTime != NULL) {
                    p.currentTime = [NSString stringWithUTF8String:strTime];
                }
                //加进可变数组
                [array addObject:p];
            }
        }
        //释放stmt
        sqlite3_finalize(stmt);
        sqlite3_close(_database);
    }
    return array;
}
@end

#import <UIKit/UIKit.h>
@interface ViewController : UIViewController
@property (weak, nonatomic) IBOutlet UITableView *tableview;

@property (weak, nonatomic) IBOutlet UITextField *textField;

@end

//
//  ViewController.m
//  保存聊天记录
//
//  Created by 736376103@qq.com on 16/4/4.
//  Copyright © 2016年 736376103@qq.com. All rights reserved.
//

#import "ViewController.h"
#import "TableViewCell.h"
#import "Text.h"
#import "TextModel.h"
//实现UITablevView和UITextField的代理
@interface ViewController ()<UITableViewDataSource,UITableViewDelegate,UITextFieldDelegate>
//数据源
@property(nonatomic,strong)NSMutableArray *dataSource;
//处理数据的类
@property(nonatomic,strong)TextModel *textModel;

@end

@implementation ViewController
//加载数据
-(void)reloadDataSource{
    if (_textModel == nil) {
        _textModel = [[TextModel alloc]init];
    }
    //获取数据
    _dataSource = [_textModel getList];
}

- (void)viewDidLoad {
    [super viewDidLoad];
    //设置代理
    _textField.delegate = self;
    //调用加载数据方法
    [self reloadDataSource];

}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
}

-(void)sendMessageContent:(NSString *)text{
    //获取时间
    NSDate *date = [NSDate date];
    NSDateFormatter *dateForMatter = [[NSDateFormatter alloc]init];
    //hh：mm是09：54这样的格式，可以自由发挥
    dateForMatter.dateFormat = @"hh:mm";
    NSString *timeString = [dateForMatter stringFromDate:date];
    
    Text *t = [[Text alloc]init];
    //把时间存到Text的属性
    t.currentTime = timeString;
    //输入的内容
    t.userText = text;
    //插入到数据库
    [_textModel insertList:t];
}
#pragma mark -- UITextField代理
-(BOOL)textFieldShouldReturn:(UITextField *)textField{
    //自定义方法，把输入的内容存到Text
    [self sendMessageContent:textField.text];
    //取消第一响应
    [textField resignFirstResponder];
    //清空键盘
    textField.text = @"";
    //插入之后，加载一次数据，相当于往数据源里加数据
    [self reloadDataSource];
    //刷新界面，显示刚插入的数据
    [_tableview reloadData];
    return YES;
}

#pragma mark -- TableView数据源
-(NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{
    return _dataSource.count;
}

-(UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath{
    
    Text *p = _dataSource[indexPath.row];
    //自定义Cell
    TableViewCell *cell = [TableViewCell tableView:tableView];
    //cell不可点击
    cell.selectionStyle = UITableViewCellSelectionStyleNone;
    cell.timeLabel.text = p.currentTime;
    cell.OtherLabel.text = p.userText;
    return cell;
}
#pragma mark -- 代理方法
-(CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath{
    return 80;
}
@end

#import <UIKit/UIKit.h>

@interface TableViewCell : UITableViewCell
//聊天内容
@property (strong, nonatomic)UILabel *OtherLabel;
//时间
@property (strong, nonatomic)UILabel *timeLabel;
//头像，这个自己随便找个图
@property(strong,nonatomic)UIImageView *meImageView;

+(instancetype)tableView:(UITableView *)tableView;

@end

//
//  TableViewCell.m
//  保存聊天记录
//
//  Created by 736376103@qq.com on 16/4/4.
//  Copyright © 2016年 736376103@qq.com. All rights reserved.
//

#import "TableViewCell.h"

@implementation TableViewCell
//初始化的时候，创建控件
-(id)initWithStyle:(UITableViewCellStyle)style reuseIdentifier:(NSString *)reuseIdentifier{
    if (self = [super initWithStyle:style reuseIdentifier:reuseIdentifier]) {
        
        CGFloat width = [UIScreen mainScreen].bounds.size.width;
        
        _OtherLabel = [[UILabel alloc]initWithFrame:CGRectMake(0, 40, width-40, 30)];
        _OtherLabel.font = [UIFont systemFontOfSize:16];
        _OtherLabel.numberOfLines = 0;
        _OtherLabel.lineBreakMode = NSLineBreakByTruncatingTail;
        _OtherLabel.textAlignment = NSTextAlignmentRight;
        [self.contentView addSubview:_OtherLabel];
        
        _timeLabel = [[UILabel alloc]initWithFrame:CGRectMake(width/2-20, 0, 40, 20)];
        _timeLabel.font = [UIFont systemFontOfSize:12];
        [self.contentView addSubview:_timeLabel];
        
        _meImageView = [[UIImageView alloc]initWithFrame:CGRectMake(width-40, 10, 30, 30)];
        _meImageView.image = [UIImage imageNamed:@"003"];
        [self.contentView addSubview:_meImageView];
    }
    return self;
}


+(instancetype)tableView:(UITableView *)tableView{
    TableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:@"ChatTableViewCell"];
    //cell复用
    if (cell == nil) {
        cell = [[TableViewCell alloc]initWithStyle:UITableViewCellStyleDefault reuseIdentifier:@"ChatTableViewCell"];
    }
    return cell;
}


@end