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});
}