如何从ViewModel安全访问Activity上下文？规避内存泄漏方案问询

我正在使用DataBinding结合ViewModel开发，ViewModel负责视图数据管理和点击事件处理。但有时需要在ViewModel中访问Activity上下文来更新视图，听说ViewModel持有Activity实例会导致内存泄漏，所以我采用了以下实现方式，想确认是否正确，若有误请提供更优方案。

当前实现代码 #

ViewModel初始版本 #

// 控制抽屉的ViewModel
internal class DrawerModel : ViewModel() {
    // 修改值时视图自动更新
    var userName: String = "Login"
    var userEmail: String = "For better experience"
    
    // 点击登出按钮时触发
    fun onLogOut() {
        FirebaseAuth.getInstance().signOut()
    }
}

规避内存泄漏的修改 #

在XML的<data>中绑定Activity实例：

<data>
    <variable
        name="drawer"
        type="com.ci.evangelian.base_page.PgModels.DrawerModel" /> <!-- ViewModel类 -->
    <variable
        name="baseActivity"
        type="com.ci.evangelian.base_page.BaseActivity" /> <!-- 需要引用的Activity -->
</data>

按钮点击事件传递Activity实例到ViewModel方法：

<Button
    android:onClick="@{() -> drawer.onLogOut(baseActivity)}"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    android:id="@+id/bt_logot" />

更新后的ViewModel方法：

internal class DrawerModel : ViewModel() {
    var userName: String = "Login"
    var userEmail: String = "For better experience"
    
    // 接收Activity参数
    fun onLogOut(baseActivity: BaseActivity) {
        FirebaseAuth.getInstance().signOut()
    }
}

Activity中绑定初始化：

class BaseActivity : AppCompatActivity(){
    internal lateinit var binding: ActivityHomeBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        binding = DataBindingUtil.setContentView(this, R.layout.activity_home)

        val model: PgModels.DrawerModel = PgModels.DrawerModel()
        binding.drawer = model
        binding.baseActivity = this
    }
}

我的逻辑是：ViewModel不保存Activity实例，仅在方法调用时临时使用，由Binding持有Activity和ViewModel实例，这样不会引发内存泄漏。

你的实现存在风险，且不符合ViewModel的设计原则：

1. 虽然ViewModel没有长期持有Activity实例，但Binding持有了Activity的强引用。如果后续在onLogOut方法中加入异步操作（比如网络请求、延迟任务），且该操作捕获了传入的Activity实例，就会导致Activity无法被GC回收，引发内存泄漏。
2. ViewModel的核心职责是处理业务逻辑，应该与视图层（Activity/Fragment）解耦。直接传递Activity实例会让ViewModel依赖具体的视图组件，降低了ViewModel的可测试性和复用性。

更优实现方案 #

方案1：使用LiveData发送事件，Activity响应UI操作 #

ViewModel负责触发业务逻辑，通过LiveData发送事件通知Activity执行UI相关操作，完全解耦ViewModel和Activity。

ViewModel代码

internal class DrawerModel : ViewModel() {
    var userName: String = "Login"
    var userEmail: String = "For better experience"
    
    // 定义私有MutableLiveData存储事件，对外暴露不可变的LiveData
    private val _logoutCompleted = MutableLiveData<Unit>()
    val logoutCompleted: LiveData<Unit> = _logoutCompleted

    fun onLogOut() {
        FirebaseAuth.getInstance().signOut()
        // 发送登出完成事件
        _logoutCompleted.value = Unit
    }
}

XML布局（无需绑定Activity）

<data>
    <variable
        name="drawer"
        type="com.ci.evangelian.base_page.PgModels.DrawerModel" />
</data>

<Button
    android:onClick="@{() -> drawer.onLogOut()}"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    android:id="@+id/bt_logot" />

Activity代码

class BaseActivity : AppCompatActivity(){
    internal lateinit var binding: ActivityHomeBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        binding = DataBindingUtil.setContentView(this, R.layout.activity_home)

        val model: PgModels.DrawerModel = ViewModelProvider(this)[PgModels.DrawerModel::class.java]
        binding.drawer = model
        binding.lifecycleOwner = this // 绑定生命周期，确保LiveData仅在Activity活跃时通知

        // 观察登出完成事件，执行UI操作
        model.logoutCompleted.observe(this) {
            // 这里可以执行需要Activity的操作，比如跳转登录页、更新视图
            startActivity(Intent(this, LoginActivity::class.java))
            finish()
        }
    }
}

方案2：使用Application Context处理非UI依赖场景 #

如果仅需要上下文执行不依赖Activity的操作（比如获取资源、初始化工具类），可以使用AndroidViewModel获取Application Context（Application生命周期与应用一致，不会引发泄漏）。

ViewModel代码

internal class DrawerModel(application: Application) : AndroidViewModel(application) {
    private val appContext = application.applicationContext

    var userName: String = "Login"
    var userEmail: String = "For better experience"

    fun onLogOut() {
        FirebaseAuth.getInstance().signOut()
        // 使用appContext执行操作，比如获取字符串资源
        val tip = appContext.getString(R.string.logout_success)
        // ...
    }
}

Activity初始化ViewModel

val model: PgModels.DrawerModel = ViewModelProvider(this)[PgModels.DrawerModel::class.java]

方案3：接口回调+弱引用（适用于复杂交互场景） #

定义回调接口，Activity实现接口，ViewModel持有接口的弱引用，避免强引用导致泄漏。

定义回调接口

interface LogoutCallback {
    fun onLogoutSuccess()
}

ViewModel代码

internal class DrawerModel : ViewModel() {
    var userName: String = "Login"
    var userEmail: String = "For better experience"
    
    // 使用WeakReference持有回调，避免内存泄漏
    private var logoutCallback: WeakReference<LogoutCallback>? = null

    fun setLogoutCallback(callback: LogoutCallback) {
        logoutCallback = WeakReference(callback)
    }

    fun onLogOut() {
        FirebaseAuth.getInstance().signOut()
        // 通知回调
        logoutCallback?.get()?.onLogoutSuccess()
    }
}

Activity实现接口并绑定

class BaseActivity : AppCompatActivity(), LogoutCallback {
    internal lateinit var binding: ActivityHomeBinding

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        binding = DataBindingUtil.setContentView(this, R.layout.activity_home)

        val model: PgModels.DrawerModel = ViewModelProvider(this)[PgModels.DrawerModel::class.java]
        binding.drawer = model
        model.setLogoutCallback(this)
    }

    override fun onLogoutSuccess() {
        // 执行UI操作
        startActivity(Intent(this, LoginActivity::class.java))
        finish()
    }
}

内容的提问来源于stack exchange，提问作者Nivin Kumar M