Update WorkManager and add InputData & Custom Configuration. (android#680)

Co-authored-by: Caren <[email protected]>

Author: pfmaggi

app/src/main/AndroidManifest.xml

@@ -16,6 +16,7 @@
   -->
 
 <manifest xmlns:android="http://schemas.android.com/apk/res/android"
+    xmlns:tools="http://schemas.android.com/tools"
     package="com.google.samples.apps.sunflower">
 
     <uses-permission android:name="android.permission.INTERNET" />
@@ -38,6 +39,11 @@
                 <category android:name="android.intent.category.LAUNCHER" />
             </intent-filter>
         </activity>
+
+        <provider
+            android:name="androidx.work.impl.WorkManagerInitializer"
+            android:authorities="${applicationId}.workmanager-init"
+            tools:node="remove" />
     </application>
 
 </manifest>

app/src/main/java/com/google/samples/apps/sunflower/MainApplication.kt

@@ -17,7 +17,13 @@
 package com.google.samples.apps.sunflower
 
 import android.app.Application
+import androidx.work.Configuration
 import dagger.hilt.android.HiltAndroidApp
 
 @HiltAndroidApp
-class MainApplication : Application()
+class MainApplication : Application(), Configuration.Provider {
+    override fun getWorkManagerConfiguration(): Configuration =
+                Configuration.Builder()
+                        .setMinimumLoggingLevel(if (BuildConfig.DEBUG) android.util.Log.DEBUG else android.util.Log.ERROR)
+                        .build()
+}

app/src/main/java/com/google/samples/apps/sunflower/data/AppDatabase.kt

@@ -24,8 +24,11 @@ import androidx.room.TypeConverters
 import androidx.sqlite.db.SupportSQLiteDatabase
 import androidx.work.OneTimeWorkRequestBuilder
 import androidx.work.WorkManager
+import androidx.work.workDataOf
 import com.google.samples.apps.sunflower.utilities.DATABASE_NAME
+import com.google.samples.apps.sunflower.utilities.PLANT_DATA_FILENAME
 import com.google.samples.apps.sunflower.workers.SeedDatabaseWorker
+import com.google.samples.apps.sunflower.workers.SeedDatabaseWorker.Companion.KEY_FILENAME
 
 /**
  * The Room database for this app
@@ -55,7 +58,9 @@ abstract class AppDatabase : RoomDatabase() {
                     object : RoomDatabase.Callback() {
                         override fun onCreate(db: SupportSQLiteDatabase) {
                             super.onCreate(db)
-                            val request = OneTimeWorkRequestBuilder<SeedDatabaseWorker>().build()
+                            val request = OneTimeWorkRequestBuilder<SeedDatabaseWorker>()
+                                    .setInputData(workDataOf(KEY_FILENAME to PLANT_DATA_FILENAME))
+                                    .build()
                             WorkManager.getInstance(context).enqueue(request)
                         }
                     }

app/src/main/java/com/google/samples/apps/sunflower/workers/SeedDatabaseWorker.kt

@@ -25,25 +25,31 @@ import com.google.gson.reflect.TypeToken
 import com.google.gson.stream.JsonReader
 import com.google.samples.apps.sunflower.data.AppDatabase
 import com.google.samples.apps.sunflower.data.Plant
-import com.google.samples.apps.sunflower.utilities.PLANT_DATA_FILENAME
-import kotlinx.coroutines.coroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 
 class SeedDatabaseWorker(
-    context: Context,
-    workerParams: WorkerParameters
+        context: Context,
+        workerParams: WorkerParameters
 ) : CoroutineWorker(context, workerParams) {
-    override suspend fun doWork(): Result = coroutineScope {
+    override suspend fun doWork(): Result = withContext(Dispatchers.IO) {
         try {
-            applicationContext.assets.open(PLANT_DATA_FILENAME).use { inputStream ->
-                JsonReader(inputStream.reader()).use { jsonReader ->
-                    val plantType = object : TypeToken<List<Plant>>() {}.type
-                    val plantList: List<Plant> = Gson().fromJson(jsonReader, plantType)
+            val filename = inputData.getString(KEY_FILENAME)
+            if (filename != null) {
+                applicationContext.assets.open(filename).use { inputStream ->
+                    JsonReader(inputStream.reader()).use { jsonReader ->
+                        val plantType = object : TypeToken<List<Plant>>() {}.type
+                        val plantList: List<Plant> = Gson().fromJson(jsonReader, plantType)
 
-                    val database = AppDatabase.getInstance(applicationContext)
-                    database.plantDao().insertAll(plantList)
+                        val database = AppDatabase.getInstance(applicationContext)
+                        database.plantDao().insertAll(plantList)
 
-                    Result.success()
+                        Result.success()
+                    }
                 }
+            } else {
+                Log.e(TAG, "Error seeding database - no valid filename")
+                Result.failure()
             }
         } catch (ex: Exception) {
             Log.e(TAG, "Error seeding database", ex)
@@ -53,5 +59,6 @@ class SeedDatabaseWorker(
 
     companion object {
         private const val TAG = "SeedDatabaseWorker"
+        const val KEY_FILENAME = "PLANT_DATA_FILENAME"
     }
 }

build.gradle

@@ -50,7 +50,7 @@ buildscript {
         testExtJunit = '1.1.0'
         uiAutomatorVersion = '2.2.0'
         viewPagerVersion = '1.0.0'
-        workVersion = '2.1.0'
+        workVersion = '2.4.0'
     }
 
     repositories {