添加websocket

app/src/main/java/com/sd/cavphmi/viewmodels/MainVm.kt

+package com.sd.cavphmi.viewmodels
+
+import android.content.Context
+import androidx.lifecycle.viewModelScope
+import com.sd.cavphmi.base.MyBaseViewModel
+import com.sd.cavphmi.bean.SpaceNoBean
+import com.sd.cavphmi.bean.VToXImgBean
+import com.sd.cavphmi.net.MyResult
+import com.sd.cavphmi.repositorys.SpaceNoRepo
+import com.sd.cavphmi.utils.MyContants
+import com.sd.cavphmi.utils.ToastHelper
+import com.sd.cavphmi.websockets.MyWebSocketClient
+import dagger.hilt.android.lifecycle.HiltViewModel
+import dagger.hilt.android.qualifiers.ApplicationContext
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.launch
+import org.java_websocket.client.WebSocketClient
+import java.net.URI
+import javax.inject.Inject
+
+
+@HiltViewModel
+class MainVm @Inject constructor(
+    @ApplicationContext var context: Context,
+    var spaceNoRepo: SpaceNoRepo
+) : MyBaseViewModel() {
+
+
+    private var client: MyWebSocketClient? = null
+
+    fun startWS() {
+        val httpHeaders = mutableMapOf<String, String>()
+        httpHeaders.put("Cookie", "username=nemo")
+
+        client = MyWebSocketClient(URI(MyContants.WS_HOST))
+        client?.connect()
+    }
+
+    fun closeWS() {
+        if (client?.isOpen ?: false) {
+            client?.close()
+        }
+    }
+
+    fun showVToTip() {
+        var imgs = VToXImgBean.imgs
+        ToastHelper.showCustViewShort(context, imgs.get("2")!!)
+    }
+
+    fun getSpaceData() {
+        viewModelScope.launch {
+            var result = spaceNoRepo.getSpaceData()
+            when (result) {
+                is MyResult.Success<SpaceNoBean> -> {
+                    var a = 0
+                }
+                else -> {
+                    var b = 0
+                }
+            }
+        }
+    }
+
+
+}
\ No newline at end of file