79 lines
1.8 KiB
Kotlin
79 lines
1.8 KiB
Kotlin
package jp.juggler.subwaytooter.api
|
|
|
|
import jp.juggler.subwaytooter.api.entity.*
|
|
import java.util.ArrayList
|
|
import java.util.HashSet
|
|
|
|
class DuplicateMap {
|
|
|
|
private val set_status_id = HashSet<Long>()
|
|
private val set_notification_id = HashSet<Long>()
|
|
private val set_report_id = HashSet<Long>()
|
|
private val set_account_id = HashSet<Long>()
|
|
private val set_status_uri = HashSet<String>()
|
|
|
|
fun clear() {
|
|
set_status_id.clear()
|
|
set_notification_id.clear()
|
|
set_report_id.clear()
|
|
set_account_id.clear()
|
|
set_status_uri.clear()
|
|
}
|
|
|
|
fun isDuplicate(o : TimelineItem) : Boolean {
|
|
|
|
when(o) {
|
|
|
|
is TootStatus ->{
|
|
val uri = o.uri
|
|
val url = o.url
|
|
when {
|
|
uri?.isNotEmpty() == true -> {
|
|
if(set_status_uri.contains(uri)) return true
|
|
set_status_uri.add(uri)
|
|
}
|
|
url?.isNotEmpty() == true -> {
|
|
// URIとURLで同じマップを使いまわすが、害はないと思う…
|
|
if(set_status_uri.contains(url)) return true
|
|
set_status_uri.add(url)
|
|
}
|
|
else -> {
|
|
if(set_status_id.contains(o.id)) return true
|
|
set_status_id.add(o.id)
|
|
}
|
|
}
|
|
}
|
|
|
|
is TootNotification -> {
|
|
if(set_notification_id.contains(o.id)) return true
|
|
set_notification_id.add(o.id)
|
|
|
|
}
|
|
|
|
is TootReport -> {
|
|
if(set_report_id.contains(o.id)) return true
|
|
set_report_id.add(o.id)
|
|
|
|
}
|
|
|
|
is TootAccount -> {
|
|
if(set_account_id.contains(o.id)) return true
|
|
set_account_id.add(o.id)
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
fun filterDuplicate(src : Collection<TimelineItem>?) : ArrayList<TimelineItem> {
|
|
val list_new = ArrayList<TimelineItem>()
|
|
if( src != null ) {
|
|
for(o in src) {
|
|
if(isDuplicate(o)) continue
|
|
list_new.add(o)
|
|
}
|
|
}
|
|
return list_new
|
|
}
|
|
}
|