Optimize FragmentInflateBindingProperty

Author: DylanCaiCoding

viewbinding-ktx/src/main/java/com/dylanc/viewbinding/Fragment.kt

@@ -18,6 +18,8 @@
 
 package com.dylanc.viewbinding
 
+import android.os.Handler
+import android.os.Looper
 import android.view.LayoutInflater
 import androidx.databinding.ViewDataBinding
 import androidx.fragment.app.Fragment
@@ -27,6 +29,8 @@ import androidx.viewbinding.ViewBinding
 import kotlin.properties.ReadOnlyProperty
 import kotlin.reflect.KProperty
 
+enum class Method { BIND, INFLATE }
+
 inline fun <reified VB : ViewBinding> Fragment.binding() =
   FragmentBindingProperty(VB::class.java)
 
@@ -47,29 +51,23 @@ class FragmentBindingProperty<VB : ViewBinding>(private val clazz: Class<VB>) :
 
 class FragmentInflateBindingProperty<VB : ViewBinding>(private val clazz: Class<VB>) : ReadOnlyProperty<Fragment, VB> {
   private var binding: VB? = null
+  private val handler by lazy { Handler(Looper.getMainLooper()) }
 
-  @Suppress("UNCHECKED_CAST")
   override fun getValue(thisRef: Fragment, property: KProperty<*>): VB {
     if (binding == null) {
       try {
+        @Suppress("UNCHECKED_CAST")
         binding = (clazz.getMethod("inflate", LayoutInflater::class.java).invoke(null, thisRef.layoutInflater) as VB)
           .also { binding -> if (binding is ViewDataBinding) binding.lifecycleOwner = thisRef.viewLifecycleOwner }
       } catch (e: IllegalStateException) {
         throw IllegalStateException("The property of ${property.name} has been destroyed.")
       }
       thisRef.viewLifecycleOwner.lifecycle.addObserver(object : DefaultLifecycleObserver {
         override fun onDestroy(owner: LifecycleOwner) {
-          if (thisRef is BindingLifecycleOwner) thisRef.onDestroyViewBinding(binding!!)
-          binding = null
+          handler.post { binding = null }
         }
       })
     }
     return binding!!
   }
 }
-
-enum class Method { BIND, INFLATE }
-
-interface BindingLifecycleOwner {
-  fun onDestroyViewBinding(destroyingBinding: ViewBinding)
-}

viewbinding-nonreflection-ktx/src/main/java/com/dylanc/viewbinding/nonreflection/Fragment.kt

@@ -18,6 +18,8 @@
 
 package com.dylanc.viewbinding.nonreflection
 
+import android.os.Handler
+import android.os.Looper
 import android.view.LayoutInflater
 import android.view.View
 import androidx.databinding.ViewDataBinding
@@ -47,6 +49,7 @@ class FragmentBindingDelegate<VB : ViewBinding>(private val bind: (View) -> VB)
 
 class FragmentInflateBindingDelegate<VB : ViewBinding>(private val inflate: (LayoutInflater) -> VB) : ReadOnlyProperty<Fragment, VB> {
   private var binding: VB? = null
+  private val handler by lazy { Handler(Looper.getMainLooper()) }
 
   override fun getValue(thisRef: Fragment, property: KProperty<*>): VB {
     if (binding == null) {
@@ -59,15 +62,10 @@ class FragmentInflateBindingDelegate<VB : ViewBinding>(private val inflate: (Lay
       }
       thisRef.viewLifecycleOwner.lifecycle.addObserver(object : DefaultLifecycleObserver {
         override fun onDestroy(owner: LifecycleOwner) {
-          if (thisRef is BindingLifecycleOwner) thisRef.onDestroyViewBinding(binding!!)
-          binding = null
+          handler.post { binding = null }
         }
       })
     }
     return binding!!
   }
 }
-
-interface BindingLifecycleOwner {
-  fun onDestroyViewBinding(destroyingBinding: ViewBinding)
-}