嵌入式linux之go语言开发（六）几行代码实现终端的远程日志诊断

package main

import (
	"archive/zip"
	"flag"
	"fmt"
	"io"
	"log"
	"os"
	"strings"

	"github.com/dutchcoders/goftp"
	"github.com/larspensjo/config"
)

var (
	ftp     *goftp.FTP
	conFile        = flag.String("ftpcfg", "/ftpcfg.ini", "config file")
	Server  string = "127.0.0.1:21"
	User    string = ""
	Pwd     string = ""
)

func checkErr(err error) {
	if err != nil {
		fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
	}
}

/**
@files：需要压缩的文件
@compreFile：压缩之后的文件
*/
func CompressZip(files []*os.File, zipfileName string) (err error) {
	zipfile, err := os.Create(zipfileName)
	if err != nil {
		return err
	}
	defer zipfile.Close()

	zw := zip.NewWriter(zipfile)
	defer zw.Close()
	for _, file := range files {
		err := compressZip(file, zw)
		if err != nil {
			return err
		}
		file.Close()
	}
	return nil
}

/**
功能：压缩文件
@file:压缩文件
@prefix：压缩文件内部的路径
@tw：写入压缩文件的流
*/
func compressZip(file *os.File, zw *zip.Writer) error {
	info, err := file.Stat()
	if err != nil {
		log.Println("压缩文件失败：", err.Error())
		return err
	}
	log.Println("filesize:", info.Size()/1024, "kb")
	// 获取压缩头信息
	head, err := zip.FileInfoHeader(info)
	if err != nil {
		log.Println("压缩文件失败：", err.Error())
		return err
	}
	// 指定文件压缩方式 默认为 Store 方式 该方式不压缩文件 只是转换为zip保存
	head.Method = zip.Deflate
	fw, err := zw.CreateHeader(head)
	if err != nil {
		log.Println("压缩文件失败：", err.Error())
		return err
	}
	// 写入文件到压缩包中
	_, err = io.Copy(fw, file)
	file.Close()
	if err != nil {
		log.Println("压缩文件失败：", err.Error())
		return err
	}
	return nil
}

func main() {

	log.Println("Hello Go")
	defer func() {
		if r := recover(); r != nil {
			log.Printf("crash occurred!capture：%s\n", r)
		}
	}()
	var fname string
	flag.StringVar(&fname, "fname", "", "file name")
	flag.Parse()
	flag.Usage()
	//获取当前路径
	file, _ := os.Getwd()
	cfg, err := config.ReadDefault(file + *conFile)
	checkErr(err)
	//获取配置文件中的配置项
	Server, err = cfg.String("SERVERCONFIG", "Server")

	User, err = cfg.String("USER", "User")
	Pwd, err = cfg.String("USER", "Pwd")

	//压缩ZIP
	var filezips []*os.File
	var filezip *os.File
	if filezip, err = os.Open(fname); err != nil {
		panic(err)
	}
	filezips = append(filezips, filezip)
	log.Println("begin compressZip...")
	err = CompressZip(filezips, fname+".zip")
	if err != nil {
		panic(err)
	}
	log.Println("compressZip ok!,name=" + fname + ".zip")
	log.Println("User:" + User)
	log.Println("->begin connect server:" + Server)
	// For debug messages: goftp.ConnectDbg("ftp.server.com:21")
	if ftp, err = goftp.ConnectDbg(Server); err != nil {
		panic(err)
	}
	//ftp.debug = true
	defer ftp.Close()
	log.Println("->Successfully connected !!")

	// Username / password authentication
	if err = ftp.Login(User, Pwd); err != nil {
		panic(err)
	}
	log.Println("->Login success!")
	if err = ftp.Cwd("/"); err != nil {
		panic(err)
	}

	var curpath string
	if curpath, err = ftp.Pwd(); err != nil {
		panic(err)
	}

	log.Printf("Current path: %s\n", curpath)

	// Upload a file
	var fileup *os.File
	if fileup, err = os.Open(fname + ".zip"); err != nil {
		panic(err)
	}
	log.Println("->begin upload file...")
	info, err := fileup.Stat()
	if err != nil {
		panic(err)
	}
	log.Println("upload filesize:", info.Size()/1024, "kb")
	fpath := fname + ".zip"
	lastidx := strings.LastIndex(fname+".zip", "/")
	if lastidx != -1 {
		fpath = fname[lastidx:] + ".zip"
	}
	log.Printf("fpath:%s\n", fpath)
	if err := ftp.Stor(fpath, fileup); err != nil {
		panic(err)
	}
	log.Println("->upload file over!")
	err = os.Remove(fname + ".zip") //上传成功，删除zip文件
	if err != nil {
		log.Println("file remove Error!")
		panic(err)
	}

}