|
|
@@ -1,234 +0,0 @@
|
|
|
-package client
|
|
|
-
|
|
|
-import (
|
|
|
- "bytes"
|
|
|
- "encoding/json"
|
|
|
- "errors"
|
|
|
- "io"
|
|
|
- "log"
|
|
|
- "mime/multipart"
|
|
|
- "net/http"
|
|
|
- "net/rpc"
|
|
|
- "strconv"
|
|
|
-
|
|
|
- "app.yhyue.com/moapp/jybase/api"
|
|
|
- "jygit.jydev.jianyu360.cn/BaseService/ossService/client/constant"
|
|
|
- "jygit.jydev.jianyu360.cn/BaseService/ossService/client/entity"
|
|
|
-)
|
|
|
-
|
|
|
-/* restful方式上传
|
|
|
- * @param domain 域名，例如：https://ossservice.jianyu360.cn
|
|
|
- * @param bucketId 桶id
|
|
|
- * @param objectName 对象名称
|
|
|
- * @param stream 文件流
|
|
|
- * @param gzip 是否压缩
|
|
|
- * @return {"error_code":0,"error_msg":"上传成功"}
|
|
|
- */
|
|
|
-func UpLoadByRestful(domain, bucketId, objectName string, stream []byte, gzip bool) (reply *api.Result) {
|
|
|
- reply = &api.Result{Error_code: -1}
|
|
|
-
|
|
|
- // 创建一个缓冲区来存储表单数据
|
|
|
- body := &bytes.Buffer{}
|
|
|
- writer := multipart.NewWriter(body)
|
|
|
- writer.WriteField("bucket_id", bucketId)
|
|
|
- writer.WriteField("object_name", objectName)
|
|
|
- writer.WriteField("gzip", strconv.FormatBool(gzip))
|
|
|
-
|
|
|
- // 创建表单字段
|
|
|
- part, err := writer.CreateFormFile("file", objectName)
|
|
|
- if err != nil {
|
|
|
- reply.Error_msg = err.Error()
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 模拟文件流
|
|
|
- fileStream := bytes.NewReader(stream)
|
|
|
-
|
|
|
- // 将文件流复制到表单字段
|
|
|
- _, err = io.Copy(part, fileStream)
|
|
|
- if err != nil {
|
|
|
- reply.Error_msg = err.Error()
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 创建 HTTP 请求
|
|
|
- if respBody, err := post(domain+constant.UploadUrl, writer, body); err != nil {
|
|
|
- reply.Error_msg = err.Error()
|
|
|
- } else {
|
|
|
- json.Unmarshal(respBody, &reply)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-/* restful方式下载
|
|
|
- * @param domain 域名，例如：https://ossservice.jianyu360.cn
|
|
|
- * @param bucketId 桶id
|
|
|
- * @param objectName 对象名称
|
|
|
- * @return {"error_code":0,"error_msg":"下载成功"}
|
|
|
- */
|
|
|
-func DownloadByRestful(domain, bucketId, objectName string) (reply *api.Result) {
|
|
|
- reply = &api.Result{}
|
|
|
- // 创建一个缓冲区来存储表单数据
|
|
|
- body := &bytes.Buffer{}
|
|
|
- writer := multipart.NewWriter(body)
|
|
|
- writer.WriteField("bucket_id", bucketId)
|
|
|
- writer.WriteField("object_name", objectName)
|
|
|
- if respBody, err := post(domain+constant.DownloadUrl, writer, body); err != nil {
|
|
|
- reply.Error_msg = err.Error()
|
|
|
- } else {
|
|
|
- reply.Error_msg = constant.DownloadSuccess
|
|
|
- reply.Data = respBody
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-/* restful方式删除
|
|
|
- * @param domain 域名，例如：https://ossservice.jianyu360.cn
|
|
|
- * @param bucketId 桶id
|
|
|
- * @param objectName 对象名称
|
|
|
- * @return {"error_code":0,"error_msg":"上传成功"}
|
|
|
- */
|
|
|
-func DeleteByRestful(domain, bucketId, objectName string) (reply *api.Result) {
|
|
|
- reply = &api.Result{}
|
|
|
- // 创建一个缓冲区来存储表单数据
|
|
|
- body := &bytes.Buffer{}
|
|
|
- writer := multipart.NewWriter(body)
|
|
|
- writer.WriteField("bucket_id", bucketId)
|
|
|
- writer.WriteField("object_name", objectName)
|
|
|
- if respBody, err := post(domain+constant.DeleteUrl, writer, body); err != nil {
|
|
|
- reply.Error_msg = err.Error()
|
|
|
- } else {
|
|
|
- json.Unmarshal(respBody, &reply)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-/* restful方式获取标讯正文
|
|
|
- * @param domain 域名，例如：https://ossservice.jianyu360.cn
|
|
|
- * @param bucketId 桶id
|
|
|
- * @param objectName 对象名称
|
|
|
- * @return {"error_code":0,"error_msg":"获取正文成功","data":"正文内容"}
|
|
|
- */
|
|
|
-func GetBidDetailByRestful(domain, bucketId, objectName string) (reply *api.Result) {
|
|
|
- reply = &api.Result{}
|
|
|
- // 创建一个缓冲区来存储表单数据
|
|
|
- body := &bytes.Buffer{}
|
|
|
- writer := multipart.NewWriter(body)
|
|
|
- writer.WriteField("bucket_id", bucketId)
|
|
|
- writer.WriteField("object_name", objectName)
|
|
|
- if respBody, err := post(domain+constant.GetBidDetailUrl, writer, body); err != nil {
|
|
|
- reply.Error_msg = err.Error()
|
|
|
- } else {
|
|
|
- reply.Error_msg = constant.GetBidDetailSuccess
|
|
|
- reply.Data = string(respBody)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-func post(url string, writer *multipart.Writer, body *bytes.Buffer) ([]byte, error) {
|
|
|
- // 关闭表单写入器
|
|
|
- if err := writer.Close(); err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- // 创建 HTTP 请求
|
|
|
- req, err := http.NewRequest("POST", url, body)
|
|
|
- if err != nil {
|
|
|
- log.Println("Error creating request:", err)
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- // 设置请求头
|
|
|
- req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
|
-
|
|
|
- // 发送请求
|
|
|
- client := &http.Client{}
|
|
|
- resp, err := client.Do(req)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- defer resp.Body.Close()
|
|
|
-
|
|
|
- // 读取响应
|
|
|
- respBody, err := io.ReadAll(resp.Body)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- if resp.StatusCode != http.StatusOK {
|
|
|
- return nil, errors.New(string(respBody))
|
|
|
- }
|
|
|
- return respBody, nil
|
|
|
-}
|
|
|
-
|
|
|
-/* rpc方式上传
|
|
|
- * @param address 域名，例如：192.168.3.206:8110
|
|
|
- * @param args 参数
|
|
|
- * @param args.BucketID 文件名
|
|
|
- * @param args.objectName 对象名称
|
|
|
- * @param args.Stream 文件流
|
|
|
- * @param args.Gzip 是否压缩
|
|
|
- * @return {"error_code":0,"error_msg":"上传成功"}
|
|
|
- * @return error 错误信息
|
|
|
- */
|
|
|
-func UpLoadByRpc(address string, args *entity.UploadArgs) (api.Result, error) {
|
|
|
- var reply api.Result
|
|
|
- err := rpcCall(address, "OSSService.Upload", args, &reply)
|
|
|
- return reply, err
|
|
|
-}
|
|
|
-
|
|
|
-/*
|
|
|
- *rpc方式下载
|
|
|
- * @param address 域名，例如：192.168.3.206:8110
|
|
|
- * @param args 参数
|
|
|
- * @param args.BucketID 文件名
|
|
|
- * @param args.objectName 对象名称
|
|
|
- * @return {"error_code":0,"error_msg":"下载成功","data":"文件流"}
|
|
|
- * @return error 错误信息
|
|
|
- */
|
|
|
-func DownloadByRpc(address string, args *entity.Args) (api.Result, error) {
|
|
|
- var reply api.Result
|
|
|
- err := rpcCall(address, "OSSService.Download", args, &reply)
|
|
|
- return reply, err
|
|
|
-}
|
|
|
-
|
|
|
-/* rpc方式删除
|
|
|
- * @param address 域名，例如：192.168.3.206:8110
|
|
|
- * @param args 参数
|
|
|
- * @param args.BucketID 文件名
|
|
|
- * @param args.objectName 对象名称
|
|
|
- * @return {"error_code":0,"error_msg":"删除成功"}
|
|
|
- * @return error 错误信息
|
|
|
- */
|
|
|
-func DeleteByRpc(address string, args *entity.Args) (api.Result, error) {
|
|
|
- var reply api.Result
|
|
|
- err := rpcCall(address, "OSSService.Delete", args, &reply)
|
|
|
- return reply, err
|
|
|
-}
|
|
|
-
|
|
|
-/*
|
|
|
- *rpc方式获取标讯正文
|
|
|
- * @param address 域名，例如：192.168.3.206:8110
|
|
|
- * @param args 参数
|
|
|
- * @param args.BucketID 文件名
|
|
|
- * @param args.objectName 对象名称
|
|
|
- * @return {"error_code":0,"error_msg":"下载成功","data":"正文内容"}
|
|
|
- * @return error 错误信息
|
|
|
- */
|
|
|
-func GetBidDetailByRpc(address string, args *entity.Args) (api.Result, error) {
|
|
|
- var reply api.Result
|
|
|
- err := rpcCall(address, "OSSService.GetBidDetail", args, &reply)
|
|
|
- return reply, err
|
|
|
-}
|
|
|
-func rpcCall(address, serviceMethod string, args any, reply any) error {
|
|
|
- client, err := rpc.DialHTTP("tcp", address)
|
|
|
- if err != nil {
|
|
|
- log.Println(err)
|
|
|
- return err
|
|
|
- }
|
|
|
- defer client.Close()
|
|
|
- err = client.Call(serviceMethod, args, reply)
|
|
|
- if err != nil {
|
|
|
- log.Println(err)
|
|
|
- return err
|
|
|
- }
|
|
|
- return nil
|
|
|
-}
|
