ossService/client/client.go

package client

import (
	"app.yhyue.com/moapp/jybase/api"
	"bytes"
	"encoding/json"
	"errors"
	"io"
	"jygit.jydev.jianyu360.cn/BaseService/ossService/constant"
	"jygit.jydev.jianyu360.cn/BaseService/ossService/entity"
	"log"
	"mime/multipart"
	"net/http"
	"net/rpc"
	"strconv"
)

/* 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 对象名称
 * @param fileBt 文件流
 * @param gzip 是否压缩
 * @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.Data = respBody
	}
	return
}

/* restful方式删除
 * @param domain 域名，例如：https://ossservice.jianyu360.cn
 * @param bucketId 桶id
 * @param objectName 对象名称
 * @param fileBt 文件流
 * @param gzip 是否压缩
 * @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
}

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