Java 11:如何从InputStream获取文件扩展名并仅处理CSV文件?
问题解答
先给你说清楚:没法直接从InputStream里获取文件扩展名。因为扩展名是文件在文件系统里的命名属性,InputStream只负责传输文件的内容数据,完全不携带文件名、扩展名这类元信息。
你的需求是区分CSV和TXT文件、只处理CSV,不能靠扩展名的话,得换思路——基于文件内容特征来判断,毕竟CSV是结构化文本,和无固定格式的普通TXT有明显区别:
- 先处理流的可重置问题
InputStream默认是单向读取的,读取后指针不会回头,所以判断前要先确保能重置流的位置,避免影响后续业务处理。可以把原始流包装成BufferedInputStream,它支持标记和重置:
BufferedInputStream bis = new BufferedInputStream(is); bis.mark(1024); // 标记当前位置,参数设为足够读取前几行的字节数即可
- 读取前几行判断CSV特征
CSV的核心是每行有多个用分隔符(通常是逗号)分隔的字段,可以读取前1-3行,检查是否符合这个特征:
String line; boolean isCsv = false; BufferedReader reader = new BufferedReader(new InputStreamReader(bis)); // 读取前3行做判断 for (int i = 0; i < 3; i++) { line = reader.readLine(); if (line == null) break; // 简单判断逻辑:行内包含逗号,且拆分后的字段数至少2个(可根据业务调整) String[] fields = line.split(","); if (fields.length >= 2) { isCsv = true; break; } } // 重置流到标记位置,供后续业务处理使用 bis.reset();
- 关于
URLConnection.guessContentTypeFromStream返回null的问题
这个方法靠文件头的「魔数」判断MIME类型,但CSV和TXT都属于纯文本,没有独特的魔数标识,所以要么返回text/plain,要么返回null,根本没法区分两者,对你的需求完全没用。
总结:放弃从流里获取扩展名的想法,转而通过内容特征判断CSV,这种方式比依赖扩展名更可靠——毕竟用户可能随便修改文件名。
内容的提问来源于stack exchange,提问作者Jeya Kumar
相关产品推荐
相关产品推荐

