liujie
2026-01-29 8ab97da36f98d75daed63cb795f037a13c530024
ruoyi-system/src/main/java/com/ruoyi/system/service/impl/UserFileServiceImpl.java
@@ -5,7 +5,13 @@
import com.ruoyi.system.mapper.UserFileMapper;
import com.ruoyi.system.model.UserFile;
import com.ruoyi.system.service.UserFileService;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.InputStreamResource;
import org.springframework.stereotype.Service;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
/**
 * <p>
@@ -18,4 +24,23 @@
@Service
public class UserFileServiceImpl extends ServiceImpl<UserFileMapper, UserFile> implements UserFileService {
    @Value("${file.upload.dirFile}")
    private Path fileLocation;
    @Override
    public InputStreamResource resolveResource(UserFile userFile){
        try {
            var filePath = fileLocation.resolve(userFile.getName() + "." + userFile.getExtension()).normalize();
            if (!Files.exists(filePath)) {
                return null;
            }
            return new InputStreamResource(Files.newInputStream(filePath));
        }catch (Exception e){
            e.printStackTrace();
        }
        return null;
    }
}