存储文件到 腾讯 COS

添加依赖

    <dependency>
      <groupId>com.qcloud</groupId>
      <artifactId>cos_api</artifactId>
      <version>5.6.155</version>
    </dependency>

SystemTxCosConfigVo-腾讯云对象存储配置信息

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

/**
 * 腾讯云对象存储配置信息
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class SystemTxCosConfigVo {
  private String domain;
  private String region;
  private String bucketName;
  private String secretId;
  private String secretKey;

}

SysConfigConstantsService 从数据库中配置获取

import com.litongjava.jfinal.plugin.activerecord.Db;
import com.litongjava.tio.boot.admin.costants.TableNames;
import com.litongjava.tio.boot.admin.vo.SystemTxCosConfigVo;
import com.litongjava.tio.utils.json.Json;

public class SysConfigConstantsService {

  public SystemTxCosConfigVo getSystemTxCosConfig() {
    String sql = "select key_value from " + TableNames.tio_boot_admin_system_constants_config + " where key_name=?";
    String systemTxCosConfigString = Db.queryStr(sql, "systemTxCosConfig");
    return Json.getJson().parse(systemTxCosConfigString, SystemTxCosConfigVo.class);
  }
}

TencentStorageService 接收文件上传并入库

package com.litongjava.tio.boot.admin.services;

import cn.hutool.core.io.file.FileNameUtil;
import cn.hutool.crypto.digest.MD5;
import com.jfinal.kit.Kv;
import com.litongjava.data.model.DbJsonBean;
import com.litongjava.data.services.DbJsonService;
import com.litongjava.data.utils.SnowflakeIdGenerator;
import com.litongjava.jfinal.aop.Aop;
import com.litongjava.tio.boot.admin.costants.TableNames;
import com.litongjava.tio.boot.admin.vo.SystemTxCosConfigVo;
import com.litongjava.tio.http.common.UploadFile;
import com.litongjava.tio.utils.http.ContentTypeUtils;
import com.litongjava.tio.utils.resp.RespBodyVo;
import com.qcloud.cos.COSClient;
import com.qcloud.cos.ClientConfig;
import com.qcloud.cos.auth.BasicCOSCredentials;
import com.qcloud.cos.auth.COSCredentials;
import com.qcloud.cos.model.ObjectMetadata;
import com.qcloud.cos.model.PutObjectRequest;
import com.qcloud.cos.model.PutObjectResult;
import com.qcloud.cos.region.Region;
import lombok.extern.slf4j.Slf4j;

import java.io.ByteArrayInputStream;

/**
 * Created by Tong Li <https://github.com/litongjava>
 */
@Slf4j
public class TencentStorageService {
  public RespBodyVo upload(UploadFile uploadFile) {
    String filename = uploadFile.getName();
    int size = uploadFile.getSize();
    byte[] fileContent = uploadFile.getData();

    // 上传文件
    long threadId = Thread.currentThread().getId();
    if (threadId > 31L) {
      threadId %= 31L;
    }

    if (threadId < 0L) {
      threadId = 0L;
    }
    long id = (new SnowflakeIdGenerator(threadId, 0L)).generateId();
    String suffix = FileNameUtil.getSuffix(filename);
    String newFilename = id + "." + suffix;

    String targetName = "public/" + newFilename;

    return uploadBytes(filename, id, targetName, fileContent, size, suffix);
  }

  public RespBodyVo uploadBytes(String filename, long id, String targetName, byte[] fileContent, int size, String suffix) {
    SystemTxCosConfigVo systemTxCosConfig = Aop.get(SysConfigConstantsService.class).getSystemTxCosConfig();

    COSClient cosClient = getCosClient(systemTxCosConfig);
    String bucketName = systemTxCosConfig.getBucketName();

    String etag = null;
    try {
      etag = upload(cosClient, bucketName, targetName, fileContent, suffix);
    } catch (Exception e) {
      log.error("Error uploading file to Tencent COS", e);
      return RespBodyVo.fail(e.getMessage());
    } finally {
      cosClient.shutdown();
    }

    // Log and save to database
    log.info("Uploaded to COS with ETag: {}", etag);
    String md5 = MD5.create().digestHex(fileContent);
    Kv kv = Kv.create()
      .set("md5", md5)
      .set("filename", filename)
      .set("file_size", size)
      .set("platform", "tencent")
      .set("region_name", systemTxCosConfig.getRegion())
      .set("bucket_name", bucketName)
      .set("target_name", targetName)
      .set("file_id", etag);

    DbJsonBean<Kv> save = DbJsonService.getInstance().save(TableNames.tio_boot_admin_system_upload_file, kv);
    String downloadUrl = getUrl(bucketName, targetName);

    Kv kvResult = Kv.create()
      .set("id", save.getData().get("id").toString())
      .set("url", downloadUrl);

    return RespBodyVo.ok(kvResult);

  }

  /**
   * upload
   */
  private String upload(COSClient cosClient, String bucketName, String targetName, byte[] fileContent, String suffix) {
    PutObjectRequest putObjectRequest = getPutObjectRequest(targetName, fileContent, suffix, bucketName);
    PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
    return putObjectResult.getETag();
  }

  /**
   * getPutObjectRequest
   */
  private PutObjectRequest getPutObjectRequest(String targetName, byte[] fileContent, String suffix, String bucketName) {
    ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(fileContent);
    ObjectMetadata objectMetadata = new ObjectMetadata();
    objectMetadata.setContentLength(fileContent.length);
    objectMetadata.setContentType(ContentTypeUtils.getContentType(suffix));

    return new PutObjectRequest(bucketName, targetName, byteArrayInputStream, objectMetadata);
  }

  /**
   * getCosClient
   */
  private COSClient getCosClient(SystemTxCosConfigVo systemTxCosConfig) {
    COSCredentials cred = new BasicCOSCredentials(systemTxCosConfig.getSecretId(), systemTxCosConfig.getSecretKey());
    ClientConfig clientConfig = new ClientConfig(new Region(systemTxCosConfig.getRegion()));
    return new COSClient(cred, clientConfig);
  }

  public String getUrl(String bucketName, String targetName) {
    SystemTxCosConfigVo systemTxCosConfig = Aop.get(SysConfigConstantsService.class).getSystemTxCosConfig();
    String region = systemTxCosConfig.getRegion();
    String domain = "https://" + bucketName + ".cos." + region + ".myqcloud.com";
    return domain + "/" + targetName;
  }
}

添加 action 上传文件

  public HttpResponse uploadToTencentCos(HttpRequest request) throws Exception {
    HttpResponse httpResponse = TioControllerContext.getResponse();
    HttpServerResponseUtils.enableCORS(httpResponse, new HttpCors());

    String method = request.getMethod();
    if ("OPTIONS".equals(method)) {
      return httpResponse;
    }

    UploadFile uploadFile = request.getUploadFile("file");
    TencentStorageService storageService = Aop.get(TencentStorageService.class);
    if (uploadFile != null) {
      RespBodyVo respVo = storageService.upload(uploadFile);
      return Resps.json(httpResponse, respVo);

    }
    return Resps.json(httpResponse, RespBodyVo.ok("Fail"));
  }