feat: sync callback for store.onInitialized

src/store/modelStore.ts

@@ -125,12 +125,16 @@ export class ModelStore extends StoreBasic<Record<string, any>> {
     this.topic.publish('unmount');
   }
 
-  onInitialized(): Promise<void> {
+  onInitialized(maybeSync?: () => void): Promise<void> {
     return new Promise((resolve) => {
       if (this._isReady) {
+        maybeSync?.();
         resolve();
       } else {
-        this.topic.subscribeOnce('ready', resolve);
+        this.topic.subscribeOnce('ready', () => {
+          maybeSync?.();
+          resolve();
+        });
       }
     });
   }

test/store.test.ts

@@ -271,7 +271,7 @@ test('Get custom compose', () => {
   process.env.NODE_ENV = prevEnv;
 });
 
-it('rxjs can observe store', () => {
+test('rxjs can observe store', () => {
   store.init();
 
   const observable = from(store);
@@ -315,3 +315,24 @@ it('rxjs can observe store', () => {
     ]),
   );
 });
+
+test('async callback for onInitialized', () => {
+  let msg = 'a';
+  store.onInitialized(() => {
+    msg += 'b';
+  });
+  msg += 'c';
+  store.init();
+  expect(msg).toBe('acb');
+});
+
+test('sync callback for onInitialized', () => {
+  store.init();
+
+  let msg = 'a';
+  store.onInitialized(() => {
+    msg += 'b';
+  });
+  msg += 'c';
+  expect(msg).toBe('abc');
+});