import 'dart:async'; import 'dart:convert'; import 'package:flutter/material.dart'; import 'package:hive_flutter/hive_flutter.dart'; import 'package:together_mobile/database/box_type.dart'; import 'package:together_mobile/models/apply_list_model.dart'; import 'package:together_mobile/models/contact_model.dart'; import 'package:together_mobile/models/init_get_it.dart'; import 'package:web_socket_channel/web_socket_channel.dart'; import 'package:web_socket_channel/status.dart' as status; class WebSocketManager extends ChangeNotifier { late Uri wsUrl; late WebSocketChannel channel; String id = ''; SocketStatus? socketStatus; Timer? heartBeatTimer; Timer? serverTimer; Timer? reconnectTimer; int reconnectCount = 30; int reconnectTimes = 0; Duration timeout = const Duration(seconds: 4); void connect(String userId) { id = userId; wsUrl = Uri.parse('ws://10.0.2.2:8000/ws/$id'); channel = WebSocketChannel.connect(wsUrl); socketStatus = SocketStatus.connected; print('websocket connected <$channel>'); heartBeatInspect(); if (reconnectTimer != null) { reconnectTimer!.cancel(); reconnectTimer = null; reconnectTimes = 0; } channel.stream.listen(onData, onError: onError, onDone: onDone); } void onData(jsonData) { heartBeatInspect(); Map data = json.decode(jsonData); switch (data['event']) { case 'one-to-one-chat': receiveFriendMsg(data); case 'apply-friend': receiveApplyFriend(data); case 'friend-added': receiveFriendAdded(data); case 'friend-deleted': receiveFriendDeleted(data); } } void onDone() { print('websocket disconnected <$channel>'); reconnect(); } void onError(Object error) {} void heartBeatInspect() { if (heartBeatTimer != null) { heartBeatTimer!.cancel(); heartBeatTimer = null; } if (serverTimer != null) { serverTimer!.cancel(); serverTimer = null; } print('start heartbeat inspect......'); heartBeatTimer = Timer(timeout, () { channel.sink.add(json.encode({'event': 'ping'})); serverTimer = Timer(timeout, () { channel.sink.close(status.internalServerError); socketStatus = SocketStatus.closed; }); }); } void reconnect() { if (socketStatus == SocketStatus.error) { if (heartBeatTimer != null) { heartBeatTimer!.cancel(); heartBeatTimer = null; } if (serverTimer != null) { serverTimer!.cancel(); serverTimer = null; } return; } print('websocket reconnecting......'); if (heartBeatTimer != null) { heartBeatTimer!.cancel(); heartBeatTimer = null; } if (serverTimer != null) { serverTimer!.cancel(); serverTimer = null; } reconnectTimer = Timer.periodic(timeout, (timer) { if (reconnectTimes < reconnectCount) { connect(id); reconnectTimes++; } else { print('reconnection times exceed the max times......'); // If it is still disconnection after reconnect 30 times, set the socket // status to error, means the network is bad, and stop reconnecting. socketStatus = SocketStatus.error; channel.sink.close(); timer.cancel(); } }); } } enum SocketStatus { connected, closed, error, } void receiveFriendMsg(Map msg) async { String senderId = msg['senderId'] as String; late Box messageTBox; try { messageTBox = Hive.box('message_$senderId'); } catch (e) { messageTBox = await Hive.openBox('message_$senderId'); } Box chatSettingBox = Hive.box('chat_setting'); ChatSetting? chatSetting = chatSettingBox.get(senderId); DateTime dateTime = DateTime.parse(msg['dateTime'] as String); if (chatSetting == null) { chatSettingBox.put( senderId, ChatSetting( senderId, false, true, false, dateTime, ), ); } else { chatSetting.isOpen = true; chatSetting.latestDateTime = dateTime; chatSettingBox.put(senderId, chatSetting); } List attachments = List.from(msg['attachments']); messageTBox.add( MessageT( senderId, msg['text'], msg['type'], DateTime.parse(msg['dateTime'] as String), msg['isShowTime'], attachments, ), ); } void receiveApplyFriend(Map msg) { getIt.get().addJson(msg); } void receiveFriendAdded(Map msg) { print('=================收到了申请好友通过事件=================='); print(msg); print('======================================='); getIt.get().addFriend(msg['friendId'], msg['setting']); getIt .get() .addAccountProfile(msg['friendId'], msg['accountProfile']); } void receiveFriendDeleted(Map msg) { print('=================收到了解除好友事件=================='); print(msg); print('======================================='); getIt.get().removeFriend(msg['friendId']); getIt.get().removeFriend(msg['friendId']); }