Android 应用开发 | 一种限制拷贝速率解决因 IO 过高导致系统卡顿的方法

文章目录

一、问题背景
二、代码实现

一、问题背景

经常做 Android 应用的小伙伴应该会有经验，就是如果应用在写入文件的时候，即使写文件的动作是在子线程，也会出现 UI 上的卡顿，这是因为文件的 IO 是由内核去完成的，此时 CPU 会从 用户态 切换到 内核态，应用将无法获得 CPU 时间片。如果 IO 较少的时候，这个切换不频繁，应用能有足够的时间片去执行绘制。但如果 IO 频繁，例如高速写入的场景下，此时 IO 占用的时间片增加，而我们知道 CPU 资源是有限的，此时会导致应用获得 时间片 将减少，导致 UI 绘制线程出现卡顿，进而影响 UI 上的卡顿。

我们可以使用 top 命令，查看 IO 占用 CPU 的时间片（即 iow）：

因此，如果不能从其它方式去解决此问题（内核层的 IO 调度和 CPU 调度机制），可以运用简单的方式：在写入文件的时候，限制写入速度，减缓 IO 占用率。

二、代码实现

例如，一个标准的拷贝文件的 Kotlin 方法如下：

kotlin 复制代码

originFile.inputStream().buffered().use { inputStream ->
    targetFile.outputStream().buffered().use { outputStream ->
        val buffer = ByteArray(1024)
        var bytesRead: Int
        while (inputStream.read(buffer).also { bytesRead = it } != -1) {
            outputStream.write(buffer, 0, bytesRead)
        }
        outputStream.flush()
    }
}

这个方法对 源文件 开启了 InputStream，并转换为 BufferInputStream，而对 目标文件 开启了 OutputStream，并转换为 BufferOutputStream，然后再从 InputStream 读取流，输出到 OutputStream，完成文件的读写。

而我们为了实现限制其写入速度，我们可以定义一个速率，例如每秒写入 20M，即每秒写入的字节数为 20 * 1024 * 1024 = 20, 971, 520 。在每次写入量达到 20M 的时候，检查一次写入这 20M 使用的时长，如果这个时长 < 1s ，则表示在这 1s 的时间里，写入的速率已经超过了 20M，需要等待一定时长，使其满足 1s 内只写入 20M 的需求。代码如下：

kotlin 复制代码

// 限制拷贝时的速率, 每秒最多拷贝的字节数
val MAX_COPY_SPEED = 20 * 1024 * 1024L

// 记录已拷贝的文件大小
var copiedSize = 0L

// 记录上次检查性能的时间 (用来计算拷贝速率 限制拷贝速率)
var lastCheckPerformanceTime = System.currentTimeMillis()
// 记录上次检查时 已拷贝的数据量
var lastCheckPerformanceSize = 0L

originFile.inputStream().buffered(1024 * 1024).use { inputStream ->
    targetFile.outputStream().buffered().use { outputStream ->
        val buffer = ByteArray(DEFAULT_BUFFER_SIZE)
        var bytesRead: Int
        while (inputStream.read(buffer).also { bytesRead = it } != -1) {
            outputStream.write(buffer, 0, bytesRead)

            // 增加已拷贝文件的大小
            copiedSize += bytesRead

            // 计算拷贝速率 每满 MAX_COPY_SPEED 计算一次速率
            if (copiedSize - lastCheckPerformanceSize >= MAX_COPY_SPEED) {
                lastCheckPerformanceSize = copiedSize
                val time = System.currentTimeMillis()
                // 计算写入 MAX_COPY_SPEED 字节数所使用的时间
                val deltaTime = time - lastCheckPerformanceTime
                lastCheckPerformanceTime = time
                // 如果速率大于 MAX_COPY_SPEED 则暂停 等待下次计算
                if (deltaTime < 1000) {
                	// 延迟 1000 - deltaTime 使满足 1s 内只写入指定量
                    delay(1000 - deltaTime)
                }
            }
        }
        outputStream.flush()
    }
}