Java操作FastDFS

Java操作FastDFS

1、创建 Springboot 工程

此处省略

2、引入 pom依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>

<dependency>
    <groupId>net.oschina.zcx7878</groupId>
    <artifactId>fastdfs-client-java</artifactId>
    <version>1.27.0.0</version>
</dependency>

3、在 Resource 下创建 FastDFS配置文件

fdfs_client.conf

connect_timeout = 60 # 连接超时时间，单位为秒
network_timeout = 60 # 通信超时时间，单位为秒。发送或接收数据时。假设在超时时间后还不能发送或接收数据，则本次网络通信失败
charset = UTF-8 # 字符集
http.tracker_http_port = 8080 # .tracker的http端口
tracker_server = FastDFS的ip地址:22122 #  tracker服务器IP和端口设置

4、在 Resource 下创建 application.yml

spring:
  servlet:
    multipart:
      max-request-size: 10MB
      max-file-size: 10MB
server:
  port: 9008

5、工具类

5.1 文件实体类

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

/**
 * description
 * author Admin
 * date 2021/11/27
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class FastDFSFile {

    /**
     * 文件名字
     */
    private String name;

    /**
     * 文件内容
     */
    private byte[] content;

    /**
     * 文件扩展名
     */
    private String ext;

    /**
     * 文件MD5摘要值
     */
    private String md5;

    /**
     * 文件创建作者
     */
    private String author;

    public FastDFSFile(String name, byte[] content, String ext, String height, String width, String author) {
        super();
        this.name = name;
        this.content = content;
        this.ext = ext;
        this.author = author;
    }

    public FastDFSFile(String name, byte[] content, String ext) {
        super();
        this.name = name;
        this.content = content;
        this.ext =ext;
    }
}

5.2 文件工具类，封装底层方法

import org.csource.common.NameValuePair;
import org.csource.fastdfs.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.io.ClassPathResource;

import java.io.ByteArrayInputStream;
import java.io.InputStream;

/**
 * description FastDFS客户端工具类
 * author Admin
 * date 2021/11/27
 */
public class FastDFSClient {
    private static Logger logger = LoggerFactory.getLogger(FastDFSFile.class);

    /**
     * 初始化加载 FastDFS的TrackerServer配置
     */
    static {
        try {
            String filePath = new ClassPathResource("fdfs_client.conf").getFile().getAbsolutePath();
            ClientGlobal.init(filePath);
        } catch (Exception e) {
            logger.error("FastDFS 初始化失败", e);
        }
    }

    /**
     * 文件上传
     * @param file
     * @return
     */
    public static String[] upload(FastDFSFile file) {
        // 获取文件的作者
        NameValuePair[] meta_list = new NameValuePair[1];
        meta_list[0] = new NameValuePair("author", file.getAuthor());

        // 接收返回数据
        String[] uploadResults = null;
        StorageClient storageClient = null;

        try {
            // 创建 StorageClient 客户端对象
            storageClient = getTrackerClient();

            /**
             * 文件上传
             * 1) 文件字节数组
             * 2) 文件扩展名
             * 3) 文件作者
             */
            uploadResults = storageClient.upload_file(file.getContent(), file.getExt(), meta_list);
        } catch (Exception e) {
            logger.error("上传文件失败: " + file.getName(), e);
        }

        if (uploadResults ==null && storageClient != null) {
            logger.error("上传文件错误，错误码: " + storageClient.getErrorCode());
        }
        // 获取组名
        String groupName = uploadResults[0];
        // 获取文件存储路径
        String remoteFileName = uploadResults[1];
        return uploadResults;
    }

    /**
     * 获取文件信息
     * @param groupName 组名
     * @param remoteFileName 文件存储完整名
     * @return
     */
    public static FileInfo getFile(String groupName, String remoteFileName) {
        try {
            StorageClient storageClient = getTrackerClient();
            return storageClient.get_file_info(groupName, remoteFileName);
        } catch (Exception e) {
            logger.error("从FastDFS中获取文件错误", e);
        }
        return null;
    }

    /**
     * 文件下载
     * @param groupName
     * @param remoteFileName
     * @return
     */
    public static InputStream downFile(String groupName, String remoteFileName) {
        try {
            // 创建 StorageClient
            StorageClient storageClient = getTrackerClient();
            // 下载文件
            byte[] fileByte = storageClient.download_file(groupName, remoteFileName);
            ByteArrayInputStream ins = new ByteArrayInputStream(fileByte);
            return ins;
        } catch (Exception e) {
            logger.error("从FastDFS获取文件失败", e);
        }
        return null;
    }

    /**
     * 文件删除
     * @param groupName
     * @param remoteFileName
     */
    public static  void deleteFile(String groupName, String remoteFileName) {
        try {
            // 创建 StorageClient
            StorageClient storageClient = getTrackerClient();
            // 删除文件
            int i = storageClient.delete_file(groupName, remoteFileName);
        } catch (Exception e) {
            logger.error("从FastDFS中文件删除失败", e);
        }
    }

    /**
     * 获取Storage组
     * @param groupName 组名
     * @return
     */
    public static StorageServer[] getStoreStorages(String groupName) {
        try {
            // 创建 TrackerClient
            TrackerClient trackerClient = new TrackerClient();
            // 获取 TrackerServer
            TrackerServer trackerServer = trackerClient.getConnection();
            //获取 Storage 组
            return trackerClient.getStoreStorages(trackerServer, groupName);
        } catch (Exception e) {
            logger.error("从FastDFS获取Storage 组失败", e);
        }
        return null;
    }

    /**
     * 获取 Storage 信息,IP和端口
     * @param groupName
     * @param remoteFileName
     * @return
     */
    public static ServerInfo[] getFetchStoages(String groupName, String remoteFileName) {
        try {
            TrackerClient trackerClient = new TrackerClient();
            TrackerServer trackerServer = trackerClient.getConnection();
            return trackerClient.getFetchStorages(trackerServer, groupName, remoteFileName);
        } catch (Exception e) {
           logger.error("获取Storage信息，IP 和端口失败", e);
        }
        return null;
    }

    /**
     * 获取 Tracker 服务地址
     * @return
     */
    public static String getTrackerUrl() {
        return "http://" + getTrackerServer().getInetSocketAddress().getHostString()+":"+ClientGlobal.getG_tracker_http_port()+ "/";
    }

    /**
     * 获取Storage 客户端
     * @return
     */
    private static StorageClient getTrackerClient() {
        TrackerServer trackerServer = getTrackerServer();
        StorageClient storageClient = new StorageClient(trackerServer, null);
        return storageClient;
    }

    /**
     * 获取 Tracker
     * @return
     */
    private static TrackerServer getTrackerServer() {
        try {
            TrackerClient trackerClient = new TrackerClient();
            TrackerServer trackerServer = trackerClient.getConnection();
            return trackerServer;
        } catch (Exception e) {
            logger.error("获取Tracker 失败", e);
        }
        return null;
    }
}

6、文件上传

@PostMapping("upload")
public Result upload(MultipartFile file) {
    try {
        // 文件上传
        // 1. 获取文件属性
        // 1.1. 原始文件名
        String originalFilename = file.getOriginalFilename();
        // 1.2. 文件内容
        byte[] fileBytes = file.getBytes();
        // 1.3 文件扩展名  test.jpg -> jpg
        String ext = originalFilename.substring(originalFilename.lastIndexOf(".") + 1);

        // 2. 创建 FastDFSFile
        FastDFSFile fastDFSFile = new FastDFSFile(originalFilename, fileBytes, ext);
        // 3. 调用工具类上传
        // uploadResult 第一个只是 group 第二个值是：物理路径
        String[] uploadResult = FastDFSClient.upload(fastDFSFile);

        // 预览url
        String url = FastDFSClient.getTrackerUrl() + uploadResult[0] + "/" + uploadResult[1];

        return new Result(true, StatusCode.OK, "上传成功",url);
    } catch (Exception e) {
        return new Result(false, StatusCode.ERROR, "上传失败" + e.getMessage());
    }
}