1. 程式人生 > 實用技巧 >go使用Gin框架利用阿里雲實現簡訊驗證碼

go使用Gin框架利用阿里雲實現簡訊驗證碼

第一步,呼叫阿里雲Api傳送簡訊並存入資料庫或者是redis當中

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
}