FastDFS--上传下载工具类

工具类基于SpringBoot JDK8

1. pom文件--git下载源码,推送至maven私服或者自行在maven仓库搜索

<!-- fastdfs客户端 -->
<dependency>
    <groupId>com.hy.fastdfs</groupId>
    <artifactId>client</artifactId>
    <!--去除日志依赖-->
    <exclusions>
        <exclusion>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-log4j12</artifactId>
        </exclusion>
    </exclusions>
</dependency>

2. application.yml

fastdfs:
  connect_timeout_in_seconds: 5
  network_timeout_in_seconds: 10
  charset: UTF-8
  http_anti_steal_token: false
  http_secret_key: FastDFS1234567890
  http_tracker_http_port: 8888
  http_server: http://127.0.0.1/
  #以下是必选项
  #多个用逗号隔开
  tracker_servers: 127.0.0.1:22122
  #url访问前缀
  default_group_name: group1
  url_prefix: group1

3. 自启动注解--用于启动类上

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@Import({FastDFSClientConfig.class})
public @interface EnableFastDFSClient {
}

4. config配置类

/**
 * @author huheng
 * @desc fastdfs配置
 * @date 2020-02-17
 */
@ConfigurationProperties(prefix = "fastdfs")
@Data
public class FastDFSConfig {
    public static final FastDFSConfig config = new FastDFSConfig();

    private String httpServer;
    // url前缀
    private static final String FASTDFS_URL_PREFIX = "fastdfs.url_prefix";
    // 连接超时
    private int connectTimeoutInSeconds = ClientGlobal.DEFAULT_CONNECT_TIMEOUT;
    // 请求超时时间
    private int networkTimeoutInSeconds = ClientGlobal.DEFAULT_NETWORK_TIMEOUT;
    // 字符集
    private String charset = ClientGlobal.DEFAULT_CHARSET;
    // token
    private boolean httpAntiStealToken = ClientGlobal.DEFAULT_HTTP_ANTI_STEAL_TOKEN;
    // key
    private String httpSecretKey = ClientGlobal.DEFAULT_HTTP_SECRET_KEY;
    private int httpTrackerHttpPort = ClientGlobal.DEFAULT_HTTP_TRACKER_HTTP_PORT;

    private String trackerServers;
    private String urlPrefix;

    private ConnectionPool connectionPool = new ConnectionPool();

    @Data
    public static class ConnectionPool {
        private boolean enabled = ClientGlobal.DEFAULT_CONNECTION_POOL_ENABLED;
        private int maxCountPerEntry = ClientGlobal.DEFAULT_CONNECTION_POOL_MAX_COUNT_PER_ENTRY;
        private int maxIdleTime = ClientGlobal.DEFAULT_CONNECTION_POOL_MAX_IDLE_TIME;
        private int maxWaitTimeInMs = ClientGlobal.DEFAULT_CONNECTION_POOL_MAX_WAIT_TIME_IN_MS;
    }

}

5. 客户端配置

/**
 * @author huheng
 * @desc fastdfs客户端配置
 * @date 2020-02-17
 */
@Configuration
@EnableConfigurationProperties(FastDFSConfig.class)
public class FastDFSClientConfig {
    @Resource
    private FastDFSConfig fastdfsConfig;

    @Bean
    public StorageClient1 storageClient1() throws Exception {
        return FastDFSUtil.init(fastdfsConfig);
    }
}

6. 工具类

/**
 * @author huheng
 * @desc fastdfs工具类
 * @date 2020-02-17
 */
public class FastDFSUtil {
    private static StorageClient1 storageClient1 = null;
    private static String fastdfsServer = null;

    public static StorageClient1 init(FastDFSConfig fastdfsConfig) throws Exception {
        Properties properties = propertiesFromConfig(fastdfsConfig);
        ClientGlobal.initByProperties(properties);

        TrackerClient trackerClient = new TrackerClient(ClientGlobal.g_tracker_group);
        TrackerServer trackerServer = trackerClient.getTrackerServer();
        StorageServer storageServer = trackerClient.getStoreStorage(trackerServer);
        storageClient1 = new StorageClient1(trackerServer, storageServer);
        fastdfsServer = fastdfsConfig.getHttpServer();
        return storageClient1;
    }

    private static Properties propertiesFromConfig(FastDFSConfig fastdfsConfig) {
        Properties properties = new Properties();
        properties.put(ClientGlobal.PROP_KEY_CONNECT_TIMEOUT_IN_SECONDS, fastdfsConfig.getConnectTimeoutInSeconds());
        properties.put(ClientGlobal.PROP_KEY_NETWORK_TIMEOUT_IN_SECONDS, fastdfsConfig.getNetworkTimeoutInSeconds());
        properties.put(ClientGlobal.PROP_KEY_CHARSET, fastdfsConfig.getCharset());
        properties.put(ClientGlobal.PROP_KEY_HTTP_SECRET_KEY, fastdfsConfig.getHttpSecretKey());
        properties.put(ClientGlobal.PROP_KEY_HTTP_ANTI_STEAL_TOKEN, fastdfsConfig.isHttpAntiStealToken());
        properties.put(ClientGlobal.PROP_KEY_HTTP_TRACKER_HTTP_PORT, fastdfsConfig.getHttpTrackerHttpPort());
        properties.put(ClientGlobal.PROP_KEY_TRACKER_SERVERS, fastdfsConfig.getTrackerServers());
        properties.put(ClientGlobal.PROP_KEY_CONNECTION_POOL_ENABLED, fastdfsConfig.getConnectionPool().isEnabled());
        properties.put(ClientGlobal.PROP_KEY_CONNECTION_POOL_MAX_COUNT_PER_ENTRY, fastdfsConfig.getConnectionPool().getMaxCountPerEntry());
        // 连接池 maxIdlePerKey 最大空闲连接数(影响并发性能)
        properties.put(ClientGlobal.PROP_KEY_CONNECTION_POOL_MAX_IDLE_TIME, fastdfsConfig.getConnectionPool().getMaxIdleTime());
        properties.put(ClientGlobal.PROP_KEY_CONNECTION_POOL_MAX_WAIT_TIME_IN_MS, fastdfsConfig.getConnectionPool().getMaxWaitTimeInMs());
        return properties;
    }

    public static String getFastdfsServer() {
        return fastdfsServer;
    }

    /**
     * 获取此{@link StorageClient1}就可以做想做的事[上传、下载、删除等]
     */
    public static StorageClient1 getStorageClient1() {
        return storageClient1;
    }

    /**
     * 流的形式上传文件
     *
     * @param inputStream
     * @param fileExt
     * @return
     */
    public static String uploadFile(InputStream inputStream, String fileExt) {
        try {
            // 将得到的文件流转二进制，便于后面直接将二进制存到fdfs服务器
            byte[] fileByte = IOUtils.toByteArray(inputStream, 1024);
            return storageClient1.upload_file1(fileByte, fileExt, null);
        } catch (Exception ex) {
            return null;
        }
    }

    /**
     * @param fileByte 文件字节数组
     * @param fileExt  文件扩展名
     * @return
     */
    public static String uploadFile(byte[] fileByte, String fileExt) {
        try {
            return storageClient1.upload_file1(fileByte, fileExt, null);
        } catch (Exception ex) {
            return null;
        }
    }


    /**
     * 根据fileId来删除一个文件（我们现在用的就是这样的方式，上传文件时直接将fileId保存在了数据库中）
     *
     * @param fileId file_id源码中的解释file_id the file id(including group name and filename);例如 group1/M00/00/00/ooYBAFM6MpmAHM91AAAEgdpiRC0012.xml
     * @return 0为成功，非0为失败，具体为错误代码
     */
    public static int deleteFile(String fileId) {
        try {
            return storageClient1.delete_file1(fileId);
        } catch (Exception ex) {
            return -1;
        }
    }

    /**
     * 修改一个已经存在的文件
     *
     * @param oldFileId 原来旧文件的fileId, file_id源码中的解释file_id the file id(including group name and filename);例如 group1/M00/00/00/ooYBAFM6MpmAHM91AAAEgdpiRC0012.xml
     * @param fileByte  新文件字节数组
     * @param fileExt   新文件扩展名
     * @return
     */
    public static String modifyFile(String oldFileId, byte[] fileByte, String fileExt) {
        String fileId = null;
        try {
            // 先上传
            fileId = uploadFile(fileByte, fileExt);
            if (fileId == null) {
                return null;
            }
            // 再删除
            int delResult = deleteFile(oldFileId);
            if (delResult != 0) {
                return null;
            }
        } catch (Exception ex) {
            return null;
        }
        return fileId;
    }

    /**
     * 文件下载
     *
     * @return 返回一个字节数组
     */
    public static byte[] downloadFile(String fileId) {
        try {
            return storageClient1.download_file1(fileId);
        } catch (Exception ex) {
            return null;
        }
    }
}