1 files changed, 8 insertions, 17 deletions

diff --git a/app/src/main/java/org/pacien/tincapp/activities/status/nodes/NodeListLiveData.kt b/app/src/main/java/org/pacien/tincapp/activities/status/nodes/NodeListLiveData.kt
index cdbdf0a..70ea54e 100644
--- a/app/src/main/java/org/pacien/tincapp/activities/status/nodes/NodeListLiveData.kt
+++ b/app/src/main/java/org/pacien/tincapp/activities/status/nodes/NodeListLiveData.kt
@@ -18,30 +18,21 @@
 
 package org.pacien.tincapp.activities.status.nodes
 
-import android.arch.lifecycle.LiveData
+import org.pacien.tincapp.activities.common.SelfRefreshingLiveData
 import org.pacien.tincapp.commands.Tinc
-import java.util.*
-import kotlin.concurrent.timer
+import java.util.concurrent.TimeUnit
 
 /**
  * @author pacien
  */
-class NodeListLiveData(private val netName: String) : LiveData<List<String>>() {
-  private val updateInterval = 2 * 1000L // in milliseconds
+class NodeListLiveData(private val netName: String) : SelfRefreshingLiveData<List<String>>(1, TimeUnit.SECONDS) {
   private val tincCtl = Tinc
-  private lateinit var updateTimer: Timer
 
-  override fun onActive() {
-    updateTimer = timer(period = updateInterval, action = { updateNodeList() })
-  }
-
-  override fun onInactive() {
-    updateTimer.apply { cancel() }.apply { purge() }
-  }
+  override fun onRefresh() {
+    val nodeList = tincCtl.dumpNodes(netName)
+      .thenApply { list -> list.map { it.substringBefore(' ') } }
+      .get()
 
-  private fun updateNodeList() {
-    tincCtl.dumpNodes(netName)
-      .thenApply { list -> list.map { it.substringBefore(' ')} }
-      .thenAccept(this::postValue)
+    postValue(nodeList)
   }
 }