Refactor get/setCurrentInterval to operator fun

Author: Rothes

bukkit/version/remapped/src/main/kotlin/io/github/rothes/esu/bukkit/module/networkthrottle/EntityUpdateInterval.kt

@@ -38,19 +38,19 @@ abstract class EntityUpdateInterval: CommonFeature<EntityUpdateInterval.FeatureC
             @Command("esu networkThrottle entityUpdateInterval entityType <entityType>")
             @ShortPerm
             fun getUpdateInterval(sender: User, entityType: EntityType<*>) {
-                val interval = getCurrentInterval(entityType)
+                val interval = this@EntityUpdateInterval[entityType]
                 sender.miniMessage("<pc>Current update interval of entity type <pdc>${entityType.toShortString()}</pdc> is <pdc>$interval")
             }
         })
     }
 
-    abstract fun getCurrentInterval(entityType: EntityType<*>): Int
-    abstract fun setCurrentInterval(entityType: EntityType<*>, interval: Int)
+    abstract operator fun get(entityType: EntityType<*>): Int
+    abstract operator fun set(entityType: EntityType<*>, interval: Int)
 
     protected fun applyUpdateInterval() {
         val config = config
         for ((type, interval) in config.entityTypeUpdateInterval) {
-            setCurrentInterval(type, interval)
+            this[type] = interval
         }
     }
 
@@ -60,8 +60,8 @@ abstract class EntityUpdateInterval: CommonFeature<EntityUpdateInterval.FeatureC
             Higher means less entity movement packets (more de-sync), but less network as deal.
         """)
         val entityTypeUpdateInterval: Map<EntityType<*>, Int> = mapOf(
-            EntityType.PIG to INSTANCE.getCurrentInterval(EntityType.PIG),
-            EntityType.ZOMBIE to INSTANCE.getCurrentInterval(EntityType.ZOMBIE),
+            EntityType.PIG to INSTANCE[EntityType.PIG],
+            EntityType.ZOMBIE to INSTANCE[EntityType.ZOMBIE],
         )
     ): BaseFeatureConfiguration()
 

bukkit/version/v1_17_1/src/main/kotlin/io/github/rothes/esu/bukkit/module/networkthrottle/v17_1/EntityUpdateIntervalImpl.kt

@@ -12,11 +12,11 @@ object EntityUpdateIntervalImpl: EntityUpdateInterval() {
 
     private val ENTITY_TYPE_UPDATE_INTERVAL = EntityType::class.java.getDeclaredField("updateInterval").usIntAccessor
 
-    override fun getCurrentInterval(entityType: EntityType<*>): Int {
+    override operator fun get(entityType: EntityType<*>): Int {
         return ENTITY_TYPE_UPDATE_INTERVAL[entityType]
     }
 
-    override fun setCurrentInterval(entityType: EntityType<*>, interval: Int) {
+    override operator fun set(entityType: EntityType<*>, interval: Int) {
         ENTITY_TYPE_UPDATE_INTERVAL[entityType] = interval
     }
 
@@ -28,7 +28,7 @@ object EntityUpdateIntervalImpl: EntityUpdateInterval() {
         override fun handleEntity(entity: Entity): Boolean {
             val tracker = entity.tracker ?: return false
             val se = TRACKED_ENTITY_SERVER_ENTITY[tracker] as ServerEntity
-            SERVER_ENTITY_UPDATE_INTERVAL[se] = getCurrentInterval(entity.type)
+            SERVER_ENTITY_UPDATE_INTERVAL[se] = this@EntityUpdateIntervalImpl[entity.type]
             return true
         }
 

bukkit/version/v1_21/src/main/kotlin/io/github/rothes/esu/bukkit/module/networkthrottle/v21__paper/ApplyCommandImpl.kt

@@ -12,7 +12,7 @@ object ApplyCommandImpl : EntityUpdateInterval.ApplyCommand() {
     override fun handleEntity(entity: Entity): Boolean {
         val tracker = entity.`moonrise$getTrackedEntity`() ?: return false
         val se = tracker.serverEntity
-        SERVER_ENTITY_UPDATE_INTERVAL[se] = EntityUpdateInterval.INSTANCE.getCurrentInterval(entity.type)
+        SERVER_ENTITY_UPDATE_INTERVAL[se] = EntityUpdateInterval.INSTANCE[entity.type]
         return true
     }