用go语言编写办公脚本——根据txt内容搜索对应文件

本文链接：https://blog.csdn.net/isluckyboy/article/details/125986414

用go语言编写办公脚本——根据txt内容搜索对应文件

最近工作中，需要处理大量的漏扫报告，需要根据不同的标准来将混在一起的报告分开，公司有个大佬直接写了.bat的批处理脚本，感叹一句，真强啊！！但是这个批处理脚本有一些玄学bug，大佬也没改出来，正好我最近想把go语言再复习复习，就用这个练练手吧。

需求

将文件名关键字筛选出来，存放到list.txt中，每行一个关键字，将混在一起的大量文件放到文件夹src中，运行脚本，将文件名包含关键字的文件放到result文件夹中。

源码

上源码！！！

····package main

import (
   "fmt"
	"os"
   "io/ioutil"
   "strings"
   "io"
)
	
//读取txt中需要查找的内容
func readfile(fileName string) (resultArray []string){
   //打开文件
   file, err := os.Open(fileName)
    if err != nil {
        panic(err)
    }
    defer file.Close()
    //读取文件内容
    content, err2 :=ioutil.ReadAll(file)
    if err2 != nil {
        panic(err2)
    }
    //fmt.Println(string(content))
    //按照回车将文件内容分割，返回一个切片
    result := strings.Split(string(content), "\n")
      //fmt.Println(result)

    return result
}

//获取src文件夹中的文件名
func readFilePath(filePath string) (filenameList []string){
   //根据文件夹路径打开文件夹
   files, _ := ioutil.ReadDir(filePath)
   //定义切片
   var filePathList []string
   //将文件名放到切片中
    for _, f := range files {
       filePathList = append(filePathList, f.Name())
    }
   return filePathList
}

//在目标文件夹中查找文件
func searchFileFromList(filenameList []string,fileList []string){
   //双重循环，filename是txt中的关键字，file是src中的文件名
   for _, filename := range filenameList {
      for _, file := range fileList {
         //字符串匹配
         if strings.Contains(file, filename) {
            //构建文件路径
            src := "./src/" + file
            dst := "./result/" + file
            //将文件复制到新文件夹
            _, err := copyFile(dst, src)
	         if err != nil {
		         fmt.Printf("copy fail:%v ", err)
		         return
	         }
	         fmt.Println(file + " copy succeed")
         }
      }
   }
}

//复制文件
func copyFile(dstName, srcName string) (written int64, err error) {
	//打开文件读取内容
   src, err := os.Open(srcName)
	if err != nil {
		fmt.Printf("open %s failed,err:%v.\n", srcName, err)
		return
	}
	defer src.Close()
   //打开目标文件
	dst, err := os.OpenFile(dstName, os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		fmt.Printf("open %s failed,err:%v.\n", dstName, err)
		return
	}
	defer dst.Close()
   //将源文件内容copy到目标文件
	return io.Copy(dst, src)
}

func main() {
   //调用函数
   searchFileFromList(readfile("list.txt"),readFilePath("./src/"))
}