add TrafficMonitor to the main activity

src/main/scala/com/github/shadowsocks/Shadowsocks.scala

@@ -40,7 +40,7 @@ package com.github.shadowsocks
 
 import java.io.{FileOutputStream, IOException, InputStream, OutputStream}
 import java.util
-import java.util.Locale
+import java.util.{Locale, Timer, TimerTask}
 
 import android.app.backup.BackupManager
 import android.app.{Activity, ProgressDialog}
@@ -147,7 +147,7 @@ class Shadowsocks
   extends AppCompatActivity {
 
   // Variables
-  private var serviceStarted = false
+  var serviceStarted = false
   var fab: FloatingActionButton = _
   var fabProgressCircle: FABProgressCircle = _
   var progressDialog: ProgressDialog = _
@@ -156,6 +156,8 @@ class Shadowsocks
   var prepared = false
   var currentProfile = new Profile
   var vpnEnabled = -1
+  var timer: Timer = null
+  val TIMER_INTERVAL = 1 // sec
 
   // Services
   var currentServiceName = classOf[ShadowsocksNatService].getName
@@ -465,6 +467,24 @@ class Shadowsocks
 
     // Check if current profile changed
     if (ShadowsocksApplication.profileId != currentProfile.id) reloadProfile()
+
+    // initialize timer
+    val task = new TimerTask {
+      def run() {
+        TrafficMonitor.update()
+        val pm = getSystemService(Context.POWER_SERVICE).asInstanceOf[PowerManager]
+        if (pm.isScreenOn) {
+          val trafficStat = getString(R.string.stat_summary).formatLocal(Locale.ENGLISH,
+            TrafficMonitor.getTxRate, TrafficMonitor.getRxRate,
+            TrafficMonitor.getTxTotal, TrafficMonitor.getRxTotal)
+            handler.post(() => {
+              preferences.findPreference(Key.stat).setSummary(trafficStat)
+            })
+        }
+      }
+    }
+    timer = new Timer(true)
+    timer.schedule(task, TIMER_INTERVAL * 1000, TIMER_INTERVAL * 1000)
   }
 
   private def setPreferenceEnabled(enabled: Boolean) {
@@ -510,6 +530,12 @@ class Shadowsocks
   override def onStop() {
     super.onStop()
     clearDialog()
+
+    // reset timer
+    if (timer != null) {
+      timer.cancel()
+      timer = null
+    }
   }
 
   override def onDestroy() {

src/main/scala/com/github/shadowsocks/ShadowsocksNatService.scala

@@ -420,6 +420,8 @@ class ShadowsocksNatService extends Service with BaseService {
 
   override def startRunner(c: Config) {
 
+    TrafficMonitor.reset()
+
     config = c
 
     // register close receiver
@@ -509,6 +511,8 @@ class ShadowsocksNatService extends Service with BaseService {
 
   override def stopRunner() {
 
+    TrafficMonitor.reset()
+
     // channge the state
     changeState(State.STOPPING)
 

src/main/scala/com/github/shadowsocks/ShadowsocksVpnService.scala

@@ -144,6 +144,8 @@ class ShadowsocksVpnService extends VpnService with BaseService {
 
   override def stopRunner() {
 
+    TrafficMonitor.reset()
+
     if (vpnThread != null) {
       vpnThread.stopThread()
       vpnThread = null
@@ -217,6 +219,8 @@ class ShadowsocksVpnService extends VpnService with BaseService {
 
   override def startRunner(c: Config) {
 
+    TrafficMonitor.reset()
+
     vpnThread = new ShadowsocksVpnThread(this)
     vpnThread.start()
 

src/main/scala/com/github/shadowsocks/utils/Constants.scala

@@ -68,6 +68,7 @@ object Key {
   val profiles = "profiles"
   val isNAT = "isNAT"
   val route = "route"
+  val stat = "stat"
 
   val isRunning = "isRunning"
   val isAutoConnect = "isAutoConnect"

src/main/scala/com/github/shadowsocks/utils/TrafficMonitor.scala

@@ -38,16 +38,16 @@ object TrafficMonitor {
     }
   }
 
-  def updateTraffic() {
+  def update() {
     val now = getTraffic
-    txRate = ((now.tx - last.tx) / 1024 / (now.timestamp - last.timestamp))
-    rxRate = ((now.rx - last.rx) / 1024 / (now.timestamp - last.timestamp))
+    txRate = (now.tx - last.tx) / (now.timestamp - last.timestamp)
+    rxRate = (now.rx - last.rx) / (now.timestamp - last.timestamp)
     txTotal += (now.tx - last.tx) / 1024
     rxTotal += (now.rx - last.rx) / 1024
     last = now
   }
 
-  def resetTraffic() {
+  def reset() {
     txRate = 0
     rxRate = 0
     txTotal = 0