解析不同的文件类型

问题描述

我正在学习 Kotlin,但在理解如何继续之前遇到了一些困难

目前我有一个从前端发送的 kml 文件,但现在我想接受 geoJson 并存储这个数据库 -> 所以我需要创建Kotlin 中用于验证文件类型并根据类型返回正确对象的函数

这是接受kml文件调用parseKmlTopolygons的函数

fun parseKmlTopolygons(file: multipartfile,applicationConfiguration: ApplicationConfiguration): Geometry {
    if (file.size > applicationConfiguration.getMaxKmlUploadFileSizeLimitInBytes()) {
        throw FileUploadSizeLimitReachedException()
    }
    return parseMultiParFiletopolygons(file.inputStream)
}

private fun parseKmlTopolygons(content: InputStream): Geometry {
    try {
        val kml = Kml.unmarshal(content) ?: throw InvalidKmlException("Failed to parse the kml file")

        return toGeometry(kml.feature)
    } catch (ex: IllegalArgumentException) {
        throw InvalidKmlException(ex.localizedMessage,ex)
    } catch (ex: InvalidGeometryException) {
        throw InvalidKmlException(ex.localizedMessage,ex)
    }
}

所以我可能需要创建一个函数来检测正确的文件,但是我可以在这里返回 Any 类型吗?另外,是否可以从 inputStream 中获取文件的类型?

private fun detectFileType():Any {

}

如果我在这里不是很清楚,我很抱歉,我所需要的只是替换获取 kml 文件函数,以便能够获取 kml 或 geoJson

更新

  //todo would be better to have detection logic separate 
private fun parseKmlTopolygons(file: multipartfile): Geometry {
    val fileExtension: String = FilenameUtils.getExtension(file.originalFilename)
    if (fileExtension == polygonFileType.KML.name) {
        return parseKmlTopolygons(file.inputStream)
    } else if (fileExtension == polygonFileType.GEOJSON.name) {
        return parseKmlTopolygons(file.inputStream)
    }
    throw FormatNotSupportedException("File format is not supported")
}

解决方法

实际上,您所说的“文件类型”是什么意思? geoJson 和 kml 这两种类型都是文本文件。它们没有任何定义类型的幻数编码。所以,我看到以下选项:

  • 使用用户上传的原始文件的扩展名。为此,您可以使用 MultipartFile.getOriginalFilename
  • 上传文件时使用 FE 设置的内容类型。 MultipartFile.getContentType。它很可能无法立即使用,您需要调整前端。
  • 检查实际文件内容。这是最复杂的选项,但由于 kml 是基于 xml 的,而 geoJson 是基于 JSON 的,所以它应该是可行的。
  • 最后是最简单的解决方案:为两种类型创建单独的端点。