github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-25 15:40:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve fuzzy search rate

Browse Source
This commit is contained in:
Him188 2020-08-29 16:07:05 +08:00
parent 2c365edb16
commit 4dc5fe9682
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/mirai-console/src/main/kotlin/net/mamoe/mirai/console/internal/command/internal.kt
View File

@ -14,6 +14,7 @@ import net.mamoe.mirai.console.command.description.CommandArgumentParserExceptio
import net.mamoe.mirai.contact.Group import net.mamoe.mirai.contact.Group
import net.mamoe.mirai.contact.Member import net.mamoe.mirai.contact.Member
import kotlin.math.max import kotlin.math.max
import kotlin.math.min
internal infix fun Array<String>.matchesBeginning(list: List<Any>): Boolean { internal infix fun Array<String>.matchesBeginning(list: List<Any>): Boolean {
@ -37,12 +38,16 @@ internal fun String.fuzzyMatchWith(target: String): Double {
} }
var match = 0 var match = 0
for (i in 0..(max(this.lastIndex, target.lastIndex))) { for (i in 0..(max(this.lastIndex, target.lastIndex))) {
val t = target.getOrNull(match) val t = target.getOrNull(match) ?: break
if (t == this.getOrNull(i) && t != null) { if (t == this.getOrNull(i)) {
match++ match++
} }
} }
return match.toDouble() / (max(this.lastIndex, target.lastIndex) + 1)
val longerLength = max(this.length, target.length)
val shorterLength = min(this.length, target.length)
return match.toDouble() / (longerLength + (shorterLength - match))
} }
internal inline fun <T : Any> Collection<T>.fuzzySearch( internal inline fun <T : Any> Collection<T>.fuzzySearch(