outaozhen 3 anos atrás
pai
commit
a1930b4eb0
2 arquivos alterados com 13 adições e 2 exclusões
  1. 2 2
      src/pages/user/Login/index.tsx
  2. 11 0
      src/pages/user/Login/userModel.ts

+ 2 - 2
src/pages/user/Login/index.tsx

@@ -32,7 +32,7 @@ const goto = () => {
 }
 }
 
 
 const Login: React.FC = () => {
 const Login: React.FC = () => {
-  // const { setAuthToken } = useModel('user')
+  const { setAuthToken } = useModel('user')
   const [userLoginState, setUserLoginState] = useState<API.LoginResult>({})
   const [userLoginState, setUserLoginState] = useState<API.LoginResult>({})
   const [type, setType] = useState<string>('account')
   const [type, setType] = useState<string>('account')
   const { initialState, setInitialState } = useModel('@@initialState')
   const { initialState, setInitialState } = useModel('@@initialState')
@@ -63,7 +63,7 @@ const Login: React.FC = () => {
   const { run: tryLogin } = useRequest((values: API.LoginParams) => login(values), {
   const { run: tryLogin } = useRequest((values: API.LoginParams) => login(values), {
     manual: true,
     manual: true,
     onSuccess: async result => {
     onSuccess: async result => {
-      // setAuthToken(result?.token)
+      setAuthToken(result?.token)
       message.success('登录成功')
       message.success('登录成功')
       await run()
       await run()
       // await fetchUserInfo()
       // await fetchUserInfo()

+ 11 - 0
src/pages/user/Login/userModel.ts

@@ -0,0 +1,11 @@
+import { useLocalStorageState } from 'ahooks'
+// import { history } from 'umi'
+
+export default () => {
+  const [, setTokenId] = useLocalStorageState('TOKEN_ID')
+  function setAuthToken(t: string) {
+    setTokenId(t)
+  }
+
+  return { setAuthToken }
+}