Observing "MemoryWarningNotification" and proxying it up to the DeviceEventEmitter

Jared Forsyth · Jared Forsyth · commit aad54006e3ba · 2015-05-22T13:09:56.000-08:00
diff --git a/Examples/UIExplorer/AppStateIOSExample.js b/Examples/UIExplorer/AppStateIOSExample.js
@@ -28,13 +28,19 @@ var AppStateSubscription = React.createClass({
     return {
       appState: AppStateIOS.currentState,
       previousAppStates: [],
+      memoryWarnings: 0,
     };
   },
   componentDidMount: function() {
     AppStateIOS.addEventListener('change', this._handleAppStateChange);
+    AppStateIOS.addEventListener('memoryWarning', this._handleMemoryWarning);
   },
   componentWillUnmount: function() {
     AppStateIOS.removeEventListener('change', this._handleAppStateChange);
+    AppStateIOS.removeEventListener('memoryWarning', this._handleMemoryWarning);
+  },
+  _handleMemoryWarning: function() {
+    this.setState({memoryWarnings: this.state.memoryWarnings + 1})
   },
   _handleAppStateChange: function(appState) {
     var previousAppStates = this.state.previousAppStates.slice();
@@ -45,6 +51,13 @@ var AppStateSubscription = React.createClass({
     });
   },
   render() {
+    if (this.props.showMemoryWarnings) {
+      return (
+        <View>
+          <Text>{this.state.memoryWarnings}</Text>
+        </View>
+      );
+    }
     if (this.props.showCurrentOnly) {
       return (
         <View>
@@ -77,4 +90,9 @@ exports.examples = [
     title: 'Previous states:',
     render(): ReactElement { return <AppStateSubscription showCurrentOnly={false} />; }
   },
+  {
+    title: 'Memory Warnings',
+    description: "In the simulator, hit Shift+Command+M to simulate a memory warning.",
+    render(): ReactElement { return <AppStateSubscription showMemoryWarnings={true} />; }
+  },
 ];
diff --git a/Libraries/AppStateIOS/AppStateIOS.ios.js b/Libraries/AppStateIOS/AppStateIOS.ios.js
@@ -16,10 +16,12 @@ var RCTDeviceEventEmitter = require('RCTDeviceEventEmitter');
 var RCTAppState = NativeModules.AppState;
 
 var logError = require('logError');
+var invariant = require('invariant');
 
-var DEVICE_APPSTATE_EVENT = 'appStateDidChange';
-
-var _appStateHandlers = {};
+var _eventHandlers = {
+  change: new Map(),
+  memoryWarning: new Map(),
+};
 
 /**
  * `AppStateIOS` can tell you if the app is in the foreground or background,
@@ -82,12 +84,23 @@ var AppStateIOS = {
     type: string,
     handler: Function
   ) {
-    _appStateHandlers[handler] = RCTDeviceEventEmitter.addListener(
-      DEVICE_APPSTATE_EVENT,
-      (appStateData) => {
-        handler(appStateData.app_state);
-      }
+    invariant(
+      ['change', 'memoryWarning'].indexOf(type) !== -1,
+      'Trying to subscribe to unknown event: "%s"', type
     );
+    if (type === 'change') {
+      _eventHandlers[type].set(handler, RCTDeviceEventEmitter.addListener(
+        'appStateDidChange',
+        (appStateData) => {
+          handler(appStateData.app_state);
+        }
+      ));
+    } else if (type === 'memoryWarning') {
+      _eventHandlers[type].set(handler, RCTDeviceEventEmitter.addListener(
+        'memoryWarning',
+        handler
+      ));
+    }
   },
 
   /**
@@ -97,19 +110,23 @@ var AppStateIOS = {
     type: string,
     handler: Function
   ) {
-    if (!_appStateHandlers[handler]) {
+    invariant(
+      ['change', 'memoryWarning'].indexOf(type) !== -1,
+      'Trying to remove listener for unknown event: "%s"', type
+    );
+    if (!_eventHandlers[type].has(handler)) {
       return;
     }
-    _appStateHandlers[handler].remove();
-    _appStateHandlers[handler] = null;
+    _eventHandlers[type].get(handler).remove();
+    _eventHandlers[type].delete(handler);
   },
 
   currentState: (null : ?String),
 
 };
 
 RCTDeviceEventEmitter.addListener(
-  DEVICE_APPSTATE_EVENT,
+  'appStateDidChange',
   (appStateData) => {
     AppStateIOS.currentState = appStateData.app_state;
   }
diff --git a/React/Modules/RCTAppState.m b/React/Modules/RCTAppState.m
@@ -53,11 +53,21 @@ - (instancetype)init
                                                selector:@selector(handleAppStateDidChange)
                                                    name:name
                                                  object:nil];
+
+      [[NSNotificationCenter defaultCenter] addObserver:self
+                                               selector:@selector(handleMemoryWarning)
+                                                   name:UIApplicationDidReceiveMemoryWarningNotification
+                                                 object:nil];
     }
   }
   return self;
 }
 
+- (void)handleMemoryWarning
+{
+  [_bridge.eventDispatcher sendDeviceEventWithName:@"memoryWarning"
+                                              body:nil];
+}
 
 - (void)dealloc
 {

Original file line number	Diff line number	Diff line change
`@@ -53,11 +53,21 @@ - (instancetype)init`
`53`	`53`	`selector:@selector(handleAppStateDidChange)`
`54`	`54`	`name:name`
`55`	`55`	`object:nil];`
	`56`	`+`
	`57`	`+ [[NSNotificationCenter defaultCenter] addObserver:self`
	`58`	`+ selector:@selector(handleMemoryWarning)`
	`59`	`+ name:UIApplicationDidReceiveMemoryWarningNotification`
	`60`	`+ object:nil];`
`56`	`61`	`}`
`57`	`62`	`}`
`58`	`63`	`return self;`
`59`	`64`	`}`
`60`	`65`
	`66`	`+- (void)handleMemoryWarning`
	`67`	`+{`
	`68`	`+ [_bridge.eventDispatcher sendDeviceEventWithName:@"memoryWarning"`
	`69`	`+ body:nil];`
	`70`	`+}`
`61`	`71`
`62`	`72`	`- (void)dealloc`
`63`	`73`	`{`