com.fhs.config.FileServiceConfig Maven / Gradle / Ivy

Go to download
package com.fhs.config;

import com.fhs.fileStorage.FileStorage;
import com.fhs.fileStorage.impl.AliyunOSSFileStorage;
import com.fhs.fileStorage.impl.DiskFileStorage;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.converter.ByteArrayHttpMessageConverter;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;

import java.util.List;

@Configuration
public class FileServiceConfig extends WebMvcConfigurerAdapter{


    /**
     * 储存类型默认是硬盘储存
     */
    @Value("${fhs.file.storageType:disk}")
    private String storageType;

    @Override
    public void configureMessageConverters(List> converters) {
        super.configureMessageConverters(converters);
        converters.add(new ByteArrayHttpMessageConverter());
    }


    /**
     * 初始化文件储存器
     * @return  根据条件初始化文件储存器
     */
    @Bean
    public FileStorage fileStorage(){
        //如果是阿里云则使用阿里云oss文件储存器
        if("oss".equals(storageType))
        {
            return new AliyunOSSFileStorage();
        }
        return new DiskFileStorage();
    }
}