|
|
@@ -0,0 +1,413 @@
|
|
|
+package tcb
|
|
|
+
|
|
|
+import (
|
|
|
+ "fmt"
|
|
|
+
|
|
|
+ "github.com/silenceper/wechat/util"
|
|
|
+)
|
|
|
+
|
|
|
+const (
|
|
|
+ //数据库导入
|
|
|
+ databaseMigrateImportURL = "https://api.weixin.qq.com/tcb/databasemigrateimport"
|
|
|
+ //数据库导出
|
|
|
+ databaseMigrateExportURL = "https://api.weixin.qq.com/tcb/databasemigrateexport"
|
|
|
+ //数据库迁移状态查询
|
|
|
+ databaseMigrateQueryInfoURL = "https://api.weixin.qq.com/tcb/databasemigratequeryinfo"
|
|
|
+ //变更数据库索引
|
|
|
+ updateIndexURL = "https://api.weixin.qq.com/tcb/updateindex"
|
|
|
+ //新增集合
|
|
|
+ databaseCollectionAddURL = "https://api.weixin.qq.com/tcb/databasecollectionadd"
|
|
|
+ //删除集合
|
|
|
+ databaseCollectionDeleteURL = "https://api.weixin.qq.com/tcb/databasecollectiondelete"
|
|
|
+ //获取特定云环境下集合信息
|
|
|
+ databaseCollectionGetURL = "https://api.weixin.qq.com/tcb/databasecollectionget"
|
|
|
+ //数据库插入记录
|
|
|
+ databaseAddURL = "https://api.weixin.qq.com/tcb/databaseadd"
|
|
|
+ //数据库删除记录
|
|
|
+ databaseDeleteURL = "https://api.weixin.qq.com/tcb/databasedelete"
|
|
|
+ //数据库更新记录
|
|
|
+ databaseUpdateURL = "https://api.weixin.qq.com/tcb/databaseupdate"
|
|
|
+ //数据库查询记录
|
|
|
+ databaseQueryURL = "https://api.weixin.qq.com/tcb/databasequery"
|
|
|
+ //统计集合记录数或统计查询语句对应的结果记录数
|
|
|
+ databaseCountURL = "https://api.weixin.qq.com/tcb/databasecount"
|
|
|
+
|
|
|
+ //冲突处理模式
|
|
|
+ ConflictModeInster ConflictMode = 1
|
|
|
+ ConflictModeUpsert ConflictMode = 2
|
|
|
+
|
|
|
+ //file_type 的合法值
|
|
|
+ FileTypeJson FileType = 1
|
|
|
+ FileTypeCsv FileType = 2
|
|
|
+)
|
|
|
+
|
|
|
+type ConflictMode int
|
|
|
+type FileType int
|
|
|
+
|
|
|
+//ValidDirections 合法的direction值
|
|
|
+var ValidDirections = []string{"1", "-1", "2dsphere"}
|
|
|
+
|
|
|
+//DatabaseMigrateExportReq 数据库出 请求参数
|
|
|
+type DatabaseMigrateExportReq struct {
|
|
|
+ Env string `json:"env,omitempty"` //云环境ID
|
|
|
+ FilePath string `json:"file_path,omitempty"` //导出文件路径(导入文件需先上传到同环境的存储中，可使用开发者工具或 HTTP API的上传文件 API上传）
|
|
|
+ FileType FileType `json:"file_type,omitempty"` //导出文件类型，文件格式参考数据库导入指引中的文件格式部分 1:json 2:csv
|
|
|
+ Query string `json:"query,omitempty"` //导出条件
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseMigrateExportRes 数据库导出 返回结果
|
|
|
+type DatabaseMigrateExportRes struct {
|
|
|
+ util.CommonError
|
|
|
+ JobID int64 `json:"job_id"` //导出任务ID，可使用数据库迁移进度查询 API 查询导入进度及结果
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseMigrateImportReq 数据库导入 请求参数
|
|
|
+type DatabaseMigrateImportReq struct {
|
|
|
+ Env string `json:"env,omitempty"` //云环境ID
|
|
|
+ CollectionName string `json:"collection_name,omitempty"` //集合名称
|
|
|
+ FilePath string `json:"file_path,omitempty"` //导出文件路径（文件会导出到同环境的云存储中，可使用获取下载链接 API 获取下载链接）
|
|
|
+ FileType FileType `json:"file_type,omitempty"` //导入文件类型，文件格式参考数据库导入指引中的文件格式部分 1:json 2:csv
|
|
|
+ StopOnError bool `json:"stop_on_error,omitempty"` //是否在遇到错误时停止导入
|
|
|
+ ConflictMode ConflictMode `json:"conflict_mode,omitempty"` //冲突处理模式 1:inster 2:UPSERT
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseMigrateImportReq 数据库导入 返回结果
|
|
|
+type DatabaseMigrateImportRes struct {
|
|
|
+ util.CommonError
|
|
|
+ JobID int64 `json:"job_id"` //导入任务ID，可使用数据库迁移进度查询 API 查询导入进度及结果
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseMigrateQueryInfoRes 数据库迁移状态查询
|
|
|
+type DatabaseMigrateQueryInfoRes struct {
|
|
|
+ util.CommonError
|
|
|
+ Status string `json:"status"` //导出状态
|
|
|
+ RecordSuccess int64 `json:"record_success"` //导出成功记录数
|
|
|
+ RecordFail int64 `json:"record_fail"` //导出失败记录数
|
|
|
+ ErrMsg string `json:"err_msg"` //导出错误信息
|
|
|
+ FileURL string `json:"file_url"` //导出文件下载地址
|
|
|
+}
|
|
|
+
|
|
|
+//UpdateIndexReq 变更数据库索引 请求参数
|
|
|
+type UpdateIndexReq struct {
|
|
|
+ Env string `json:"env,omitempty"` //云环境ID
|
|
|
+ CollectionName string `json:"collection_name,omitempty"` //集合名称
|
|
|
+ CreateIndexes []CreateIndex `json:"create_indexes,omitempty"` //新增索引
|
|
|
+ DropIndexes []DropIndex `json:"drop_indexes,omitempty"` //删除索引
|
|
|
+}
|
|
|
+
|
|
|
+//CreateIndex 新增索引
|
|
|
+type CreateIndex struct {
|
|
|
+ Name string `json:"name,omitempty"` //索引名
|
|
|
+ Unique bool `json:"unique,omitempty"` //是否唯一
|
|
|
+ Keys []CreateIndexKey `json:"keys,omitempty"` //索引字段
|
|
|
+}
|
|
|
+
|
|
|
+//CreateIndexKey create index key
|
|
|
+type CreateIndexKey struct {
|
|
|
+ Name string `json:"name,omitempty"` //字段名
|
|
|
+ Direction string `json:"direction,omitempty"` //字段排序
|
|
|
+}
|
|
|
+
|
|
|
+//DropIndex 删除索引
|
|
|
+type DropIndex struct {
|
|
|
+ Name string `json:"name,omitempty"`
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCollectionReq 新增/删除集合请求参数
|
|
|
+type DatabaseCollectionReq struct {
|
|
|
+ Env string `json:"env,omitempty"` //云环境ID
|
|
|
+ CollectionName string `json:"collection_name,omitempty"` //集合名称
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCollectionGetReq 获取特定云环境下集合信息请求
|
|
|
+type DatabaseCollectionGetReq struct {
|
|
|
+ Env string `json:"env,omitempty"` //云环境ID
|
|
|
+ Limit int64 `json:"limit,omitempty"` //获取数量限制
|
|
|
+ Offset int64 `json:"offset,omitempty"` //偏移量
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCollectionGetRes 获取特定云环境下集合信息结果
|
|
|
+type DatabaseCollectionGetRes struct {
|
|
|
+ util.CommonError
|
|
|
+ Pager struct {
|
|
|
+ Limit int64 `json:"limit"` //单次查询限制
|
|
|
+ Offset int64 `json:"offset"` //偏移量
|
|
|
+ Total int64 `json:"total"` //符合查询条件的记录总数
|
|
|
+ } `json:"pager"`
|
|
|
+ Collections []struct {
|
|
|
+ Name string `json:"name"` //集合名
|
|
|
+ Count int64 `json:"count"` //表中文档数量
|
|
|
+ Size int64 `json:"size"` //表的大小（即表中文档总大小），单位：字节
|
|
|
+ IndexCount int64 `json:"index_count"` //索引数量
|
|
|
+ IndexSize int64 `json:"index_size"` //索引占用大小，单位：字节
|
|
|
+ } `json:"collections"`
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseReq 数据库插入/删除/更新/查询/统计记录请求参数
|
|
|
+type DatabaseReq struct {
|
|
|
+ Env string `json:"env,omitempty"` //云环境ID
|
|
|
+ Query string `json:"query,omitempty"` //数据库操作语句
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseAddRes 数据库插入记录返回结果
|
|
|
+type DatabaseAddRes struct {
|
|
|
+ util.CommonError
|
|
|
+ IDList []string `json:"id_list"` //插入成功的数据集合主键_id。
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseDeleteRes 数据库删除记录返回结果
|
|
|
+type DatabaseDeleteRes struct {
|
|
|
+ util.CommonError
|
|
|
+ Deleted int64 `json:"deleted"` //删除记录数量
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseUpdateRes 数据库更新记录返回结果
|
|
|
+type DatabaseUpdateRes struct {
|
|
|
+ util.CommonError
|
|
|
+ Matched int64 `json:"matched"` //更新条件匹配到的结果数
|
|
|
+ Modified int64 `json:"modified"` //修改的记录数，注意：使用set操作新插入的数据不计入修改数目
|
|
|
+ ID string `json:"id"`
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseQueryRes 数据库查询记录 返回结果
|
|
|
+type DatabaseQueryRes struct {
|
|
|
+ util.CommonError
|
|
|
+ Pager struct {
|
|
|
+ Limit int64 `json:"limit"` //单次查询限制
|
|
|
+ Offset int64 `json:"offset"` //偏移量
|
|
|
+ Total int64 `json:"total"` //符合查询条件的记录总数
|
|
|
+ } `json:"pager"`
|
|
|
+ Data []string `json:"data"`
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCountRes 统计集合记录数或统计查询语句对应的结果记录数 返回结果
|
|
|
+type DatabaseCountRes struct {
|
|
|
+ util.CommonError
|
|
|
+ Count int64 `json:"count"` //记录数量
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseMigrateImport 数据库导入
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseMigrateImport.html
|
|
|
+func (tcb *Tcb) DatabaseMigrateImport(req *DatabaseMigrateImportReq) (*DatabaseMigrateImportRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseMigrateImportURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, req)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseMigrateImportRes := &DatabaseMigrateImportRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseMigrateImportRes, "DatabaseMigrateImport")
|
|
|
+ return databaseMigrateImportRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseMigrateExport 数据库导出
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseMigrateExport.html
|
|
|
+func (tcb *Tcb) DatabaseMigrateExport(req *DatabaseMigrateExportReq) (*DatabaseMigrateExportRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseMigrateExportURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, req)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseMigrateExportRes := &DatabaseMigrateExportRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseMigrateExportRes, "DatabaseMigrateExport")
|
|
|
+ return databaseMigrateExportRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//databaseMigrateQueryInfo 数据库迁移状态查询
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseMigrateQueryInfo.html
|
|
|
+func (tcb *Tcb) DatabaseMigrateQueryInfo(env string, jobID int64) (*DatabaseMigrateQueryInfoRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseMigrateQueryInfoURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, map[string]interface{}{
|
|
|
+ "env": env,
|
|
|
+ "job_id": jobID,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseMigrateQueryInfoRes := &DatabaseMigrateQueryInfoRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseMigrateQueryInfoRes, "DatabaseMigrateQueryInfo")
|
|
|
+ return databaseMigrateQueryInfoRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//updateIndex 变更数据库索引
|
|
|
+//https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/updateIndex.html
|
|
|
+func (tcb *Tcb) UpdateIndex(req *UpdateIndexReq) error {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", updateIndexURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, req)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ return util.DecodeWithCommonError(response, "UpdateIndex")
|
|
|
+}
|
|
|
+
|
|
|
+//databaseCollectionAdd 新增集合
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseCollectionAdd.html
|
|
|
+func (tcb *Tcb) DatabaseCollectionAdd(env, collectionName string) error {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseCollectionAddURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseCollectionReq{
|
|
|
+ Env: env,
|
|
|
+ CollectionName: collectionName,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ return util.DecodeWithCommonError(response, "DatabaseCollectionAdd")
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCollectionDelete 删除集合
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseCollectionDelete.html
|
|
|
+func (tcb *Tcb) DatabaseCollectionDelete(env, collectionName string) error {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseCollectionDeleteURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseCollectionReq{
|
|
|
+ Env: env,
|
|
|
+ CollectionName: collectionName,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ return util.DecodeWithCommonError(response, "DatabaseCollectionDelete")
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCollectionGet 获取特定云环境下集合信息
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseCollectionGet.html
|
|
|
+func (tcb *Tcb) DatabaseCollectionGet(env string, limit, offset int64) (*DatabaseCollectionGetRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseCollectionGetURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseCollectionGetReq{
|
|
|
+ Env: env,
|
|
|
+ Limit: limit,
|
|
|
+ Offset: offset,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseCollectionGetRes := &DatabaseCollectionGetRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseCollectionGetRes, "DatabaseCollectionGet")
|
|
|
+ return databaseCollectionGetRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//databaseAdd 数据库插入记录
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseAdd.html
|
|
|
+func (tcb *Tcb) DatabaseAdd(env, query string) (*DatabaseAddRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseAddURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseReq{
|
|
|
+ Env: env,
|
|
|
+ Query: query,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseAddRes := &DatabaseAddRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseAddRes, "DatabaseAdd")
|
|
|
+ return databaseAddRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseDelete 数据库插入记录
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseDelete.html
|
|
|
+func (tcb *Tcb) DatabaseDelete(env, query string) (*DatabaseDeleteRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseDeleteURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseReq{
|
|
|
+ Env: env,
|
|
|
+ Query: query,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseDeleteRes := &DatabaseDeleteRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseDeleteRes, "DatabaseDelete")
|
|
|
+ return databaseDeleteRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseUpdate 数据库插入记录
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseUpdate.html
|
|
|
+func (tcb *Tcb) DatabaseUpdate(env, query string) (*DatabaseUpdateRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseUpdateURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseReq{
|
|
|
+ Env: env,
|
|
|
+ Query: query,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseUpdateRes := &DatabaseUpdateRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseUpdateRes, "DatabaseUpdate")
|
|
|
+ return databaseUpdateRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseQuery 数据库查询记录
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseQuery.html
|
|
|
+func (tcb *Tcb) DatabaseQuery(env, query string) (*DatabaseQueryRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseQueryURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseReq{
|
|
|
+ Env: env,
|
|
|
+ Query: query,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseQueryRes := &DatabaseQueryRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseQueryRes, "DatabaseQuery")
|
|
|
+ return databaseQueryRes, err
|
|
|
+}
|
|
|
+
|
|
|
+//DatabaseCount 统计集合记录数或统计查询语句对应的结果记录数
|
|
|
+//reference:https://developers.weixin.qq.com/miniprogram/dev/wxcloud/reference-http-api/database/databaseCount.html
|
|
|
+func (tcb *Tcb) DatabaseCount(env, query string) (*DatabaseCountRes, error) {
|
|
|
+ accessToken, err := tcb.GetAccessToken()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ uri := fmt.Sprintf("%s?access_token=%s", databaseCountURL, accessToken)
|
|
|
+ response, err := util.PostJSON(uri, &DatabaseReq{
|
|
|
+ Env: env,
|
|
|
+ Query: query,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ databaseCountRes := &DatabaseCountRes{}
|
|
|
+ err = util.DecodeWithError(response, databaseCountRes, "DatabaseCount")
|
|
|
+ return databaseCountRes, err
|
|
|
+}
|
silenceper