Refactor DataLayer method to use coroutines rather than callback. (#777)

* Add missing snippet and add more code for formatting

* Adds coroutines for Data Layer sample

* fix format

Author: kul3r4

gradle/libs.versions.toml

@@ -236,6 +236,7 @@ kotlin-stdlib = { module = "org.jetbrains.kotlin:kotlin-stdlib", version.ref = "
 kotlin-test = { module = "org.jetbrains.kotlin:kotlin-test", version.ref = "kotlin" }
 kotlinx-coroutines-android = { module = "org.jetbrains.kotlinx:kotlinx-coroutines-android", version.ref = "coroutines" }
 kotlinx-coroutines-guava = { module = "org.jetbrains.kotlinx:kotlinx-coroutines-guava", version.ref = "kotlinxCoroutinesGuava" }
+kotlinx-coroutines-play-services = { module = "org.jetbrains.kotlinx:kotlinx-coroutines-play-services", version.ref = "coroutines" }
 kotlinx-coroutines-test = { module = "org.jetbrains.kotlinx:kotlinx-coroutines-test", version.ref = "coroutines" }
 kotlinx-metadata-jvm = { module = "org.jetbrains.kotlin:kotlin-metadata-jvm", version.ref = "kotlin" }
 kotlinx-serialization-json = { module = "org.jetbrains.kotlinx:kotlinx-serialization-json", version.ref = "kotlinxSerializationJson" }

wear/build.gradle.kts

@@ -75,6 +75,7 @@ dependencies {
 
     implementation(libs.compose.ui.tooling)
     implementation(libs.play.services.wearable)
+    implementation(libs.kotlinx.coroutines.play.services)
     implementation(libs.androidx.tiles)
     implementation(libs.androidx.wear)
     implementation(libs.androidx.wear.ongoing)

wear/src/main/java/com/example/wear/snippets/datalayer/DataLayerActivity.kt

@@ -35,6 +35,7 @@ import com.google.android.gms.wearable.Wearable
 import java.io.ByteArrayOutputStream
 import java.io.InputStream
 import java.util.concurrent.ExecutionException
+import kotlinx.coroutines.tasks.await
 
 class DataLayerActivity : ComponentActivity(), DataClient.OnDataChangedListener {
     private val dataClient by lazy { Wearable.getDataClient(this) }
@@ -170,24 +171,24 @@ class DataLayerActivity2 : ComponentActivity(), DataClient.OnDataChangedListener
 }
 
 // [START android_wear_datalayer_async_call]
-private fun Context.sendDataAsync(count: Int) {
-    // Create a data item with the path and data to be sent
-    val putDataReq: PutDataRequest = PutDataMapRequest.create("/count").run {
-        dataMap.putInt("count_key", count)
-        asPutDataRequest()
+private suspend fun Context.sendDataAsync(count: Int) {
+    try {
+        val putDataReq: PutDataRequest = PutDataMapRequest.create("/count").run {
+            dataMap.putInt("count_key", count)
+            asPutDataRequest()
+        }
+        val dataItem = Wearable.getDataClient(this).putDataItem(putDataReq).await()
+        handleDataItem(dataItem)
+    } catch (e: Exception) {
+        handleDataItemError(e)
+    } finally {
+        handleTaskComplete()
     }
-    // Create a task to send the data to the data layer
-    val task: Task<DataItem> = Wearable.getDataClient(this).putDataItem(putDataReq)
-
-    // Using Kotlin function references
-    task.addOnSuccessListener(::handleDataItem)
-    task.addOnFailureListener(::handleDataItemError)
-    task.addOnCompleteListener(::handleTaskComplete)
 }
 
 private fun handleDataItem(dataItem: DataItem) { }
 private fun handleDataItemError(exception: Exception) { }
-private fun handleTaskComplete(task: Task<DataItem>) { }
+private fun handleTaskComplete() { }
 // [END android_wear_datalayer_async_call]
 
 // [START android_wear_datalayer_sync_call]