第一步，調用阿里雲Api發送短信並存入數據庫或者是redis當中
-
使用json配置文件
-
// 配置文件 "sms": { "sign_name": "虛擬雲餐廳", "template_code": "SMS_205471746", "app_key": "", "app_secret": "", "region_id": "cn-hangzhou" } - 配置文件解析
func ParseConfig(path string) (*Config, error) { file, err := os.Open(path) if err != nil { panic(err) } defer file.Close() newReader := bufio.NewReader(file) decoder := json.NewDecoder(newReader) err = decoder.Decode(&config) if err != nil { panic(err) } return config, nil } -
-
編寫路由controller層
-
//發送手機驗證碼 engine.GET("/api/sendcode",mc.SendSmsCode ) //手機號和短信登錄 engine.POST("/api/login_sms", mc.smsLogin) //刪除驗證碼 -
發送短信
func (mc *MemBerController) SendSmsCode(context *gin.Context) {
phone, err := context.GetQuery("phone")
if !err {
context.JSON(200, map[string]interface{}{
"code": 0,
"message": "手機號不存在",
})
return
}
ms:=Service.MemberService{}
isSend := ms.Sendcode(phone)//調用service層方法發送驗證碼
if isSend {
context.JSON(200, map[string]interface{}{
"code":1,
"msg":"發送成功",
})
}else {
context.JSON(200, map[string]interface{}{
"code": 0,
"msg": "發送失敗",
})
}
}
- service層
func (msi * MemberService)Sendcode(phone string) bool {
//隨機產生一個驗證碼
code := fmt.Sprintf("%06v", rand.New(rand.NewSource(time.Now().UnixNano())).Int31n(1000000))
smsConfig := config.GetConfig().Sms
//調用阿里雲短信接口發送短信
//client, err := dysmsapi.NewClientWithAccessKey("cn-hangzhou", "<accessKeyId>", "<accessSecret>")
client, err := dysmsapi.NewClientWithAccessKey(smsConfig.RegionId, smsConfig.AppKey, smsConfig.AppSecret)
if err != nil {
panic(err)
}
request := dysmsapi.CreateSendSmsRequest()
//request屬性設置
request.Scheme = "https"
request.SignName = smsConfig.SignName
request.TemplateCode = smsConfig.TemplateCode
request.PhoneNumbers = phone
//使用json字符串發送驗證碼
par, err := json.Marshal(map[string]interface{}{
"code": code,
})
//設置驗證碼
request.TemplateParam = string(par)
response, err := client.SendSms(request)
fmt.Println( response)
if err != nil {
err.Error()
return false
}
//檢查返回結果，並判斷發生狀態
if response.Code != "OK" {
smsCode := model.SmsCode{Phone: phone, Code: code, BizId: response.BizId, CreateTime: time.Now().Unix()}
//插入數據庫操作
memberDao := dao.MemberDao{util.DdEngie}
insertCode := memberDao.InsertCode(smsCode)
return insertCode>0
return true
}
return true
}
- Dao層
- 數據庫表結構
//驗證碼表結構 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"` } //用戶表結構 //用戶類 type Member struct { Id int64 `xorm:"pk autoincr "json:"id"` UserName string `xorm:"varchar(20) "json:"user_name"` Mobile string `xorm:"varchar(11)" json:"mobile"` Password string `xorm:"varchar(255)" json:"password"` RegisterTime int64 `xorm:"bigint" json:"register_time"` Avatar string `xorm:"varchar(255)" json:"avatar"` Balance float64 `xorm:"double" json:"balance"` IsActive int8 `xorm:"tinyint" json:"is_active"` City string `xorm:"varchar(10)" json:"city"` }
type MemberDao struct {
* util.Orm
}
//插入數據
func (md *MemberDao) InsertCode(sms model.SmsCode) int64 {
one, err := md.InsertOne(&sms)
if err != nil {
panic(err)
}
return one
}
//插入用戶
func (md *MemberDao) InsertMember(member model.Member) int64 {
result, err := md.InsertOne(&member)
if err != nil {
util.Error(err.Error())
}
return result
}
驗證碼登錄
- 路由
engine.POST("/api/login_sms", mc.smsLogin)
//短信登錄
func (mc *MemBerController) smsLogin(context *gin.Context) {
var smsParam param.SmsLoginParam
//解析請求參數
// type SmsLoginParam struct {
// Phone string `json:"phone"`
// Code string `json:"code"`
// }
err := util.Decode(context.Request.Body, &smsParam)
if err != nil {
util.Failed(context, "參數解析錯誤")
return
}
//完成手機加短信的登錄方法
us := Service.MemberService{}
member := us.SmsLogin(smsParam)
if member != nil {
util.Success(context, member)
return
}
util.Failed(context, "登錄失敗")
}
- service層
func (msi *MemberService) SmsLogin(param param.SmsLoginParam) *model.Member {
dao := dao.MemberDao{}
//查詢封裝的驗證碼表結構
sms := dao.ValidateSmsCode(param.Phone, param.Code)
//判斷時間
if sms == nil || time.Now().Unix()-sms.CreateTime > 300 {
return nil
}
//查詢用戶
member := dao.QueryByPhone(param.Phone)
if member != nil {
return nil
}
user := model.Member{}
user.UserName = param.Phone
user.Mobile = param.Phone
user.RegisterTime = time.Now().Unix()
user.Id = dao.InsertMember(user)
//返回用戶
return &user
}
- dao層
//查詢語句
func (md *MemberDao) ValidateSmsCode(phone string, code string) *model.SmsCode {
var sms model.SmsCode
if err := md.Where(" phone = ? and code = ? ", phone, code).Find(&sms); err != nil {
util.Error(err.Error())
}
return &sms
}
func (md *MemberDao) QueryByPhone(phone string) interface{} {
var sms model.Member
err := md.Where("phone=?", phone).Find(&sms)
if err != nil {
util.Error(err.Error())
}
return sms
}