From da4b9d628a7a3370739aa5dac68be8934732c14a Mon Sep 17 00:00:00 2001
From: Kevin Zhang <kevinzhanglulu@gmail.com>
Date: Tue, 24 Mar 2020 10:15:47 +0800
Subject: [PATCH] fix: the invalid initialization

---
 src/components/Header/Header.tsx | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/components/Header/Header.tsx b/src/components/Header/Header.tsx
index 9577fd8..6b51ffc 100644
--- a/src/components/Header/Header.tsx
+++ b/src/components/Header/Header.tsx
@@ -1,4 +1,10 @@
-import React, { useState, useMemo, useEffect, useCallback } from "react";
+import React, {
+  useState,
+  useMemo,
+  useEffect,
+  useCallback,
+  useLayoutEffect,
+} from "react";
 import styles from "./header.module.css";
 import { Button, Selector } from "../Core";
 import { notifier } from "../Core/Notification";
@@ -12,7 +18,6 @@ import { proxy, Subscription } from "../../reducers/proxyReducer";
 import { ipcRenderer } from "electron";
 import { useLocation } from "react-router-dom";
 import { updateSubscription } from "../../utils/helper";
-import { useOnMount } from "../../hooks";
 import { LoadingDialog } from "../Dialogs/LoadingDialog";
 
 const Header = () => {
@@ -26,7 +31,7 @@ const Header = () => {
   const dispatch = useDispatch();
 
   //Init app
-  useOnMount(() => {
+  useLayoutEffect(() => {
     if (isUpdateSubscriptionsOnOpen) {
       const updateSubscriptions = async () => {
         await Promise.all(
@@ -74,7 +79,8 @@ const Header = () => {
     return () => {
       ipcRenderer.removeAllListeners("message");
     };
-  });
+    //Only be fired once to init app.
+  }, []);
   const customizedRulesDirPath = useSelector<AppState, string>(
     (state) => state.setting.rule.dirPath
   );