Question

在C ++中，我可以使用朋友operator>>重载从文件中加载类的集合：

friend std::istream& operator>>(std::istream& is_a, College& college_a) {
    return is_a >> college_a.id_,
        is_a.get(),
        std::getline(is_a,college_a.name_);
}

std::set<College> colleges {};
std::copy(std::istream_iterator<College> {std::cin},
    std::istream_iterator<College> {},
    std::inserter(colleges, colleges.begin()));

来自这样的文件：

0707 Rowan Technical College
0980 University of Saskatchewan
1058 Belmont University
1072 Belmont Technical College

用Kotlin怎么办？

Answer 1

在Kotlin / Java中，有很多方法可以解析文件并拆分行。下一个示例使用正则表达式将行分为id和大学名称：

data class College(val id: String, val name: String)

val colleges = File("input.txt").useLines { lines ->
    val regex = Regex("(\\d+) (.+)")
    lines.mapNotNull { line ->
        regex.matchEntire(line)?.let {
            College(it.groupValues[1], it.groupValues[2])
        }
    }.toList()
}

如果您需要阅读标准输入，则将第三行替换为：

val colleges = InputStreamReader(System.`in`).useLines { lines ->

更新：隐藏详细信息

在Java / Kotlin中，没有将文本文件解析为对象的通用方法。对于XML和Json文件，存在一些标准API，您可以为每种类型注册Serializer / Deserializer。

如果您需要对面向行的简单文本文件进行抽象，则必须自己构建。下面的示例可能会给您一个想法：

fun <T : Any> File.parseLines(lineParser: (String) -> T?): List<T> =
    useLines { it.mapNotNull(lineParser).toList() }

该函数使用File方法扩展parseLines类。

类College的行解析器的实现如下：

val collegeLineParser: (String) -> College? = { line ->
    val regex = Regex("(\\d+) (.+)")
    regex.matchEntire(line)?.let {
        College(it.groupValues[1], it.groupValues[2])
    }
}

或者如果您想缓存RegEx：

val collegeLineParserCachedRegex = object : (String) -> College? {
    val regex = Regex("(\\d+) (.+)")
    override fun invoke(line: String): College? =
        regex.matchEntire(line)?.let {
            College(it.groupValues[1], it.groupValues[2])
        }
}

现在您可以在parseLines上调用方法File：

val colleges = File("input.txt").parseLines(collegeLineParser)

更新：解析器注册表

如果您不想为每次通话都提供lineParser，则可以创建一个注册表：

object LineParserRegistry {
    val parsers = ConcurrentHashMap<KClass<*>, (String) -> Any?>()

    inline fun <reified T> register(noinline parser : (String) -> T?) {
        parsers[T::class] = parser
    }

    inline fun <reified T> get(): (String) -> T? {
        // force companion initializer
        Class.forName(T::class.java.name)
        return parsers[T::class] as (String) -> T??
    }
}

inline fun <reified T : Any> File.parseLines(): List<T> =
    useLines { it.mapNotNull(LineParserRegistry.get<T>()).toList() }

如果您想自动注册解析器，则需要使用init方法添加伴随对象：

data class College(val id: String, val name: String) {
    companion object {
        init {
            val collegeLineParser: (String) -> College? = { line ->
                val regex = Regex("(\\d+) (.+)")
                regex.matchEntire(line)?.let {
                    College(it.groupValues[1], it.groupValues[2])
                }
            }
            LineParserRegistry.register(collegeLineParser)
        }
    }
}

此init方法在第一次使用College类时被调用。但是反射并不重要，因此我们不得不在注册表的Class.forName(T::class.java.name)方法中添加get。此调用将强制初始化伴随对象。

现在，您无需进行任何准备就可以调用解析器：

val colleges = File("input.txt").parseLines<College>()

Kotlin是否从文件加载类的集合？

1 个答案: