背景介绍
上节课我们已经接入了第三方的短信服务SDK功能，可以接受短信验证码。在用户接受到验证码以后，输入验证码进行登录，我们需要验证用户输入的验证码是否正确。
因此，我们需要将发送过的验证码通过持久化的方式保存下来，方便我们进行校验。
我们选择通过数据库来存储用户手机验证码。
xorm介绍
在项目开发过程中，我们会使用一些成熟的框架来操作数据库。xorm就是一个比较流行的数据库操作orm框架。
xorm安装及mysql驱动
使用如下命令安装xorm：
go get github.com/go-xorm/xorm
安装mysql驱动：
go get github.com/go-sql-driver/mysql
连接数据库
在连接数据库之前，首先要创建数据库。在mysql中创建cloudrestaurant数据库：
create database cloudrestaurant;
创建完数据库并安装好xorm库以后，使用xorm进行连接数据库。具体的连接操作如下所示：
import (
"github.com/go-xorm/xorm"
_ "github.com/go-sql-driver/mysql"
)
database := cfg.Database
conn := database.User + ":" + database.Password + "@tcp(" + database.Host + ":" + database.Port + ")/" + database.DbName + "?charset=" + database.Charset
engine, err := xorm.NewEngine(database.Driver, conn)
if err != nil {
return nil, err
}
...
连接数据库有些参数需要自己指定，比如说驱动类型，登录数据库的用户名，密码，数据库名等。将这些变量配置在app.json配置文件中，如下所示：
{
...
"database": {
"driver": "mysql",
"user": "root",
"password": "12345678",
"host": "127.0.0.1",
"port": "3306",
"db_name": "cloudrestaurant",
"charset": "utf8mb4",
"show_sql": true
}
...
}
并在Config结构体中添加对dtabase的解析：
type Config struct {
AppName string json:"app_name"
AppMode string json:"app_mode"
AppHost string json:"app_host"
AppPort string json:"app_port"
Database DatabaseConfig json:"database"
Sms SmsConfig json:"sms"
}
type DatabaseConfig struct {
Driver string json:"driver"
User string json:"user"
Password string json:"password"
Host string json:"host"
Port string json:"port"
DbName string json:"db_name"
Charset string json:"charset"
ShowSql bool json:"show_sql"
}
...
创建SmsCode
要存储验证码，需要在数据库中创建表结构进行存储。我们可以创建SmsCode结构体，并通过tag设置数据库字段约束，具体的SmsCode定义如下:
package model
type SmsCode struct {
Id int64 xorm:"pk autoincr" json:"id"
Phone string xorm:"varchar(11)" json:"phone"
BizId string xorm:"varchar(30)" json:"biz_id"
Code string xorm:"varchar(4)" json:"code"
CreateTime int64 xorm:"bigint" json:"create_time"
}
通过tag的xorm设置字段数据类型以及约束。
pk：表示主键
autoinc：表示自增
bigint：整形变量
varchar：字符串类型
Sync2同步生成数据库表
可以调用engine.Sync2方法，将结构体类型同步映射到数据库中，生成数据库表。
err = engine.Sync2(new(model.SmsCode))
if err != nil {
return nil, err
}
...
将验证码数据保存到数据库中
在MemberService的SendCode方法，添加保存验证码到数据库的操作：
func (msi *MemberServiceImpl) SendCode(phone string) string {
code := fmt.Sprintf("%06v", rand.New(rand.NewSource(time.Now().UnixNano())).Int31n(1000000))
...
dao := impl.NewMemberDao()
smsCode := model.SmsCode{Phone: phone, Code: code, BizId: response.BizId, CreateTime: time.Now().Unix()}
if result := dao.InsertCode(smsCode); result > 0 {
return code
}
return ""
}
