14、扩展函数与高阶函数在数据库中的应用：Room的扩展函数、SQLite的DSL封装、数据库迁移的简化。

数据库操作，说白了就是跟数据打交道。在Android开发中，Room已经帮我们封装了很多东西，但有些场景下，你仍然会觉得它不够「顺手」。比如，你写一个复杂的查询，或者做数据库迁移，代码量还是不少。

我个人习惯用扩展函数和高阶函数来「驯服」这些繁琐操作。今天我们就聊聊，怎么用Kotlin的特性，让Room和SQLite用起来更爽。

14.1 Room的扩展函数：让DAO更简洁

Room的DAO接口，通常是用注解来定义SQL查询。但有些操作，比如批量插入、条件更新，写起来挺啰嗦的。我曾在项目中遇到一个场景：需要批量插入几百条数据，如果用@Insert注解，每条数据都得单独调用一次，性能很差。

这时候，扩展函数就派上用场了。

14.1.1 批量插入的扩展

Room的@Insert注解支持传入集合，但如果你需要更灵活的控制，比如插入前先清空旧数据，或者插入后返回主键列表，扩展函数能帮你搞定。

// 定义一个扩展函数，批量插入并返回主键列表
fun <T> RoomDatabase.bulkInsert(
    dao: BaseDao<T>,
    entities: List<T>
): List<Long> {
    return withTransaction {
        entities.map { dao.insert(it) }
    }
}

// 使用示例
val ids = database.bulkInsert(userDao, userList)

嗯，这里要注意：withTransaction是Room提供的高阶函数，它保证所有操作在同一个事务中执行。如果你不用事务，批量插入的性能会差很多。我曾经踩过这个坑，后来才意识到事务的重要性。

14.1.2 条件更新的扩展

Room的@Update注解默认是根据主键更新。但有时候，你想根据某个条件来更新，比如「把所有状态为未同步的数据标记为已同步」。这时候，扩展函数可以帮你封装一个通用的条件更新方法。

// 定义一个扩展函数，根据条件更新
fun <T> RoomDatabase.updateByCondition(
    dao: BaseDao<T>,
    condition: String,
    newValue: T
) {
    withTransaction {
        // 先查询符合条件的记录
        val entities = dao.queryByCondition(condition)
        // 更新每条记录
        entities.forEach { dao.update(it.apply { /* 应用新值 */ }) }
    }
}

你想想看，如果没有这个扩展函数，你每次都得写一遍查询、遍历、更新的逻辑。有了它，一行代码就搞定了。

14.2 SQLite的DSL封装：用代码写SQL

Room虽然好用，但有些复杂查询，比如多表联查、子查询，用注解写起来还是有点别扭。我更喜欢用DSL（领域特定语言）来封装SQLite操作，让查询语句看起来像在写代码。

14.2.1 一个简单的查询DSL

我们可以用高阶函数和扩展函数，构建一个类似这样的DSL：

// 定义一个查询构建器
class QueryBuilder {
    private val conditions = mutableListOf<String>()
    private var orderBy: String? = null
    private var limit: Int? = null

    fun where(condition: String) {
        conditions.add(condition)
    }

    fun orderBy(field: String, ascending: Boolean = true) {
        orderBy = "$field ${if (ascending) "ASC" else "DESC"}"
    }

    fun limit(count: Int) {
        limit = count
    }

    fun build(): String {
        val sb = StringBuilder("SELECT * FROM table")
        if (conditions.isNotEmpty()) {
            sb.append(" WHERE ${conditions.joinToString(" AND ")}")
        }
        orderBy?.let { sb.append(" ORDER BY $it") }
        limit?.let { sb.append(" LIMIT $it") }
        return sb.toString()
    }
}

// 使用DSL构建查询
fun query(block: QueryBuilder.() -> Unit): String {
    val builder = QueryBuilder()
    builder.block()
    return builder.build()
}

// 实际使用
val sql = query {
    where("age > 18")
    where("status = 'active'")
    orderBy("name")
    limit(10)
}

你看，这个DSL让SQL查询变得像在写配置一样清晰。我个人习惯把这种DSL放在一个工具类里，项目里所有复杂查询都用它来生成。这样，SQL语句的拼写错误就少了很多。

14.2.2 结合Room的SQLiteDatabase

Room底层其实还是用的SQLiteDatabase。如果你需要直接操作SQLite，可以用扩展函数来简化：

// 为SQLiteDatabase添加扩展函数
fun SQLiteDatabase.queryWithBuilder(
    table: String,
    block: QueryBuilder.() -> Unit
): Cursor {
    val sql = query(block) // 使用上面定义的DSL
    return rawQuery(sql, null)
}

// 使用示例
val cursor = database.queryWithBuilder("users") {
    where("age > 18")
    orderBy("name")
}

这样做的好处是，你既保留了Room的便利性，又能在需要时直接操作SQLite，灵活性大大提升。

14.3 数据库迁移的简化：用高阶函数管理版本

数据库迁移，是每个Android开发者都会遇到的痛点。Room的迁移是通过Migration类来实现的，每个版本迁移都要写一个类，代码量不小。

我曾在项目中经历过一次大版本升级，数据库从版本1升到版本5，中间有4次迁移。如果每个迁移都写一个类，文件会变得很臃肿。后来我用高阶函数来简化这个过程。

14.3.1 用高阶函数定义迁移

// 定义一个高阶函数，用于创建迁移
fun migration(
    startVersion: Int,
    endVersion: Int,
    migrate: (SupportSQLiteDatabase) -> Unit
): Migration {
    return object : Migration(startVersion, endVersion) {
        override fun migrate(database: SupportSQLiteDatabase) {
            migrate(database)
        }
    }
}

// 使用示例
val migration1to2 = migration(1, 2) { db ->
    db.execSQL("ALTER TABLE users ADD COLUMN age INTEGER NOT NULL DEFAULT 0")
}

val migration2to3 = migration(2, 3) { db ->
    db.execSQL("CREATE TABLE IF NOT EXISTS posts (id INTEGER PRIMARY KEY, title TEXT)")
}

你看，这样每个迁移就是一个函数调用，而不是一个类。代码量减少了，而且逻辑更集中。

14.3.2 批量注册迁移

如果迁移很多，你还可以用扩展函数来批量注册：

// 为RoomDatabase.Builder添加扩展函数
fun RoomDatabase.Builder<*>.addMigrations(vararg migrations: Migration) {
    migrations.forEach { addMigration(it) }
}

// 使用示例
databaseBuilder.addMigrations(
    migration1to2,
    migration2to3,
    migration3to4
)

嗯，这里要注意：迁移的顺序很重要。Room会按照版本号从小到大依次执行迁移。如果你跳过了某个版本，Room会抛出异常。我曾经因为忘记注册中间版本的迁移，导致App崩溃，后来才加上。

14.4 避坑指南：扩展函数与高阶函数的陷阱

虽然扩展函数和高阶函数很好用，但也有一些坑需要注意。

14.5 本章小结

扩展函数和高阶函数，说白了就是Kotlin给我们的两把利器。在数据库操作中，它们能帮我们写出更简洁、更灵活的代码。无论是Room的DAO扩展，还是SQLite的DSL封装，亦或是数据库迁移的简化，这些技巧都能让你的代码质量上一个台阶。

我个人建议，你在项目中多尝试用扩展函数来封装重复操作，用高阶函数来抽象流程控制。你会发现，代码的可读性和可维护性都会大大提升。