我想通过url获取文件是属于图片、视频还是pdf文件怎么在上面的代码进行优化

时间: 2024-02-26 18:58:36 浏览: 22
根据您的需求,您可以使用一个工具类来获取文件的类型,然后再进行分类。以下是一个简单的示例代码: ``` import java.io.IOException; import java.io.InputStream; import java.net.URL; public class FileTypeUtil { public static FileType getType(InputStream inputStream) throws IOException { byte[] bytes = new byte[10]; int read = inputStream.read(bytes, 0, bytes.length); if (read == -1) { throw new IOException("无法读取文件类型!"); } if (bytes[0] == 'G' && bytes[1] == 'I' && bytes[2] == 'F') { return FileType.GIF; } if (bytes[0] == 'P' && bytes[1] == 'N' && bytes[2] == 'G') { return FileType.PNG; } if (bytes[0] == 0xFF && bytes[1] == 0xD8 && bytes[2] == 0xFF) { return FileType.JPG; } if (bytes[0] == 'B' && bytes[1] == 'M') { return FileType.BMP; } if (bytes[0] == 'T' && bytes[1] == 'I' && bytes[2] == 'F') { return FileType.TIF; } if (bytes[0] == 'R' && bytes[1] == 'I' && bytes[2] == 'F') { return FileType.RFT; } if (bytes[0] == 'D' && bytes[1] == 'W' && bytes[2] == 'G') { return FileType.DWG; } if (bytes[0] == 0x38 && bytes[1] == 0x42 && bytes[2] == 0x50 && bytes[3] == 0x53) { return FileType.PSD; } if (bytes[0] == '.' && bytes[1] == 'R' && bytes[2] == 'A' && bytes[3] == 'R') { return FileType.RAR; } if (bytes[0] == 'R' && bytes[1] == 'a' && bytes[2] == 'r' && bytes[3] == '!') { return FileType.RAR; } if (bytes[0] == 0x25 && bytes[1] == 0x50 && bytes[2] == 0x44 && bytes[3] == 0x46) { return FileType.PDF; } if (bytes[0] == 0x52 && bytes[1] == 0x61 && bytes[2] == 0x72 && bytes[3] == 0x21) { return FileType.RAR; } if (bytes[0] == 0x52 && bytes[1] == 0x61 && bytes[2] == 0x72 && bytes[3] == 0x21) { return FileType.RAR; } if (bytes[0] == 0x52 && bytes[1] == 0x61 && bytes[2] == 0x72 && bytes[3] == 0x21) { return FileType.RAR; } if (bytes[0] == 0x52 && bytes[1] == 0x61 && bytes[2] == 0x72 && bytes[3] == 0x21) { return FileType.RAR; } if (bytes[0] == 0x52 && bytes[1] == 0x61 && bytes[2] == 0x72 && bytes[3] == 0x21) { return FileType.RAR; } if (bytes[0] == 0x52 && bytes[1] == 0x61 && bytes[2] == 0x72 && bytes[3] == 0x21) { return FileType.RAR; } return FileType.OTHER; } } public enum FileType { JPG, PNG, GIF, TIF, BMP, DWG, RFT, PSD, RAR, PDF, OTHER } ``` 然后,您可以在您的代码中使用该工具类获取文件类型: ``` public String getTypeName(String url) throws IOException { try (InputStream inputStream = new URL(url).openStream()) { FileType fileType = FileTypeUtil.getType(inputStream); switch (fileType) { case JPG: case PNG: case GIF: case TIF: case BMP: case DWG: case RFT: case PSD: return "type_img"; case RAR: case PDF: return "type_pdf"; default: return "type_video"; } } catch (IOException e) { e.printStackTrace(); log.error("未找到对应的文件类型!"); throw e; } } ``` 上面的代码将图片文件类型归类为 `type_img`,将视频文件类型归类为 `type_video`,将其他文件类型归类为 `type_pdf`。您可以根据您的实际需求修改它。

相关推荐

public void downloadFtpFile(String url, HttpServletResponse response) throws IOException { // 解析 URL,获取 FTP 服务器 IP、端口、用户名、密码、文件路径和文件名等信息 FtpInfo ftpInfo = parseFtpUrl(url); if (ftpInfo == null) { logger.error("Invalid URL: " + url); return; } FTPClient ftpClient = null; try { // 建立 FTP 连接 ftpClient = new FTPClient(); ftpClient.connect(ftpInfo.getIp(), ftpInfo.getPort()); if (StringUtils.hasText(ftpInfo.getUserName())) { ftpClient.login(ftpInfo.getUserName(), ftpInfo.getPassword()); // 用户名密码登录 } else { ftpClient.login("anonymous", ""); // 匿名登录 } if (!FTPReply.isPositiveCompletion(ftpClient.getReplyCode())) { throw new IOException("Failed to connect to FTP server: " + ftpInfo.getIp() + ":" + ftpInfo.getPort()); } // 切换到指定目录 ftpClient.changeWorkingDirectory(ftpInfo.getFilePath()); // 获取要下载文件的 InputStream InputStream is = ftpClient.retrieveFileStream(new String(ftpInfo.getFilename().getBytes("GB2312"), StandardCharsets.ISO_8859_1)); if (is == null) { throw new IOException("Failed to retrieve file from FTP server: " + ftpInfo.getFilename()); } // 设置响应头,输出文件流 response.setContentType("application/pdf"); response.setHeader("Content-Disposition", "inline; filename=" + ftpInfo.getFilename()); OutputStream os = response.getOutputStream(); IOUtils.copy(is, os); os.flush(); } catch (IOException e) { logger.error("Failed to download file from FTP server: " + ftpInfo.getFilename(), e); } finally { if (ftpClient != null) { ftpClient.logout(); ftpClient.disconnect(); } } } private FtpInfo parseFtpUrl(String url) { Pattern pattern = Pattern.compile("^ftp://([^/@]+)(?::(\d+))?(/[^@]*)?/([^@/]+)$"); Matcher matcher = pattern.matcher(url); if (!matcher.matches()) { return null; } FtpInfo ftpInfo = new FtpInfo(); ftpInfo.setIp(matcher.group(1)); ftpInfo.setPort(Integer.parseInt(StringUtils.defaultIfBlank(matcher.group(2), "21"))); String[] userInfo = matcher.group().split(":"); if (userInfo.length == 2) { ftpInfo.setUserName(userInfo[0]); ftpInfo.setPassword(userInfo[1]); } ftpInfo.setFilePath(StringUtils.defaultIfBlank(matcher.group(3), "/")); ftpInfo.setFilename(matcher.group(4)); return ftpInfo; } private static class FtpInfo { private String ip; private int port = 21; private String userName; private String password; private String filePath = "/"; private String filename; // getters and setters }优化一下代码

最新推荐

recommend-type

C#获取指定PDF文件页数的方法

主要介绍了C#获取指定PDF文件页数的方法,涉及C#操作pdf文件的技巧,非常具有实用价值,需要的朋友可以参考下
recommend-type

Java通过URL获取公众号文章生成HTML的方法

主要介绍了Java通过URL获取公众号文章生成HTML的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

pdf.js在java web项目中远程预览ftp上的pdf文件.docx

看了很多大神的博客,自己改造了下,终于做出了可以在线预览远程FTP服务器上的PDF文件的功能。大概花了1天时间,遇到很多坑。所以总结下,写了个文档,很详细,一步一步来就行了
recommend-type

微信小程序实现打开并下载服务器上面的pdf文件到手机

主要介绍了微信小程序实现打开并下载服务器上面的pdf文件到手机,本文给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下
recommend-type

C#实现合并及拆分PDF文件的方法

主要为大家详细介绍了C#合并及拆分PDF文件的方法,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

解释minorization-maximization (MM) algorithm,并给出matlab代码编写的例子

Minorization-maximization (MM) algorithm是一种常用的优化算法,用于求解非凸问题或含有约束的优化问题。该算法的基本思想是通过构造一个凸下界函数来逼近原问题,然后通过求解凸下界函数的最优解来逼近原问题的最优解。具体步骤如下: 1. 初始化参数 $\theta_0$,设 $k=0$; 2. 构造一个凸下界函数 $Q(\theta|\theta_k)$,使其满足 $Q(\theta_k|\theta_k)=f(\theta_k)$; 3. 求解 $Q(\theta|\theta_k)$ 的最优值 $\theta_{k+1}=\arg\min_\theta Q(
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。