feat(lane): add subscribe, unsubscribe apis

Author: ArrayZoneYour

__test__/lane/lane.spec.ts

@@ -62,6 +62,44 @@ describe('lane model', () => {
     })
   })
 
+  test('subscribe model', () => {
+    let subscribeTimes = 0
+    const { subscribe, unsubscribe, getState } = createStore(() => {
+      const [count, setCount] = useModel(1)
+      return { count, setCount }
+    })
+
+    const callback = () => {
+      subscribeTimes += 1
+    }
+
+    subscribe(callback)
+
+    act(() => {
+      expect(subscribeTimes).toEqual(0)
+    })
+
+    act(() => {
+      getState().setCount(5)
+    })
+
+    act(() => {
+      expect(subscribeTimes).toEqual(1)
+      expect(getState().count).toBe(5)
+    })
+
+    unsubscribe(callback)
+
+    act(() => {
+      getState().setCount(15)
+    })
+
+    act(() => {
+      expect(subscribeTimes).toEqual(1)
+      expect(getState().count).toBe(15)
+    })
+  })
+
   test('pass function to useModel ', async () => {
     const { useStore } = createStore(() => {
       const [count, setCount] = useModel(() => 1)

src/index.d.ts

@@ -156,6 +156,8 @@ interface API<MT extends ModelType = ModelType<any, any, {}>> {
 interface LaneAPI<S> {
   useStore: () => S
   getState: () => S
+  subscribe: (callback: () => void) => void
+  unsubscribe: (callback: () => void) => void
 }
 
 interface APIs<M extends Models> {

src/index.tsx

@@ -113,7 +113,21 @@ function createStore<S>(n: any, u?: any): LaneAPI<S> {
   return {
     // TODO: support selector
     useStore: () => useStore(storeId, selector),
-    getState: () => selector()
+    getState: () => selector(),
+    subscribe: (callback: () => void) => {
+      if (!Global.subscriptions[storeId]) {
+        Global.subscriptions[storeId] = []
+      }
+      Global.subscriptions[storeId].push(callback)
+    },
+    unsubscribe: (callback?: () => void) => {
+      if (Global.subscriptions[storeId]) {
+        if (callback) {
+          const idx = Global.subscriptions[storeId].indexOf(callback)
+          if (idx >= 0) Global.subscriptions[storeId].splice(idx)
+        }
+      }
+    }
   }
 }
 

src/middlewares.ts

@@ -114,7 +114,10 @@ const stateUpdater: Middleware = async (context, restMiddlewares) => {
 
 const subscription: Middleware = async (context, restMiddlewares) => {
   const { modelName, actionName, next, Global } = context
-  const subscriptions = Global.subscriptions[`${modelName}_${actionName}`]
+  const subscriptions =
+    context.type === 'u'
+      ? Global.subscriptions[modelName]
+      : Global.subscriptions[`${modelName}_${actionName}`]
   if (subscriptions) {
     subscriptions.forEach((callback) => {
       callback(context)