iamge_manager.go

package main

import (
	"fmt"
	"os"
	"io/ioutil"
	"path"
	"strings"
	"syscall"
	"time"
	"path/filepath"
	"encoding/json"
	"text/tabwriter"
	"qsrdocker/container"

	log "github.com/sirupsen/logrus"
	"github.com/urfave/cli"
)

// imageCmd 关于 镜像的相关操作
var imageCmd = cli.Command {
		Name: "image",
		Usage: "qsrdocker image COMMAND",
		Subcommands: []cli.Command {
			imageLsCmd,
	},
}

// imageLsCmd 打印所有的镜像
var imageLsCmd = cli.Command { 
	Name: "ls",
	Usage: "List images",
	ArgsUsage: "[]",
	Action: func(context *cli.Context) error {
		listImage()
		return nil
	},
}

// listImage 显示镜像列表
func listImage() {
	// 创建反序列化载体
	var imageConfig map[string]map[string]string

	// 配置文件路径
	imageConfigPath := path.Join(container.ImageDir, container.ImageInfoFile)


	exist, err := container.PathExists(imageConfigPath)

	// 配置文件不存在时直接返回imageName
	if err != nil || !exist {

		log.Errorf("%v imageConfig is not exits", imageConfigPath)
		return
	}

	//ReadFile函数会读取文件的全部内容，并将结果以[]byte类型返回
	data, err := ioutil.ReadFile(imageConfigPath)
	if err != nil {
		log.Errorf("imageConfig Can't open imageConfig : %v", imageConfigPath)
		return
	}

	//读取的数据为json格式，需要进行解码
	err = json.Unmarshal(data, &imageConfig)
	if err != nil {
		log.Debugf("Can't Unmarshal : %v", imageConfigPath)
		return
	}

	// 使用 tabwriter.NewWriter 在 终端 打出容器信息，打印对齐的表格
	w := tabwriter.NewWriter(os.Stdout, 20, 1, 3, ' ', 0)
	fmt.Fprint(w, "IMAGE NAME\tTAG\tIMAGE ID\tSIZE\tCREATE TIME\n")

	for imageName, imageTagMap := range imageConfig {
		for imageTag, imageLower := range imageTagMap {
			imageLowers := strings.Split(imageLower, ":")
			fmt.Fprintf( w, "%s\t%s\t%s\t%s\t%s\n",
				imageName,
				imageTag,
				imageLowers[0],
				getImageSize(imageLowers),
				getCreateTime(imageLowers[0]),
			)
		}
	}

	if err := w.Flush(); err != nil {
		log.Errorf("Flush error %v", err)
	}
}


// getImageSize 获取image
func getImageSize(imageLowers []string) string {
	
	// 从零开始技术
	imageSizeByte := int64(0) 

	for _, imageID := range imageLowers {
		
		imagePath := path.Join(container.ImageDir, imageID)

		if exist, _ := container.PathExists(imagePath); exist{
			// 遍历目录获取大小
			// Walk 自动遍历所有目录 子目录
			filepath.Walk(imagePath, func(_ string, info os.FileInfo, err error) error {
				if !info.IsDir() {
					imageSizeByte += info.Size()
				}
				return err
			})		
		} else {

			// 镜像还未被使用的情况
			imageTarPath := strings.Join([]string{imagePath, ".tar"},"")

			// 获取 log 文件信息
			fileInfo, err := os.Stat(imageTarPath)

			if err != nil {
				log.Errorf("Get %v info err : %v", imageTarPath, err) 
			}

			imageSizeByte += fileInfo.Size()
		}
		

	}

	switch {
		case imageSizeByte >=  int64(1024*1024*1024): return fmt.Sprintf("%.2f GB", float64(imageSizeByte)/(1024.0*1024.0*1024.0))
		case imageSizeByte >=  int64(1024*1024): return fmt.Sprintf("%.2f MB", float64(imageSizeByte)/(1024.0*1024.0))
		case imageSizeByte >=  int64(1024): return fmt.Sprintf("%.2f KB", float64(imageSizeByte)/1024.0)
		case imageSizeByte >=  int64(0): return fmt.Sprintf("%d B", imageSizeByte)
	}
	
	return ""
}

// getCreateTime 获取文件创建时间
func getCreateTime(imageID string) string {

	imagePath := path.Join(container.ImageDir, imageID)

	if exist, _ := container.PathExists(imagePath); !exist{
		// 存在 image 未使用
		imagePath = strings.Join([]string{imagePath, ".tar"},"")

	}

	fileInfo, err := os.Stat(imagePath)
	
	if err != nil {
		log.Errorf("Get %v info err : %v", imagePath, err) 
	}

	creatUnixTime := fileInfo.Sys().(*syscall.Stat_t).Ctim

	return time.Unix(creatUnixTime.Sec, creatUnixTime.Nsec).Format("2006-01-02 15:04:05")

}