vue+elementui导入Excel文件(基于vue-element-admin中的uploadExcel组件), 然后 go-zero进行逻辑处理功能

7 篇文章 0 订阅

#前端代码

html代码

<el-button v-waves class="filter-item" type="primary" icon="add"  @click="downLoadExlce">模板下载</el-button>
<el-upload class="upload" action :multiple="false" :show-file-list="false"   accept=".xls"  :drag="false" :http-request="importHandle">
  <el-button slot="trigger"  type="primary">批量导入</el-button>
 </el-upload>

js代码

import { uploadConfig } from 'upload'
downLoadExlce () {
      var hrefStr = encodeURI('下载地址的url');
      window.location.href = hrefStr;
    },
  // 导入方法
importHandle(item) {
      const form = new FormData();
      const fileObj = item.file;
      console.log(fileObj);
      form.append('file', fileObj);

      uploadConfig(form).then(() => {
        // 其他逻辑处理, 比如重新获取页面列表接口
        this.$notify({
          title: '成功',
          message: '更新成功',
          type: 'success'
        })
      })
    },    
    

请求接口的upload.js

import request from '@/utils/request'
/**
 * 导入上传文件
 * @param data
 */
export function uploadConfig(data) {
    return request({
        url: '/api/upload/fundConfig',
        method: 'post',
        headers: {
            'Content-Type': 'multipart/form-data'
        },
        data
    })
}

上传的模板

在这里插入图片描述

go-zero 处理逻辑

配置一个config.api文件, 然后执行生成命令

syntax = "v1"

info(
    title: "配置"
    author: ""
)

type (
    UploadResp struct {
        Code string `json:"code"`
        Message string `json:"message"`
    }


)

@server(
    prefix: /api/upload
)
service admin-api {

    @doc "广告金配置导入"
    @handler UploadConfig
    post /fundConfig  returns (UploadResp)
}

找到对应的handler 目录下 uploadconfighandler.go 代码

原来的代码
l := internalConfig.NewUploadConfigLogic(r.Context(), svcCtx)
改为 
l := internalConfig.NewUploadConfigLogic(r, r.Context(), svcCtx)

找到对应的logic目录下 uploadconfiglogic.go 代码 部分逻辑

import (
	"context"
	"fmt"
	"net/http"
	"strconv"
	"time"
	"github.com/xuri/excelize/v2"    # 这个是关键
	"github.com/zeromicro/go-zero/core/logx"
)

type UploadConfigLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
	r      *http.Request
}

func NewAdFundUploadLogic(r *http.Request, ctx context.Context, svcCtx *svc.ServiceContext) *UploadConfigLogic{
	return &UploadConfigLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
		r:      r,
	}
}

func (l *UploadConfigLogic) UploadConfig() (resp *types.UploadResp, err error) {

	file, handler, err := l.r.FormFile("file")
	if err != nil {
		return nil, errorx.NewDefaultError("上传失败:" + err.Error())
	}

	defer file.Close()

	logx.Infof("upload file: %+v, file size: %d, MIME header: %+v",
		handler.Filename, handler.Size, handler.Header)

	//读excel流
	xlsx, errs := excelize.OpenReader(file)
	if errs != nil {
		return nil, errorx.NewDefaultError("open excel error:[%s]" + err.Error())
	}

	//解析excel的数据
	if err2 := l.ReadExcel(xlsx); err2 != nil {
		return nil, errorx.NewDefaultError("上传失败②:" + err2.Error())
	}

	return &types.UploadResp{
		Code:    "000000",
		Message: "添加成功",
	}, nil
}

//Employee .
type Employee struct {
	StatDatetime time.Time `db:"stat_datetime"` // 日期。格式:yyyy-MM-dd
	StatDate     string    `db:"stat_date"`     // 日期。格式:yyyy-MM-dd
	Appid        string    `db:"appid"`         // 小程序appid
	GameName     string    `db:"game_name"`     // 游戏名称
	AdFund       float64   `db:"ad_fund"`       // 广告金
}

//ReadExcel .
func (l *UploadConfigLogic) ReadExcel(xlsx *excelize.File) error {
	//根据名字获取cells的内容,返回的是一个[][]string
	rows, _ := xlsx.GetRows(xlsx.GetSheetName(xlsx.GetActiveSheetIndex()))
	//声明一个数组
	var datas []Employee
	for i, row := range rows {
		// 去掉第一行是execl表头部分
		if i == 0 {
			continue
		}
		var data Employee
		for k, v := range row {
			// 第一列是日期
			if k == 0 {
				statDatetime2 := DateStringToDate2(v, "2006-01-02")
				data.StatDatetime = statDatetime2
				data.StatDate = v
			}
			// 第二列是Appid
			if k == 1 {
				data.Appid = v
			}
			// 第三列是游戏名称
			if k == 2 {
				data.GameName = v
			}
			// 第四列是广告金
			if k == 3 {
				// string 转float64
				n, _ := strconv.ParseFloat(v, 64)
				data.AdFund = n
			}

		}
		//将数据追加到datas数组中
		datas = append(datas, data)
	}

	//logx.Info("datas:", datas)
	//循环遍历datas数组
	for k, employee := range datas {
		key := k + 1
		appid := employee.Appid
		statDate := employee.StatDate
		gameName := employee.GameName
		statDatetime2 := employee.StatDatetime
		adFund := employee.AdFund
		configOne, errs := l.svcCtx.GameAdFundConfigModel.FindByAppidStatDatetime(appid, statDate)
		if len(statDate) == 0 {
			return errorx.NewDefaultError(fmt.Sprintf("第 %s 行,没有填写日期", key))
		}
		if len(appid) == 0 {
			return errorx.NewDefaultError(fmt.Sprintf("第 %s 行,没有填写appid", key))
		}
		if len(gameName) == 0 {
			return errorx.NewDefaultError(fmt.Sprintf("第 %s 行,没有填写应用名称", key))
		}
		if adFund < 0 {
			return errorx.NewDefaultError(fmt.Sprintf("第 %s 行,广告金不能为负数", key))
		}

	   # 处理落库逻辑
	   
	}
	return nil
}


//2022-05-10 字符串 转为 时间类型的 2022-05-10或者20220510
func DateStringToDate2(timeStr string, layout string) time.Time {
	tt, _ := time.Parse(layout, timeStr)
	return tt
}

excelize 安装与使用

Excelize官方文档

安装命令

go get github.com/xuri/excelize
如果您使用 Go Modules 管理软件包,请使用下面的命令来安装最新版本。
go get github.com/xuri/excelize/v2

创建 Excel 文档

下面是一个创建 Excel 文档的简单例子:

package main
import (
    "fmt"
    "github.com/xuri/excelize/v2"
)

func main() {
    f := excelize.NewFile()
    // 创建一个工作表
    index := f.NewSheet("Sheet2")
    // 设置单元格的值
    f.SetCellValue("Sheet2", "A2", "Hello world.")
    f.SetCellValue("Sheet1", "B2", 100)
    // 设置工作簿的默认工作表
    f.SetActiveSheet(index)
    // 根据指定路径保存文件
    if err := f.SaveAs("Book1.xlsx"); err != nil {
        fmt.Println(err)
    }
}

读取 Excel 文档

下面是读取 Excel 文档的例子:

package main

import (
    "fmt"

    "github.com/xuri/excelize/v2"
)

func main() {
    f, err := excelize.OpenFile("Book1.xlsx")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer func() {
        if err := f.Close(); err != nil {
            fmt.Println(err)
        }
    }()
    // 获取工作表中指定单元格的值
    cell, err := f.GetCellValue("Sheet1", "B2")
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(cell)
    // 获取 Sheet1 上所有单元格
    rows, err := f.GetRows("Sheet1")
    if err != nil {
        fmt.Println(err)
        return
    }
    for _, row := range rows {
        for _, colCell := range row {
            fmt.Print(colCell, "\t")
        }
        fmt.Println()
    }
}

个人捣鼓的小程序,欢迎多提提意见:
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值