From 58003515cae7b870f80deb0c36a024e2259a4273 Mon Sep 17 00:00:00 2001
From: Jeff Sharkey <jsharkey@android.com>
Date: Thu, 14 May 2015 21:08:48 -0700
Subject: [PATCH] Initial pass at storage benchmarks.

After adopting a disk, run benchmarks to compare its performance
against internal storage.

Bug: 21172095
Change-Id: I7c9a5a89e37f7156677fc76bd1ba193eead6665e
---
 .../StorageWizardFormatProgress.java           | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/com/android/settings/deviceinfo/StorageWizardFormatProgress.java b/src/com/android/settings/deviceinfo/StorageWizardFormatProgress.java
index 9ab714a7158..e2e6dbe8929 100644
--- a/src/com/android/settings/deviceinfo/StorageWizardFormatProgress.java
+++ b/src/com/android/settings/deviceinfo/StorageWizardFormatProgress.java
@@ -55,12 +55,23 @@ public class StorageWizardFormatProgress extends StorageWizardBase {
         new PartitionTask().execute();
     }
 
-    public class PartitionTask extends AsyncTask<Void, Void, Exception> {
+    public class PartitionTask extends AsyncTask<Void, Integer, Exception> {
         @Override
         protected Exception doInBackground(Void... params) {
             try {
                 if (mFormatPrivate) {
                     mStorage.partitionPrivate(mDisk.getId());
+                    publishProgress(40);
+
+                    final long internalBench = mStorage.benchmark(null);
+                    publishProgress(60);
+
+                    final VolumeInfo privateVol = findFirstVolume(VolumeInfo.TYPE_PRIVATE);
+                    final long privateBench = mStorage.benchmark(privateVol.id);
+
+                    // TODO: plumb through to user when below threshold
+                    final float pct = (float) internalBench / (float) privateBench;
+                    Log.d(TAG, "New volume is " + pct + "x the speed of internal");
                 } else {
                     mStorage.partitionPublic(mDisk.getId());
                 }
@@ -70,6 +81,11 @@ public class StorageWizardFormatProgress extends StorageWizardBase {
             }
         }
 
+        @Override
+        protected void onProgressUpdate(Integer... progress) {
+            setCurrentProgress(progress[0]);
+        }
+
         @Override
         protected void onPostExecute(Exception e) {
             final Context context = StorageWizardFormatProgress.this;