Feature support metadata cache to memory or redis (#1051)

* Fix entity class deserialization error

* Added metadata cache function, which support memory or redis

* add time filed on Result return

* Redesign the metadata UI and support caching

* fix merge

* Add scroll bar when database exceeds seven

* Add apache Licensed

Co-authored-by: steve <woai1998>

dlink-admin/pom.xml

@@ -127,10 +127,10 @@
             <artifactId>sa-token-spring-boot-starter</artifactId>
         </dependency>
         <!-- sa-token 持久化 -->
-        <!--        <dependency>-->
-        <!--            <groupId>org.springframework.boot</groupId>-->
-        <!--            <artifactId>spring-boot-starter-data-redis</artifactId>-->
-        <!--        </dependency>-->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-redis</artifactId>
+        </dependency>
         <!--        <dependency>-->
         <!--            <groupId>cn.dev33</groupId>-->
         <!--            <artifactId>sa-token-dao-redis-jackson</artifactId>-->

dlink-admin/src/main/java/com/dlink/Dlink.java

@@ -21,6 +21,7 @@ package com.dlink;
 
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.cache.annotation.EnableCaching;
 import org.springframework.transaction.annotation.EnableTransactionManagement;
 
 /**
@@ -31,6 +32,7 @@ import org.springframework.transaction.annotation.EnableTransactionManagement;
  */
 @EnableTransactionManagement
 @SpringBootApplication
+@EnableCaching
 public class Dlink {
 
     public static void main(String[] args) {

dlink-admin/src/main/java/com/dlink/common/result/Result.java

@@ -23,6 +23,7 @@ import com.dlink.model.CodeEnum;
 
 import java.io.Serializable;
 
+import cn.hutool.core.date.DateTime;
 import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.NoArgsConstructor;
@@ -41,6 +42,7 @@ public class Result<T> implements Serializable {
     private T datas;
     private Integer code;
     private String msg;
+    private String time;
 
     public static <T> Result<T> succeed(String msg) {
         return of(null, CodeEnum.SUCCESS.getCode(), msg);
@@ -55,7 +57,7 @@ public class Result<T> implements Serializable {
     }
 
     public static <T> Result<T> of(T datas, Integer code, String msg) {
-        return new Result<>(datas, code, msg);
+        return new Result<>(datas, code, msg,new DateTime().toString());
     }
 
     public static <T> Result<T> failed(String msg) {

dlink-admin/src/main/java/com/dlink/configure/CacheConfigure.java

+/*
+ *
+ *  Licensed to the Apache Software Foundation (ASF) under one or more
+ *  contributor license agreements.  See the NOTICE file distributed with
+ *  this work for additional information regarding copyright ownership.
+ *  The ASF licenses this file to You under the Apache License, Version 2.0
+ *  (the "License"); you may not use this file except in compliance with
+ *  the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ */
+
+package com.dlink.configure;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.cache.RedisCacheConfiguration;
+import org.springframework.data.redis.serializer.RedisSerializationContext;
+import org.springframework.data.redis.serializer.RedisSerializer;
+import org.springframework.data.redis.serializer.StringRedisSerializer;
+
+
+/**
+ * CacheCoonfigure
+ *
+ * @author ikiler
+ * @since 2022/09/24 11:23
+ */
+@Configuration
+public class CacheConfigure {
+
+    /**
+     * 配置Redis缓存注解的value序列化方式
+     */
+    @Bean
+    public RedisCacheConfiguration cacheConfiguration() {
+        return RedisCacheConfiguration.defaultCacheConfig()
+                //序列化为json
+                .serializeValuesWith(
+                        RedisSerializationContext.SerializationPair.fromSerializer(RedisSerializer.json())
+                )
+                .serializeKeysWith(RedisSerializationContext.SerializationPair.fromSerializer(new StringRedisSerializer()));
+    }
+
+
+    //    /**
+    //     * 配置RedisTemplate的序列化方式
+    //     */
+    //    @Bean
+    //    public RedisTemplate redisTemplate(RedisConnectionFactory factory) {
+    //        RedisTemplate redisTemplate = new RedisTemplate();
+    //        redisTemplate.setConnectionFactory(factory);
+    //        // 指定key的序列化方式：string
+    //        redisTemplate.setKeySerializer(RedisSerializer.string());
+    //        // 指定value的序列化方式：json
+    //        redisTemplate.setValueSerializer(RedisSerializer.json());
+    //        return redisTemplate;
+    //    }
+
+}

dlink-admin/src/main/java/com/dlink/controller/DataBaseController.java

@@ -35,6 +35,8 @@ import java.util.List;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.cache.annotation.CacheEvict;
+import org.springframework.cache.annotation.Cacheable;
 import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -167,11 +169,21 @@ public class DataBaseController {
     /**
      * 获取元数据的表
      */
+    @Cacheable(cacheNames = "metadata_schema",key = "#id")
     @GetMapping("/getSchemasAndTables")
     public Result getSchemasAndTables(@RequestParam Integer id) {
         return Result.succeed(databaseService.getSchemasAndTables(id), "获取成功");
     }
 
+    /**
+     * 清除元数据表的缓存
+     */
+    @CacheEvict(cacheNames = "metadata_schema",key = "#id")
+    @GetMapping("/unCacheSchemasAndTables")
+    public Result unCacheSchemasAndTables(@RequestParam Integer id) {
+        return Result.succeed("clear cache", "success");
+    }
+
     /**
      * 获取元数据的指定表的列
      */

dlink-admin/src/main/resources/application.yml

@@ -11,6 +11,20 @@ spring:
       matching-strategy: ant_path_matcher
   main:
     allow-circular-references: true
+
+#  默认使用内存缓存元数据信息，
+#  dlink支持redis缓存，如有需要请把simple改为redis，并打开下面的redis连接配置
+#  子配置项可以按需要打开或自定义配置
+  cache:
+    type: simple
+##    如果type配置为redis，则该项可按需配置
+#    redis:
+##      是否缓存空值，保存默认即可
+#      cache-null-values: false
+##      缓存过期时间，24小时
+#      time-to-live: 86400
+
+
   #  flyway:
   #    enabled: false
   #    clean-disabled: true

dlink-common/src/main/java/com/dlink/model/Schema.java

@@ -45,6 +45,12 @@ public class Schema implements Serializable, Comparable<Schema> {
     private List<String> userFunctions = new ArrayList<>();
     private List<String> modules = new ArrayList<>();
 
+    /**
+     * 需要保留一个空构造方法，否则序列化有问题
+     * */
+    public Schema() {
+    }
+
     public Schema(String name) {
         this.name = name;
     }

dlink-common/src/main/java/com/dlink/model/Table.java

@@ -22,6 +22,7 @@ package com.dlink.model;
 import com.dlink.assertion.Asserts;
 import com.dlink.utils.SqlUtil;
 
+import java.beans.Transient;
 import java.io.Serializable;
 import java.util.ArrayList;
 import java.util.Date;
@@ -75,10 +76,12 @@ public class Table implements Serializable, Comparable<Table> {
         this.columns = columns;
     }
 
+    @Transient
     public String getSchemaTableName() {
         return Asserts.isNullString(schema) ? name : schema + "." + name;
     }
 
+    @Transient
     public String getSchemaTableNameWithUnderline() {
         return Asserts.isNullString(schema) ? name : schema + "_" + name;
     }
@@ -100,6 +103,7 @@ public class Table implements Serializable, Comparable<Table> {
         return new Table(name, schema, columns);
     }
 
+    @Transient
     public String getFlinkTableWith(String flinkConfig) {
         String tableWithSql = "";
         if (Asserts.isNotNullString(flinkConfig)) {
@@ -109,10 +113,12 @@ public class Table implements Serializable, Comparable<Table> {
         return tableWithSql;
     }
 
+    @Transient
     public String getFlinkTableSql(String flinkConfig) {
         return getFlinkDDL(flinkConfig, name);
     }
 
+    @Transient
     public String getFlinkDDL(String flinkConfig, String tableName) {
         StringBuilder sb = new StringBuilder();
         sb.append("CREATE TABLE IF NOT EXISTS " + tableName + " (\n");
@@ -162,6 +168,7 @@ public class Table implements Serializable, Comparable<Table> {
         return sb.toString();
     }
 
+    @Transient
     public String getFlinkTableSql(String catalogName, String flinkConfig) {
         StringBuilder sb = new StringBuilder("DROP TABLE IF EXISTS ");
         String fullSchemaName = catalogName + "." + schema + "." + name;
@@ -213,6 +220,7 @@ public class Table implements Serializable, Comparable<Table> {
         return sb.toString();
     }
 
+    @Transient
     public String getSqlSelect(String catalogName) {
         StringBuilder sb = new StringBuilder("SELECT\n");
         for (int i = 0; i < columns.size(); i++) {
@@ -239,6 +247,7 @@ public class Table implements Serializable, Comparable<Table> {
         return sb.toString();
     }
 
+    @Transient
     public String getCDCSqlInsert(String targetName, String sourceName) {
         StringBuilder sb = new StringBuilder("INSERT INTO ");
         sb.append(targetName);

dlink-web/src/components/Studio/StudioEvent/DDL.ts

@@ -222,6 +222,11 @@ export function showAlertGroup(dispatch: any) {
 export function showMetaDataTable(id: number) {
   return getData('api/database/getSchemasAndTables', {id: id});
 }
+/*--- 清理 元数据表缓存 ---*/
+export function clearMetaDataTable(id: number) {
+  return getData('api/database/unCacheSchemasAndTables', {id: id});
+}
+
 /*--- 刷新 数据表样例数据 ---*/
 export function showTableData(id: number,schemaName:String,tableName:String,option:{}) {
   return postAll('api/database/queryData', {id: id,schemaName:schemaName,tableName:tableName,option:option});

dlink-web/src/components/Studio/StudioTree/Function.ts

@@ -32,6 +32,8 @@ export interface TreeDataNode extends DataNode {
   taskId:number;
   parentId:number;
   path:string[];
+  schema:string;
+  table:string;
 }
 
 export function convertToTreeData(data:TreeDataNode[], pid:number,path?:string[]) {

dlink-web/src/pages/DataCenter/MetaData/TableData/index.less

+/*
+ *
+ *  Licensed to the Apache Software Foundation (ASF) under one or more
+ *  contributor license agreements.  See the NOTICE file distributed with
+ *  this work for additional information regarding copyright ownership.
+ *  The ASF licenses this file to You under the Apache License, Version 2.0
+ *  (the "License"); you may not use this file except in compliance with
+ *  the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ */
+
 
 .margin_10{
   margin: 10px;

dlink-web/src/pages/DataCenter/MetaData/TableData/index.tsx

+/*
+ *
+ *  Licensed to the Apache Software Foundation (ASF) under one or more
+ *  contributor license agreements.  See the NOTICE file distributed with
+ *  this work for additional information regarding copyright ownership.
+ *  The ASF licenses this file to You under the Apache License, Version 2.0
+ *  (the "License"); you may not use this file except in compliance with
+ *  the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ */
+
+
 import {useEffect, useState} from "react";
 import {Alert, AutoComplete, Button, Col, Input, Row, Spin, Table, Tooltip} from "antd";
 import {showTableData} from "@/components/Studio/StudioEvent/DDL";
 import styles from './index.less';
-import { SearchOutlined} from "@ant-design/icons";
+import {SearchOutlined} from "@ant-design/icons";
 import Divider from "antd/es/divider";
 
 const TableData = (props: any) => {
 
   // 数据库id，数据库名称，表名称
-  const {dbId, table, schema,rows} = props;
+  const {dbId, table, schema} = props;
   // 表数据
-  const [tableData, setableData] = useState<{ columns: {}[],rowData: {}[] }>({columns: [], rowData: []});
+  const [tableData, setableData] = useState<{ columns: {}[], rowData: {}[] }>({columns: [], rowData: []});
   // 加载状态
   const [loading, setLoading] = useState<boolean>(false);
   // 列名和列信息数据
   const [columns, setColumns] = useState<string[]>([]);
-  const [errMsg, setErrMsg] = useState<{isErr:boolean,msg:string}>({isErr:false,msg:""});
+  const [errMsg, setErrMsg] = useState<{ isErr: boolean, msg: string }>({isErr: false, msg: ""});
   //使用多选框进行自由列选择，屏蔽不需要看见的列，目前实现有些问题，暂时屏蔽
   // 是否展开字段选择
   // const [showFilter, setShowFilter] = useState(false);
   // 是否全选，全选状态相关
   // const [state, setState] = useState<{ checkedList: [], indeterminate: boolean, checkAll: boolean }>({checkedList: [], indeterminate: true, checkAll: false});
   // 条件查询时联想输入使用
-  const [options, setOptions] = useState<{whereOption: {}[],orderOption:{}[] }>({whereOption:[],orderOption:[]});
+  const [options, setOptions] = useState<{ whereOption: {}[], orderOption: {}[] }>({whereOption: [], orderOption: []});
 // where输入框内容
-const [optionInput, setOptionInput] = useState<{whereInput:string,orderInput:string}>({whereInput:"",orderInput:""});
-const [page, setPage] = useState<{ page:number,pageSize:number }>({page:0 ,pageSize:10});
+  const [optionInput, setOptionInput] = useState<{ whereInput: string, orderInput: string }>({
+    whereInput: "",
+    orderInput: ""
+  });
+  const [page, setPage] = useState<{ page: number, pageSize: number }>({page: 0, pageSize: 10});
 // const [defaultInput,setDefaultInput]
 // 获取数据库数据
-const fetchData = async (whereInput:string,orderInput:string) => {
+  const fetchData = async (whereInput: string, orderInput: string) => {
 
     setLoading(true);
-  let temp= {rowData: [], columns: []}
+    let temp = {rowData: [], columns: []}
 
-  let option = {where:whereInput,
-    order:orderInput,limitStart:"0",limitEnd:"500"}
+    let option = {
+      where: whereInput,
+      order: orderInput, limitStart: "0", limitEnd: "500"
+    }
 
     await showTableData(dbId, schema, table, option).then(result => {
-    if (result.code == 1){
-      setErrMsg({isErr:true,msg:result.datas.error})
-    }else {
-      setErrMsg({isErr:false,msg:""})
+      if (result.code == 1) {
+        setErrMsg({isErr: true, msg: result.datas.error})
+      } else {
+        setErrMsg({isErr: false, msg: ""})
       }
       let data = result.datas;
       setColumns(data.columns)
@@ -61,21 +86,21 @@ const fetchData = async (whereInput:string,orderInput:string) => {
     })
     setableData(temp);
     setLoading(false)
-};
+  };
 
-useEffect(() => {
+  useEffect(() => {
     setColumns([])
     setableData({columns: [], rowData: []})
-  setErrMsg({isErr:false,msg:""})
-  setOptions({whereOption:[],orderOption:[]})
-  setOptionInput({whereInput:"",orderInput:""})
-  setPage({page:0 ,pageSize:10})
+    setErrMsg({isErr: false, msg: ""})
+    setOptions({whereOption: [], orderOption: []})
+    setOptionInput({whereInput: "", orderInput: ""})
+    setPage({page: 0, pageSize: 10})
     setLoading(false)
 
-  fetchData("","");
+    fetchData("", "");
 
 
-}, [dbId, table, schema]);
+  }, [dbId, table, schema]);
 
 //使用多选框进行自由列选择，屏蔽不需要看见的列，目前实现有些问题，暂时屏蔽
 // // 单项选择监听
@@ -99,8 +124,8 @@ useEffect(() => {
 
 
 // 条件查询时反馈联想信息
-const handleRecommend = (value: string) => {
-  if (columns==null){
+  const handleRecommend = (value: string) => {
+    if (columns == null) {
       return []
     }
     let inputSplit: string[] = value.split(" ");
@@ -110,44 +135,43 @@ const handleRecommend = (value: string) => {
     console.log(inputSplit)
     for (let column of columns) {
       if (column.startsWith(lastWord)) {
-      let msg = inputSplit.join("") + " "+column
-      recommend.push({value: msg })
+        let msg = inputSplit.join("") + " " + column
+        recommend.push({value: msg})
       }
     }
     return recommend;
-};
+  };
 
-const handleWhere = (value:string) =>{
+  const handleWhere = (value: string) => {
     let result = handleRecommend(value)
     setOptions({
-    whereOption:result,
-    orderOption:[]
+      whereOption: result,
+      orderOption: []
     })
-}
+  }
 
-const handleOrder = (value:string) =>{
+  const handleOrder = (value: string) => {
     let result = handleRecommend(value)
     setOptions({
-    orderOption:result,
-    whereOption:[]
+      orderOption: result,
+      whereOption: []
     })
-}
-
+  }
 
 
-return (
+  return (
     <div>
       <Spin spinning={loading} delay={500}>
 
         <div className={styles.mrgin_top_40}>
-        {errMsg.isErr?(
+          {errMsg.isErr ? (
             <Alert
               message="Error"
               description={errMsg.msg}
               type="error"
               showIcon
             />
-        ):<></>}
+          ) : <></>}
           <Row>
             <Col span={12}>
               <AutoComplete
@@ -155,15 +179,19 @@ return (
                 options={options.whereOption}
                 style={{width: "100%"}}
                 onSearch={handleWhere}
-              onSelect={(value:string, option)=>{
-                setOptionInput({whereInput:value,
-                  orderInput:optionInput.orderInput})
+                onSelect={(value: string, option) => {
+                  setOptionInput({
+                    whereInput: value,
+                    orderInput: optionInput.orderInput
+                  })
                 }}
               >
                 <Input addonBefore="WHERE" placeholder="查询条件"
-                     onChange={(value)=>{
-                setOptionInput({whereInput:value.target.value,
-                  orderInput:optionInput.orderInput})
+                       onChange={(value) => {
+                         setOptionInput({
+                           whereInput: value.target.value,
+                           orderInput: optionInput.orderInput
+                         })
                        }}
                 />
               </AutoComplete>
@@ -175,14 +203,18 @@ return (
                 options={options.orderOption}
                 style={{width: "100%"}}
                 onSearch={handleOrder}
-              onSelect={(value:string, option)=>{
-                setOptionInput({whereInput:optionInput.whereInput,
-                  orderInput:value})
+                onSelect={(value: string, option) => {
+                  setOptionInput({
+                    whereInput: optionInput.whereInput,
+                    orderInput: value
+                  })
                 }}
               >
-              <Input addonBefore="ORDER BY"  placeholder="排序" onChange={(value)=>{
-                setOptionInput({whereInput:optionInput.whereInput,
-                  orderInput:value.target.value})
+                <Input addonBefore="ORDER BY" placeholder="排序" onChange={(value) => {
+                  setOptionInput({
+                    whereInput: optionInput.whereInput,
+                    orderInput: value.target.value
+                  })
                 }}/>
               </AutoComplete>
             </Col>
@@ -216,8 +248,8 @@ return (
             {/*</Col>*/}
             <Col span={1}>
               <Tooltip title="查询">
-              <Button type="primary" shape="circle" icon={<SearchOutlined/>} size="middle" onClick={(event)=>{
-                fetchData(optionInput.whereInput,optionInput.orderInput)
+                <Button type="primary" shape="circle" icon={<SearchOutlined/>} size="middle" onClick={(event) => {
+                  fetchData(optionInput.whereInput, optionInput.orderInput)
                 }}/>
               </Tooltip>
             </Col>
@@ -231,10 +263,12 @@ return (
           <Table style={{height: '95vh'}}
                  columns={tableData.columns}
                  dataSource={tableData.rowData}
-               pagination={{pageSize:page.pageSize,
+                 pagination={{
+                   pageSize: page.pageSize,
                    onChange: (pageNum, pageSize) => {
-                   setPage({page:pageNum,pageSize:pageSize});
-                 }}}
+                     setPage({page: pageNum, pageSize: pageSize});
+                   }
+                 }}
                  scroll={{y: "80vh", x: true}}
           />
         </div>
@@ -242,7 +276,7 @@ return (
 
     </div>
 
-)
+  )
 };
 
 export default TableData

dlink-web/src/pages/DataCenter/MetaData/index.less

+/*
+ *
+ *  Licensed to the Apache Software Foundation (ASF) under one or more
+ *  contributor license agreements.  See the NOTICE file distributed with
+ *  this work for additional information regarding copyright ownership.
+ *  The ASF licenses this file to You under the Apache License, Version 2.0
+ *  (the "License"); you may not use this file except in compliance with
+ *  the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ */
 
+
+//主体样式
 .container {
+  margin-top: 5px;
   background: #ffffff;
   height: 95vh;
   padding: 8px
 }
+//顶部数据库列表父组件样式，覆盖原有seagement样式
+.headerBarContent {
+  overflow-x: scroll;
+  width: 100%;
+  background-color:#fafafa;
+  :global{
+    .ant-segmented:not(.ant-segmented-disabled):hover, .ant-segmented:not(.ant-segmented-disabled):focus{
+      background-color:#fafafa;
+    }
+  }
+}
+//顶部数据库列表样式
+.headerBar {
+  background: #fafafa;
+  padding: 8px;
+}
+//数据库列表单条car样式
+.headerCard {
+  width:200px;
+  :global{
+    .ant-card-body{
+      padding: 8px;
+    }
+  }
+}
+
+//table标题样式
+.tableListHead{
+  background-color: #f9f9f9;
+  padding: 8px;
+  :global{
+    .ant-card-meta-title{
+      font-size: 14px;
+    }
+    .ant-card-meta-description{
+      color: rgba(0, 0, 0, 0.45);
+      font-size: 10px;
 
-.content-height{
-  height: 100%;
+    }
+  }
 }

dlink-web/src/pages/DataCenter/MetaData/index.tsx

-import { PageContainer } from '@ant-design/pro-layout'; //
+/*
+ *
+ *  Licensed to the Apache Software Foundation (ASF) under one or more
+ *  contributor license agreements.  See the NOTICE file distributed with
+ *  this work for additional information regarding copyright ownership.
+ *  The ASF licenses this file to You under the Apache License, Version 2.0
+ *  (the "License"); you may not use this file except in compliance with
+ *  the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ */
+
+
+
+import {PageContainer} from '@ant-design/pro-layout'; //
 
 import styles from './index.less';
-import { Row, Col, Tabs, Select, Tag, Empty, Tree, Spin } from 'antd';
-import React, { Key, useEffect, useState } from 'react';
-import { showMetaDataTable } from '@/components/Studio/StudioEvent/DDL';
-import { Scrollbars } from 'react-custom-scrollbars';
-import { getData } from '@/components/Common/crud';
+import {Row, Col, Tabs, Tag, Empty, Tree, Segmented, Card, Image, Spin, Button} from 'antd';
+import React, {Key, useEffect, useState} from 'react';
+import {clearMetaDataTable, showMetaDataTable} from '@/components/Studio/StudioEvent/DDL';
+import {getData} from '@/components/Common/crud';
 import {
-  BarsOutlined,
+  BarsOutlined, CheckCircleOutlined,
   ConsoleSqlOutlined,
   DatabaseOutlined,
-  DownOutlined,
+  DownOutlined, ExclamationCircleOutlined,
   ReadOutlined,
   TableOutlined,
 } from '@ant-design/icons';
-import { TreeDataNode } from '@/components/Studio/StudioTree/Function';
+import {TreeDataNode} from '@/components/Studio/StudioTree/Function';
 import Tables from '@/pages/DataBase/Tables';
 import Columns from '@/pages/DataBase/Columns';
 import Divider from 'antd/es/divider';
 import Generation from '@/pages/DataBase/Generation';
 import TableData from '@/pages/DataCenter/MetaData/TableData';
-const { DirectoryTree } = Tree;
-const { TabPane } = Tabs;
+import {FALLBACK, getDBImage} from "@/pages/DataBase/DB";
+import Meta from "antd/lib/card/Meta";
+
+const {DirectoryTree} = Tree;
+const {TabPane} = Tabs;
 
 const Container: React.FC<{}> = (props: any) => {
-  // const { dispatch} = props;
-  let [database, setDatabase] = useState([
-    {
-      id: '',
+
+
+  let [database, setDatabase] = useState<[{
+    id: number,
+    name: string,
+    alias: string,
+    type: string,
+    enabled: string,
+    groupName: string,
+    status: string,
+    time: string
+  }]>([{
+    id: -1,
     name: '',
     alias: '',
     type: '',
     enabled: '',
-    },
-  ]);
+    groupName: '',
+    status: '',
+    time:''
+  }]);
+
   const [databaseId, setDatabaseId] = useState<number>();
-  const [treeData, setTreeData] = useState<[]>([]);
+  const [treeData, setTreeData] = useState<{tables:[],updateTime:string}>({tables:[],updateTime:"none"});
   const [row, setRow] = useState<TreeDataNode>();
   const [loadingDatabase, setloadingDatabase] = useState(false);
 
@@ -58,22 +93,22 @@ const Container: React.FC<{}> = (props: any) => {
         for (let i = 0; i < tables.length; i++) {
           tables[i].title = tables[i].name;
           tables[i].key = tables[i].name;
-          tables[i].icon = <DatabaseOutlined />;
+          tables[i].icon = <DatabaseOutlined/>;
           tables[i].children = tables[i].tables;
 
           for (let j = 0; j < tables[i].children.length; j++) {
             tables[i].children[j].title = tables[i].children[j].name;
             tables[i].children[j].key = tables[i].name + '.' + tables[i].children[j].name;
-            tables[i].children[j].icon = <TableOutlined />;
+            tables[i].children[j].icon = <TableOutlined/>;
             tables[i].children[j].isLeaf = true;
             tables[i].children[j].schema = tables[i].name;
             tables[i].children[j].table = tables[i].children[j].name;
           }
         }
 
-        setTreeData(tables);
+        setTreeData({tables:tables,updateTime:result.time});
       } else {
-        setTreeData([]);
+        setTreeData({tables:[],updateTime:"none"});
       }
     });
     setloadingDatabase(false);
@@ -83,31 +118,16 @@ const Container: React.FC<{}> = (props: any) => {
     fetchDatabaseList();
   }, []);
 
-  const getDataBaseOptions = () => {
-    return (
-      <>
-        {database.map(({ id, name, alias, type, enabled }) => (
-          <Select.Option
-            key={id}
-            value={id}
-            label={
-              <>
-                <Tag color={enabled ? 'processing' : 'error'}>{type}</Tag>
-                {alias === '' ? name : alias}
-              </>
-            }
-          >
-            <Tag color={enabled ? 'processing' : 'error'}>{type}</Tag>
-            {alias === '' ? name : alias}
-          </Select.Option>
-        ))}
-      </>
-    );
+  const onChangeDataBase = (value: string|number) => {
+    onRefreshTreeData(Number(value));
+    setRow(null);
   };
 
-  const onChangeDataBase = (value: number) => {
-    onRefreshTreeData(value);
-    setRow(null);
+  const refeshDataBase = (value: string | number) => {
+    setloadingDatabase(true);
+    clearMetaDataTable(Number(databaseId)).then(result=>{
+      onChangeDataBase(Number(value));
+    })
   };
 
   const showTableInfo = (selected: boolean, node: TreeDataNode) => {
@@ -121,44 +141,107 @@ const Container: React.FC<{}> = (props: any) => {
     fetchDatabase(databaseId);
   };
 
+
+  const buildDatabaseBar = () => {
+    return database.map((item) => {
+        return {
+          label: (
+            <Card className={styles.headerCard}
+                  hoverable={false} bordered={false}>
+              <Row>
+                <Col span={11}>
+                  <Image style={{float: "left", paddingRight: "10px"}}
+                         height={50}
+                         preview={false}
+                         src={getDBImage(item.type)}
+                         fallback={FALLBACK}
+                  />
+                </Col>
+                <Col span={11}>
+                  <div>
+                    <p>{item.alias}</p>
+                    <Tag color="blue" key={item.groupName}>
+                      {item.groupName}
+                    </Tag>
+                    {(item.status) ?
+                      (<Tag icon={<CheckCircleOutlined/>} color="success">
+                        正常
+                      </Tag>) :
+                      <Tag icon={<ExclamationCircleOutlined/>} color="warning">
+                        异常
+                      </Tag>}
+                  </div>
+                </Col>
+              </Row>
+            </Card>
+          ),
+          value: item.id,
+        }
+      }
+    )
+  };
+
+  const buildListTitle = () => {
+    for (let item of database) {
+      if (item.id == databaseId) {
+        return (
+          <div>
+            <div style={{position: "absolute", right: "10px"}}>
+              <Button type="link" size="small"
+                      loading={loadingDatabase}
+                      onClick={()=>refeshDataBase(databaseId)}
+              >刷新</Button>
+            </div>
+            <div>{item.alias}</div>
+          </div>
+        )
+      }
+    }
+    return (<div>未选择数据库</div>)
+  }
+
+
+
+
+
   return (
-    <div className={styles.container}>
     <div>
-        <>
+      <div className={styles.headerBarContent}>
+        <Segmented className={styles.headerBar}
+                   options={buildDatabaseBar()}
+                   onChange={onChangeDataBase}
+        />
+      </div>
+      <div className={styles.container}>
         <Row gutter={24}>
           <Col span={4}>
-              <Select
-                style={{
-                  width: '90%',
-                }}
-                placeholder="选择数据源"
-                optionLabelProp="label"
-                onChange={onChangeDataBase}
-              >
-                {getDataBaseOptions()}
-              </Select>
             <Spin spinning={loadingDatabase} delay={500}>
-                <Scrollbars
-                  style={{
-                    height: '90vh',
-                  }}
+              <Card
+                type="inner"
+                bodyStyle={{padding: 0}}
               >
-                  {treeData.length > 0 ? (
+                <Meta title={buildListTitle()}
+                      className={styles.tableListHead}
+                      description={"上次更新："+treeData.updateTime}
+                />
+
+                {treeData.tables.length > 0 ? (
                   <DirectoryTree
+                    height={850}
                     showIcon
-                      switcherIcon={<DownOutlined />}
-                      treeData={treeData}
+                    switcherIcon={<DownOutlined/>}
+                    treeData={treeData.tables}
                     onSelect={(
                       selectedKeys: Key[],
-                        { event, selected, node, selectedNodes, nativeEvent }
+                      {event, selected, node, selectedNodes, nativeEvent}
                     ) => {
                       showTableInfo(selected, node);
                     }}
                   />
                 ) : (
-                    <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} />
+                  <Empty image={Empty.PRESENTED_IMAGE_SIMPLE}/>
                 )}
-                </Scrollbars>
+              </Card>
             </Spin>
           </Col>
 
@@ -169,7 +252,7 @@ const Container: React.FC<{}> = (props: any) => {
                   <TabPane
                     tab={
                       <span>
-                          <ReadOutlined />
+                          <ReadOutlined/>
                           描述
                         </span>
                     }
@@ -179,49 +262,49 @@ const Container: React.FC<{}> = (props: any) => {
                       表信息
                     </Divider>
                     {row ? (
-                        <Tables table={row} />
+                      <Tables table={row}/>
                     ) : (
-                        <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} />
+                      <Empty image={Empty.PRESENTED_IMAGE_SIMPLE}/>
                     )}
                     <Divider orientation="left" plain>
                       字段信息
                     </Divider>
                     {row ? (
-                        <Columns dbId={databaseId} schema={row.schema} table={row.table} />
+                      <Columns dbId={databaseId} schema={row.schema} table={row.table}/>
                     ) : (
-                        <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} />
+                      <Empty image={Empty.PRESENTED_IMAGE_SIMPLE}/>
                     )}
                   </TabPane>
 
                   <TabPane
                     tab={
                       <span>
-                          <BarsOutlined />
+                          <BarsOutlined/>
                           数据查询
                         </span>
                     }
                     key="exampleData"
                   >
                     {row ? (
-                        <TableData dbId={databaseId} schema={row.schema} table={row.table} rows={row.rows}/>
+                      <TableData dbId={databaseId} schema={row.schema} table={row.table} />
                     ) : (
-                        <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} />
+                      <Empty image={Empty.PRESENTED_IMAGE_SIMPLE}/>
                     )}
                   </TabPane>
 
                   <TabPane
                     tab={
                       <span>
-                          <ConsoleSqlOutlined />
+                          <ConsoleSqlOutlined/>
                           SQL 生成
                         </span>
                     }
                     key="sqlGeneration"
                   >
                     {row ? (
-                        <Generation dbId={databaseId} schema={row.schema} table={row.table} />
+                      <Generation dbId={databaseId} schema={row.schema} table={row.table}/>
                     ) : (
-                        <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} />
+                      <Empty image={Empty.PRESENTED_IMAGE_SIMPLE}/>
                     )}
                   </TabPane>
 
@@ -230,7 +313,6 @@ const Container: React.FC<{}> = (props: any) => {
             </div>
           </Col>
         </Row>
-        </>
       </div>
     </div>
   );
@@ -239,7 +321,7 @@ const Container: React.FC<{}> = (props: any) => {
 export default () => {
   return (
     <PageContainer>
-      <Container />
+      <Container/>
     </PageContainer>
   );
 };