TF-1360 Fix can not add data to stream controller when re-login

This commit is contained in:
dab246
2023-01-09 14:11:19 +07:00
committed by Dat H. Pham
parent d2f03a9567
commit eadc0c4ddc
20 changed files with 260 additions and 148 deletions
@@ -1,14 +1,12 @@
import 'package:core/utils/app_logger.dart';
import 'package:core/utils/build_utils.dart';
import 'package:firebase_messaging/firebase_messaging.dart';
import 'package:flutter/foundation.dart';
import 'package:flutter_dotenv/flutter_dotenv.dart';
import 'package:tmail_ui_user/features/push_notification/domain/exceptions/fcm_exception.dart';
import 'package:tmail_ui_user/features/push_notification/presentation/services/fcm_service.dart';
import 'package:tmail_ui_user/main/utils/app_utils.dart';
@pragma('vm:entry-point')
Future<void> handleFirebaseBackgroundMessage(RemoteMessage message) async {
log('FcmReceiver::handleFirebaseBackgroundMessage():messageId: ${message.messageId}');
log('FcmReceiver::handleFirebaseBackgroundMessage(): ${message.data}');
FcmService.instance.handleFirebaseBackgroundMessage(message);
}
@@ -30,12 +28,13 @@ class FcmReceiver {
void getFcmToken() async {
try {
final token = await FirebaseMessaging.instance.getToken(vapidKey: kIsWeb ? dotenv.get('FIREBASE_WEB_VAPID_PUBLIC_KEY', fallback: '') : null);
log('FcmReceiver::onFcmToken():token: $token');
FcmService.instance.handleRefreshToken(token);
final currentToken = await FirebaseMessaging.instance.getToken(vapidKey: AppUtils.fcmVapidPublicKey);
log('FcmReceiver::onFcmToken():currentToken: $currentToken');
if (BuildUtils.isWeb) {
FcmService.instance.handleGetToken(currentToken);
}
} catch(e) {
log('FcmReceiver::onFcmToken():exception: $e');
throw NotLoadedFCMTokenException();
}
}
@@ -8,44 +8,84 @@ import 'package:tmail_ui_user/features/push_notification/presentation/controller
class FcmService {
static const int durationMessageComing = 2000;
static const int durationRefreshToken = 2000;
final StreamController<RemoteMessage> foregroundMessageStreamController = StreamController<RemoteMessage>.broadcast();
late StreamController<RemoteMessage> foregroundMessageStreamController;
Stream<RemoteMessage> get foregroundMessageStream => foregroundMessageStreamController.stream;
final StreamController<RemoteMessage> backgroundMessageStreamController = StreamController<RemoteMessage>.broadcast();
late StreamController<RemoteMessage> backgroundMessageStreamController;
Stream<RemoteMessage> get backgroundMessageStream => backgroundMessageStreamController.stream;
final StreamController<String?> fcmTokenStreamController = StreamController<String?>.broadcast();
late StreamController<String?> fcmTokenStreamController;
Stream<String?> get fcmTokenStream => fcmTokenStreamController.stream;
FcmService._internal();
FcmService._internal() {
foregroundMessageStreamController = StreamController<RemoteMessage>.broadcast();
backgroundMessageStreamController = StreamController<RemoteMessage>.broadcast();
fcmTokenStreamController = StreamController<String?>.broadcast();
}
static final FcmService _instance = FcmService._internal();
static FcmService get instance => _instance;
void handleFirebaseForegroundMessage(RemoteMessage newRemoteMessage) {
log('FcmService::handleFirebaseForegroundMessage():messageId: ${newRemoteMessage.messageId}');
log('FcmService::handleFirebaseForegroundMessage():message: ${newRemoteMessage.data}');
foregroundMessageStreamController.add(newRemoteMessage);
if (!foregroundMessageStreamController.isClosed) {
foregroundMessageStreamController.add(newRemoteMessage);
}
}
void handleFirebaseBackgroundMessage(RemoteMessage newRemoteMessage) {
FcmController.instance.initialize();
backgroundMessageStreamController.add(newRemoteMessage);
if (!backgroundMessageStreamController.isClosed) {
backgroundMessageStreamController.add(newRemoteMessage);
}
}
void handleRefreshToken(String? newToken) {
fcmTokenStreamController.add(newToken);
void handleGetToken(String? currentToken) async {
log('FcmService::handleGetToken():currentToken: $currentToken');
if (fcmTokenStreamController.isClosed) {
log('FcmService::handleGetToken():fcmTokenStreamController: isClosed');
fcmTokenStreamController = StreamController<String?>.broadcast();
await FcmController.instance.listenTokenStream();
}
if (!fcmTokenStreamController.isClosed) {
fcmTokenStreamController.add(currentToken);
}
}
void _closeStream() {
void handleRefreshToken(String? newToken) async {
log('FcmService::handleRefreshToken():newToken: $newToken');
if (fcmTokenStreamController.isClosed) {
log('FcmService::handleRefreshToken():fcmTokenStreamController: isClosed');
fcmTokenStreamController = StreamController<String?>.broadcast();
await FcmController.instance.listenTokenStream();
}
if (!fcmTokenStreamController.isClosed) {
fcmTokenStreamController.add(newToken);
}
}
Future<void> recreateStreamController() async {
if (foregroundMessageStreamController.isClosed) {
foregroundMessageStreamController = StreamController<RemoteMessage>.broadcast();
await FcmController.instance.listenForegroundMessageStream();
}
if (backgroundMessageStreamController.isClosed) {
backgroundMessageStreamController = StreamController<RemoteMessage>.broadcast();
await FcmController.instance.listenBackgroundMessageStream();
}
if (fcmTokenStreamController.isClosed) {
fcmTokenStreamController = StreamController<String?>.broadcast();
await FcmController.instance.listenTokenStream();
}
return Future.value();
}
void closeStream() {
foregroundMessageStreamController.close();
backgroundMessageStreamController.close();
fcmTokenStreamController.close();
}
void dispose() {
_closeStream();
}
}