Victorique/vtqe/attach.go

package main

import (
	"io"
	"os"

	"b612.me/starlog"
	"github.com/spf13/cobra"
)

var attachcmd = &cobra.Command{
	Use:   "attach",
	Short: "合并多个文件",
	Long:  "合并多个文件",
	Run: func(this *cobra.Command, args []string) {
		src, _ := this.Flags().GetStringArray("src")
		out, _ := this.Flags().GetString("out")
		if len(src) < 2 || out == "" {
			starlog.Criticalln("请输入至少2个输入路径，一个输出路径")
			os.Exit(1)
		}
		os.Exit(runAttach(src, out))
	},
}

func runAttach(src []string, out string) int {
	fpOut, err := os.Create(out)
	if err != nil {
		starlog.Errorln("Err:无法创建输出文件！", err)
		return 2
	}
	defer fpOut.Close()
	for _, file := range src {
		fpFile, err := os.OpenFile(file, os.O_RDONLY, 0644)
		if err != nil {
			starlog.Errorln("Err:Cannot Openfile:", err)
			return 3
		}
		stats, err := fpFile.Stat()
		if err != nil {
			starlog.Errorln("Err:Cannot Get File Stats:", err)
			return 4
		}
		if stats.Size() == 0 {
			starlog.Warningf("文件:%s为空文件，跳过！\n", stats.Name())
			continue
		}
		buf := make([]byte, 65535)
		sumAll := 0
		for {
			n, err := fpFile.Read(buf)
			if err != nil && err != io.EOF {
				starlog.Errorln("Err:Error Occured While ReadFile:", err)
				fpFile.Close()
				return 5
			}
			sumAll += n
			_, errW := fpOut.Write(buf[:n])
			if errW != nil {
				starlog.Errorln("Error While Write Data to OutFile", errW)
				return 6
			}
			starlog.StdPrintf([]starlog.Attr{starlog.FgGreen}, "文件%s，已完成：%.2f%%\r", stats.Name(), (float64(sumAll) / float64(stats.Size()) * 100.0000))
			if err == io.EOF {
				starlog.StdPrintf([]starlog.Attr{starlog.FgGreen}, "文件：%v，已完成：100.00%% \n", stats.Name())
				fpFile.Close()
				break
			}
		}
	}
	starlog.StdPrintln([]starlog.Attr{starlog.FgGreen}, "Ok!文件合并完成")
	return 0
}

func init() {
	attachcmd.Flags().StringArrayP("src", "s", []string{}, "源文件路径")
	attachcmd.Flags().StringP("out", "o", "", "输出文件路径")
}