serverless＋腾讯云短信实现短信验证码登录

/**
 * 
 * @param {*} 功能：通过 SDK 发送短信 
 * @param {*} 参数：手机号、短信验证码 
 */
async function sendSms(phone, code) {
    const tencentcloud = require('tencentcloud-sdk-nodejs');
    const SmsClient = tencentcloud.sms.v20190711.Client;
    const Credential = tencentcloud.common.Credential;
    const ClientProfile = tencentcloud.common.ClientProfile;
    const HttpProfile = tencentcloud.common.HttpProfile;
    //腾讯云账户 secretId，secretKey，切勿泄露
    const secretId = "secretId";//需要配置为真实的 secretId
    const secretKey = "secretKey";//需要配置为真实的 secretKey
    let cred = new Credential(secretId, secretKey);
    let httpProfile = new HttpProfile();
    httpProfile.endpoint = "sms.tencentcloudapi.com";
    let clientProfile = new ClientProfile();
    clientProfile.httpProfile = httpProfile;
    let client = new SmsClient(cred, "ap-guangzhou", clientProfile);
    phone = "+86" + phone;//国内手机号
    let req = {
        PhoneNumberSet: [phone],//发送短信的手机号
        TemplateID: "",//<a href="#Step1_2">步骤1.2</a> 中创建并记录的模板 ID
        Sign: "",//<a href="#Step1_1">步骤1.1</a> 中创建的签名
        TemplateParamSet: [code],//随机验证码
        SmsSdkAppid: ""//短信应用 ID
    }
    function smsPromise() {
        return new Promise((resolve, reject) => {
            client.SendSms(req, function (errMsg, response) {
                if (errMsg) {
                    reject(errMsg)
                } else {
                    if (response.SendStatusSet && response.SendStatusSet[0] && response.SendStatusSet[0].Code === "Ok") {
                        resolve({
                            errorCode: 0,
                            errorMessage: response.SendStatusSet[0].Message,
                            data: {
                                codeStr: response.SendStatusSet[0].Code,
                                requestId: response.RequestId
                            }
                        })
                    } else {
                        resolve({
                            errorCode: -1003,//短信验证码发送失败
                            errorMessage: response.SendStatusSet[0].Message,
                            data: {
                                codeStr: response.SendStatusSet[0].Code,
                                requestId: response.RequestId
                            }
                        })
                    }
                }
            });
        })
    }
    let queryResult = await smsPromise()
    return queryResult
}

/*
 * 功能：根据手机号获取短信验证码
 */
async function getSms(queryString) {
  const code = Math.random().toString().slice(-6);//生成6位数随机验证码
  const sessionId = Math.random().toString().slice(-8);//生成8位随机数
  const sessionCode = {
      code: code,
      sessionId: sessionId,
      sendTime: new Date().getTime(),
      num: 0,//验证次数，最多可验证3次
      used: 1//1-未使用，2-已使用
  }
  clearCacheCode()
cacheCode[queryString.phone] = sessionCode

/*



功能：登录

*/

async function loginSms(queryString) {

const connection = mysql.createConnection({

host: '', // The ip address of cloud database instance, 云数据库实例 IP 地址

user: '', // The name of cloud database, for example, root, 云数据库用户名，例如 root

password: '', // Password of cloud database, 云数据库密码

database: '' // Name of the cloud database, 数据库名称

});

connection.connect();
if(queryString.token) {

return await verifyToken(connection, queryString)

}
if(!queryString.code || !queryString.sessionId) {

return {

errorCode: -1001,

errorMessage: "缺少参数"

}

}
let result = cacheCode[queryString.phone]

if(!result || result.used === 2 || result.num >= 3) {

return {

errorCode: -1100,

errorMessage: "验证码已失效"

}

}

if(result.sessionId !== queryString.sessionId) {

return {

errorCode: -1103,

errorMessage: "sessionId不匹配"

}

}
if(result.code == queryString.code) {

cacheCode[queryString.phone].used = 2;//将验证码更新为已使用

const queryInfoSql = select * from info where phone = ?

let queryInfoResult = await wrapPromise(connection, queryInfoSql, [queryString.phone])

if(queryInfoResult.length === 0) {//没有找到记录，未注册

return await generateInfo(connection, queryString)

} else {

let infoResult = queryInfoResult[0]

return {

errorCode: 0,

errorMessage: "登录成功",

data: {

phone: infoResult.phone,

token: getToken(infoResult.userId, infoResult),

name: infoResult.name,

avatar: infoResult.avatar,

userId: infoResult.userId.toString()

}

}

}

} else {

updateCacheCode(queryString.phone, result)

return {

errorCode: -1102,

errorMessage: "验证码错误，请重新输入"

}

}

}

/*



功能：利用 json web token 签发一个 token

*/

function getToken(userId, infoResult) {

return jwt.sign({

phone: infoResult.phone,

userId: userId,

name: infoResult.name,

avatar: infoResult.avatar

}, privateKey, {expiresIn: tokenExpireTime});

}