From c27d7d4e1ee86d7d8b4e6581e7d2ef96717d028d Mon Sep 17 00:00:00 2001
From: xiangheng <11675084@qq.com>
Date: Mon, 26 Feb 2024 12:39:33 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=AF=BC=E5=85=A5=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E6=95=B0=E6=8D=AE=E4=B8=8B=E6=A0=87=E4=B8=8D=E5=AF=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 admin/src/api/perms/admin.ts                  |  7 ++-
 admin/src/components/upload/index.vue         | 16 ++++-
 .../verifition/Verify/VerifySlide.vue         |  2 +
 .../views/setting/permission/admin/index.vue  | 20 +++++-
 server/admin/system/admin/admin.go            | 45 ++++++++++++++
 server/admin/system/enter.go                  |  2 +
 server/main.go                                |  1 +
 server/util/excel/excel_import.go             | 61 ++++++++++++-------
 8 files changed, 129 insertions(+), 25 deletions(-)

diff --git a/admin/src/api/perms/admin.ts b/admin/src/api/perms/admin.ts
index 566e56e..66aef9e 100644
--- a/admin/src/api/perms/admin.ts
+++ b/admin/src/api/perms/admin.ts
@@ -1,6 +1,7 @@
 import request from '@/utils/request'
 import queryString from 'query-string'
 import { getToken } from '@/utils/auth'
+import config from '@/config'
 // 管理员列表
 export function adminLists(params: any) {
     return request.get({ url: '/system/admin/list', params })
@@ -36,9 +37,13 @@ export function adminListByDeptId(params: any) {
     return request.get({ url: '/system/admin/ListByDeptId', params })
 }
 
+// 导入
+export const adminImportFile = '/system/admin/ImportFile'
+
 // 导出
 export function adminExportFile(params: any) {
     // return request.get({ url: '/system/admin/ExportFile', params })
     return (window.location.href =
-        `/api/admin/system/admin/ExportFile?token=${getToken()}&` + queryString.stringify(params))
+        `${config.baseUrl}${config.urlPrefix}/system/admin/ExportFile?token=${getToken()}&` +
+        queryString.stringify(params))
 }
diff --git a/admin/src/components/upload/index.vue b/admin/src/components/upload/index.vue
index 8f4a5e9..7728e4e 100644
--- a/admin/src/components/upload/index.vue
+++ b/admin/src/components/upload/index.vue
@@ -49,6 +49,11 @@ import { RequestCodeEnum } from '@/enums/requestEnums'
 export default defineComponent({
     components: {},
     props: {
+        // 上传地址
+        url: {
+            type: String,
+            default: ''
+        },
         // 上传文件类型
         type: {
             type: String,
@@ -79,7 +84,16 @@ export default defineComponent({
     setup(props, { emit }) {
         const userStore = useUserStore()
         const uploadRefs = shallowRef<InstanceType<typeof ElUpload>>()
-        const action = ref(`${config.baseUrl}${config.urlPrefix}/common/upload/${props.type}`)
+        let action = ''
+        if (props.url) {
+            if (props.url.startsWith('http')) {
+                action = props.url
+            } else {
+                action = `${config.baseUrl}${config.urlPrefix}${props.url}`
+            }
+        } else {
+            action = `${config.baseUrl}${config.urlPrefix}/common/upload/${props.type}`
+        }
         const headers = computed(() => ({
             token: userStore.token,
             version: config.version
diff --git a/admin/src/components/verifition/Verify/VerifySlide.vue b/admin/src/components/verifition/Verify/VerifySlide.vue
index c4d0bf0..208bf1c 100644
--- a/admin/src/components/verifition/Verify/VerifySlide.vue
+++ b/admin/src/components/verifition/Verify/VerifySlide.vue
@@ -10,6 +10,7 @@
                 :style="{ width: setSize.imgWidth, height: setSize.imgHeight }"
             >
                 <img
+                    v-show="backImgBase"
                     :src="'data:image/png;base64,' + backImgBase"
                     alt=""
                     style="width: 100%; height: 100%; display: block"
@@ -74,6 +75,7 @@
                         }"
                     >
                         <img
+                            v-show="blockBackImgBase"
                             :src="'data:image/png;base64,' + blockBackImgBase"
                             alt=""
                             style="
diff --git a/admin/src/views/setting/permission/admin/index.vue b/admin/src/views/setting/permission/admin/index.vue
index f724c83..22a6d7f 100644
--- a/admin/src/views/setting/permission/admin/index.vue
+++ b/admin/src/views/setting/permission/admin/index.vue
@@ -32,6 +32,17 @@
                 </template>
                 新增
             </el-button>
+            <upload
+                class="mr-3"
+                :url="adminImportFile"
+                :data="{ cid: 0 }"
+                type="file"
+                :show-progress="true"
+                @change="resetPage"
+            >
+                <el-button type="primary">导入</el-button>
+            </upload>
+
             <el-button type="primary" @click="exportFile">
                 <template #icon>
                     <icon name="el-icon-Download" />
@@ -98,7 +109,13 @@
 </template>
 
 <script lang="ts" setup>
-import { adminLists, adminDelete, adminStatus, adminExportFile } from '@/api/perms/admin'
+import {
+    adminLists,
+    adminDelete,
+    adminStatus,
+    adminExportFile,
+    adminImportFile
+} from '@/api/perms/admin'
 import { roleAll } from '@/api/perms/role'
 import { useDictOptions } from '@/hooks/useDictOptions'
 import { usePaging } from '@/hooks/usePaging'
@@ -135,6 +152,7 @@ const handleAdd = async () => {
     await nextTick()
     editRef.value?.open('add')
 }
+
 const exportFile = async () => {
     await feedback.confirm('确定要导出？')
     await adminExportFile(formData)
diff --git a/server/admin/system/admin/admin.go b/server/admin/system/admin/admin.go
index 7c09eb9..31c2803 100644
--- a/server/admin/system/admin/admin.go
+++ b/server/admin/system/admin/admin.go
@@ -1,6 +1,10 @@
 package admin
 
 import (
+	"bytes"
+	"fmt"
+	"io"
+	"net/http"
 	"strconv"
 	"x_admin/config"
 	"x_admin/core/request"
@@ -10,6 +14,7 @@ import (
 	"x_admin/util"
 
 	"github.com/gin-gonic/gin"
+	"github.com/xuri/excelize/v2"
 )
 
 // func AdminRoute(rg *gin.RouterGroup) {
@@ -67,6 +72,46 @@ func (ah AdminHandler) ExportFile(c *gin.Context) {
 	// f.Write(c.Writer)
 }
 
+// 导入文件
+func (ah AdminHandler) ImportFile(c *gin.Context) {
+	// file, err := c.FormFile("file")
+	// if err != nil {
+	// 	c.String(500, "上传文件出错")
+	// }
+	// fmt.Println(file)
+
+	// 单文件
+	file, _, err := c.Request.FormFile("file")
+	if err != nil {
+		response.FailWithMsg(c, response.SystemError, "文件不存在")
+		return
+	}
+	defer file.Close()
+	// 创建缓冲区
+	buf := new(bytes.Buffer)
+
+	// 将文件内容复制到缓冲区
+	_, err = io.Copy(buf, file)
+	if err != nil {
+		c.String(http.StatusInternalServerError, "Failed to read file")
+		return
+	}
+	// 创建Excel文件对象
+	f, err := excelize.OpenReader(bytes.NewReader(buf.Bytes()))
+	if err != nil {
+		c.String(http.StatusInternalServerError, "Failed to open Excel file")
+		return
+	}
+	importList := []SystemAuthAdminResp{}
+	err = excel.ImportExcel(f, &importList, 1, 2)
+	if err != nil {
+		fmt.Println(err)
+	}
+	for _, t := range importList {
+		fmt.Printf("%#v", t)
+	}
+}
+
 // list 管理员列表
 func (ah AdminHandler) List(c *gin.Context) {
 	var page request.PageReq
diff --git a/server/admin/system/enter.go b/server/admin/system/enter.go
index f92187d..70938da 100644
--- a/server/admin/system/enter.go
+++ b/server/admin/system/enter.go
@@ -32,6 +32,8 @@ func AdminRoute(rg *gin.RouterGroup) {
 
 	rg.GET("/admin/ExportFile", middleware.RecordLog("管理员导出"), handle.ExportFile)
 
+	rg.POST("/admin/ImportFile", handle.ImportFile)
+
 }
 func RoleRoute(rg *gin.RouterGroup) {
 	// db := core.GetDB()
diff --git a/server/main.go b/server/main.go
index dbda4f0..b18a1f5 100644
--- a/server/main.go
+++ b/server/main.go
@@ -29,6 +29,7 @@ func initRouter() *gin.Engine {
 	// 初始化gin
 	gin.SetMode(config.Config.GinMode)
 	router := gin.New()
+	router.MaxMultipartMemory = 8 << 20 // 8 MiB
 	// 设置静态路径
 	router.Static(config.Config.PublicPrefix, config.Config.UploadDirectory)
 
diff --git a/server/util/excel/excel_import.go b/server/util/excel/excel_import.go
index b41e21f..a62a8ae 100644
--- a/server/util/excel/excel_import.go
+++ b/server/util/excel/excel_import.go
@@ -30,29 +30,39 @@ func ImportBySheet(f *excelize.File, dst interface{}, sheetName string, headInde
 	return
 }
 
-// 判断数组中是否包含指定元素
-func IsContain(items interface{}, item interface{}) bool {
-	switch items.(type) {
-	case []int:
-		intArr := items.([]int)
-		for _, value := range intArr {
-			if value == item.(int) {
-				return true
-			}
+// 获取在数组中得下标
+func GetIndex(items []string, item string) int {
+	for i, v := range items {
+		if v == item {
+			return i
 		}
-	case []string:
-		strArr := items.([]string)
-		for _, value := range strArr {
-			if value == item.(string) {
-				return true
-			}
-		}
-	default:
-		return false
 	}
-	return false
+	return -1
 }
 
+// 判断数组中是否包含指定元素
+// func IsContain(items interface{}, item interface{}) bool {
+// 	switch items.(type) {
+// 	case []int:
+// 		intArr := items.([]int)
+// 		for _, value := range intArr {
+// 			if value == item.(int) {
+// 				return true
+// 			}
+// 		}
+// 	case []string:
+// 		strArr := items.([]string)
+// 		for _, value := range strArr {
+// 			if value == item.(string) {
+// 				return true
+// 			}
+// 		}
+// 	default:
+// 		return false
+// 	}
+// 	return false
+// }
+
 // 解析数据
 func importData(f *excelize.File, dst interface{}, sheetName string, headIndex, startRow int) (err error) {
 	rows, err := f.GetRows(sheetName) // 获取所有行
@@ -98,11 +108,12 @@ func importData(f *excelize.File, dst interface{}, sheetName string, headIndex,
 				}
 				cellValue = row[excelizeIndex] // 获取单元格的值
 			} else { // 否则根据表头名称来拿数据
-				if IsContain(heads, excelTag.Name) { // 当tag里的表头名称和excel表格里面的表头名称相匹配时
-					if i >= len(row) { // 防止下标越界
+				var index = GetIndex(heads, excelTag.Name)
+				if index != -1 {
+					if index >= len(row) { // 防止下标越界
 						continue
 					}
-					cellValue = row[i] // 获取单元格的值
+					cellValue = row[index] // 获取单元格的值
 				}
 			}
 			// 根据字段类型设置值
@@ -110,6 +121,12 @@ func importData(f *excelize.File, dst interface{}, sheetName string, headIndex,
 			case reflect.Int:
 				v, _ := strconv.ParseInt(cellValue, 10, 64)
 				newData.Field(i).SetInt(v)
+			case reflect.Uint:
+				v, _ := strconv.ParseUint(cellValue, 10, 64)
+				newData.Field(i).SetUint(v)
+			// case reflect.Uint8:
+			// 	v, _ := strconv.ParseUint(cellValue, 10, 64)
+			// 	newData.Field(i).SetUint(v)
 			case reflect.String:
 				newData.Field(i).SetString(cellValue)
 			}