定期自动将云端数据库备份至云存储中

微信小程序使用的是云开发,之前都是通过手工保存云端数据库中多个表单的方式,以此进行数据备份。显然不够智能,也比较浪费时间。所以,此次写了个云函数,命名为autoBackupDatabase,设置好触发器后,即可自动将云数据库保存至云存储中。

从云数据库备份至云存储的主要步骤:

1. 取得access_token;
2. 取得数据库表单查询后的job_id;
3. 通过取得的job_id获取到对应的数据库备份文件的url;
4. 通过数据库备份文件的url,取得该url对应的文件内容;
5. 将取得的文件内容上传至云存储中;
6. 上传云函数并设置触发器。

以下为autoBackupDatabase云函数的全部代码:

// 云函数入口文件
const cloud = require('wx-server-sdk');
const request = require('request');
cloud.init({ env: 'xxxxxx' }); // 使用当前云环境

// 1.
// 获得access_token
async function getAccessToken(appid, secret) {
    return new Promise((resolve, reject) => {
        request.get(
            `https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=${appid}&secret=${secret}`,
            (err, res, body) => {
                if (err) {
                    reject(err)
                }
                resolve(JSON.parse(body))
            }
        )
    })
}

// 2.
// 进行数据库的查询操作,并导出一个job_id以获取文件地址
async function createExportJob(access_token, env, collection) {
    const date = new Date().toISOString();

    return new Promise((resolve, reject) => {
        request.post(
            `https://api.weixin.qq.com/tcb/databasemigrateexport?access_token=${access_token}`,
            {
                body: JSON.stringify({
                    env, // 云开发的环境ID
                    file_path: `${date}.json`, // 准备生成的文件名称
                    file_type: '1', // '1' 为 JSON, '2' 为 CSV
                    query: `db.collection("${collection}").get()` // 预设的查询命令
                })
            },
            (err, res, body) => {
                if (err) {
                    reject(err)
                }

                resolve(JSON.parse(body))
            }
        )
    })
}

// 3.
// 从 access_token 和 job_id 获取到 目标文件的url,
// 防止无法取到值,设置了一个500ms的setInterval去不断执行。
async function waitJobFinished(access_token, env, job_id) {
    return new Promise((resolve, reject) => {
        const timer = setInterval(() => {
            request.post(
                `https://api.weixin.qq.com/tcb/databasemigratequeryinfo?access_token=${access_token}`,
                {
                    body: JSON.stringify({
                        env,
                        job_id
                    })
                },
                (err, res, body) => {
                    if (err) {
                        reject(err)
                    }

                    const { status, file_url } = JSON.parse(body)

                    console.log("查询")

                    if (status === 'success') {
                        clearInterval(timer)
                        resolve(file_url)
                    }
                }
            )
        }, 500)
    })
}

// 4.
// 从目标文件的url取得具体内容
async function getJsonContent(file_url) {
    return new Promise((resolve, reject) => {
        request.get(
            `${file_url}`,
            (err, res, body) => {
                if (err) {
                    reject(err)
                }
                resolve(body)
            }
        )
    })
}

// 云函数入口函数
exports.main = async (event, context) => {
    const appid = "xxxxxxx";
    const secret = "xxxxxxx";
    const envId = "xxxxxxx" // 云开发的环境ID
    // 需要备份的集合名称
    const backupFromTables = ["table1", "table2", "table3"];
    // 云端备份所存放的目录
    const folderName = "databaseBackup"
    const dateString = (new Date()).toISOString()

    // count是为统计backupFromTables中已完成备份多少张表单的计数器
    var count = 0
    var access_token = ""

    // 1. 
    try {
        const result = await getAccessToken(appid, secret)
        access_token = result.access_token

        if (typeof access_token == null || access_token== "" || access_token == "undefined") {
            throw new Error(`获取 access_token 失败: ${result.errmsg}` || '获取 access_token 为空')
        }
    } catch (e) {
        throw new Error(`获取 access_token: ${e.message}`);
    }

    for(index in backupFromTables) {
        const backupFromTable = backupFromTables[index]

        // 2.
        createExportJob(access_token, envId, backupFromTable)
        .then(res => {
            console.log(res)
            if(res.errcode !== 0) {
                throw new Error("Can't get job_id")
            }
            console.log(res.job_id)
            // 3.
            return waitJobFinished(access_token, envId, res.job_id)
        })
        .then(file_url => {
            // 4.
            return getJsonContent(file_url)
        })
        .then(fileContent => {
            // 5.
            return cloud.uploadFile({
                cloudPath: './' + folderName + '/' + dateString.slice(0,10) + '/' + dateString + '_' + backupFromTable + '.json',
                fileContent: fileContent
            })
        })
        .then(res => {
            count++
            console.log("--->---- " + count + "/" + backupFromTables.length+ ":" + backupFromTable + "已完成备份 ---<----")
        })
        .catch( error => {
            throw new Error(`导出数据库异常: ${error.message}`);
        })
    }

}