Step.07-Solution-Create-a-Worker

Author: Jill Heske

app/src/main/java/com/example/android/devbyteviewer/work/RefreshDataWork.kt

@@ -17,11 +17,26 @@
 
 package com.example.android.devbyteviewer.work
 
-// TODO (01) Create the RefreshDataWorker class, extend it from CoroutineWorker, and
-// pass in a Context and WorkerParams.
+import android.content.Context
+import androidx.work.CoroutineWorker
+import androidx.work.WorkerParameters
+import com.example.android.devbyteviewer.database.getDatabase
+import com.example.android.devbyteviewer.repository.VideosRepository
+import retrofit2.HttpException
 
-// TODO (02) Override the required doWork() method, and create variables for the
-// database and the repository.
-
-// TODO (03) Inside doWork(), in a try-catch block, refresh the videos, and
-// use Payload() to return SUCCESS or RETRY result.
+class RefreshDataWorker(appContext: Context, params: WorkerParameters):
+        CoroutineWorker(appContext, params) {
+    /**
+     * A coroutine-friendly method to do your work.
+     */
+    override suspend fun doWork(): Payload {
+        val database = getDatabase(applicationContext)
+        val repository = VideosRepository(database)
+        return try {
+            repository.refreshVideos()
+            Payload(Result.SUCCESS)
+        } catch (e: HttpException) {
+            Payload(Result.RETRY)
+        }
+    }
+}