异想家博客图片批量压缩程序

为了方便给自己的博客配图，用Golang写了一个脚本处理，现分享出来，有需要的朋友也可以参考修改使用。

压缩规则

1、图片都等比例压缩，不破坏长宽比。

2、如果是横屏图片，压缩到宽度为1280，高度适应。

3、如果是竖屏图片，压缩到高度为1000，宽度适应。

4、如果分辨率小于这个，不压缩。

5、支持png、jpg、jpeg。

使用方法

go build jfzBlogPicCompress.go

原图放在当前目录下的raw目录中，压缩后的图片生成在thumb目录下，运行生成的二进制文件即可，压缩完5s后程序退出。

源码

文件名jfzBlogPicCompress.go：

package main

import (
	"fmt"
	"github.com/nfnt/resize"
	"image"
	"image/jpeg"
	"image/png"
	"io"
	"log"
	"os"
	"path/filepath"
	"strings"
	"time"
)

type InputArgs struct {
	OutputPath string /** 输出目录 */
	LocalPath  string /** 输入的目录或文件路径 */
	Quality    int    /** 质量 */
	Width      int    /** 宽度尺寸，像素单位 */
	Format     string /** 格式 */
}

var inputArgs InputArgs

func main() {
	execute()
	fmt.Printf("处理完成，5秒后自动退出……")
	time.Sleep(5 * time.Second) /** 如果不是自己点击退出，延时5s */
}

//  执行压缩
func execute() {
	fmt.Println("开始批量压缩...")

	inputArgs.LocalPath = "./raw/"
	inputArgs.OutputPath = "./thumb/"
	inputArgs.Quality = 90
	inputArgs.Width = 1280
	fmt.Println("压缩规则：宽度1280，如果是竖图，高度1000  压缩质量：", inputArgs.Quality)

	GetFilelist(inputArgs.LocalPath)
	fmt.Println("图片保存在：" + inputArgs.OutputPath)
}

// 遍历输入原图目录的图片
func GetFilelist(path string) {
	/** 创建输出目录 */
	errC := os.MkdirAll(inputArgs.OutputPath, 0777)
	if errC != nil {
		fmt.Printf("%s", errC)
		return
	}
	err := filepath.Walk(path, func(pathFound string, f os.FileInfo, err error) error {
		if f == nil {
			return err
		}
		// 是否是目录
		if f.IsDir() {
			return nil
		}
		// 文件是否是图片
		localPath, format, _ := isPictureFormat(pathFound)
		outputPath := strings.Replace(localPath, "raw", "thumb", 1)
		if localPath != "" {
			if !imageCompress(
				func() (io.Reader, error) {
					return os.Open(localPath)
				},
				func() (*os.File, error) {
					return os.Open(localPath)
				},
				outputPath,
				inputArgs.Quality,
				inputArgs.Width,
				format) {
				fmt.Println("生成缩略图失败")
			} else {
				fmt.Println("生成缩略图成功 " + outputPath)
			}
		}
		return nil
	})
	if err != nil {
		fmt.Printf("输入的路径信息有误 %v
", err)
	}
}

// 压缩算法
func imageCompress(
	getReadSizeFile func() (io.Reader, error),
	getDecodeFile func() (*os.File, error),
	to string,
	Quality,
	base int,
	format string) bool {
	// 读取文件
	file_origin, err := getDecodeFile()
	defer file_origin.Close()
	if err != nil {
		fmt.Println("os.Open(file)错误")
		log.Fatal(err)
		return false
	}
	var origin image.Image
	var config image.Config
	var temp io.Reader

	// 读取尺寸
	temp, err = getReadSizeFile()
	if err != nil {
		fmt.Println("os.Open(temp)")
		log.Fatal(err)
		return false
	}
	var typeImage int64
	format = strings.ToLower(format)
	if format == "jpg" || format == "jpeg" {
		// jpg 格式 1
		typeImage = 1
		origin, err = jpeg.Decode(file_origin)
		if err != nil {
			fmt.Println("jpeg.Decode(file_origin)")
			log.Fatal(err)
			return false
		}
		temp, err = getReadSizeFile()
		if err != nil {
			fmt.Println("os.Open(temp)")
			log.Fatal(err)
			return false
		}
		config, err = jpeg.DecodeConfig(temp)
		if err != nil {
			fmt.Println("jpeg.DecodeConfig(temp)")
			return false
		}

	} else if format == "png" {
		// png 格式 0
		typeImage = 0
		origin, err = png.Decode(file_origin)
		if err != nil {
			fmt.Println("png.Decode(file_origin)")
			log.Fatal(err)
			return false
		}
		temp, err = getReadSizeFile()
		if err != nil {
			fmt.Println("os.Open(temp)")
			log.Fatal(err)
			return false
		}
		config, err = png.DecodeConfig(temp)
		if err != nil {
			fmt.Println("png.DecodeConfig(temp)")
			return false
		}
	}
	// 等比缩放（压缩到1280的宽，如果图片是竖着的，限定高最多1000）
	fixBase := base
	heightBase := fixBase * config.Height / config.Width
	if config.Height > config.Width {
		heightBase = 1000
		fixBase = heightBase * config.Width / config.Height
	}

	// 基准
	width := uint(fixBase)
	height := uint(heightBase)

	canvas := resize.Thumbnail(width, height, origin, resize.Lanczos3)
	file_out, err := os.Create(to)
	defer file_out.Close()
	if err != nil {
		log.Fatal(err)
		return false
	}
	if typeImage == 0 {
		err = png.Encode(file_out, canvas)
		if err != nil {
			fmt.Println("压缩图片失败")
			return false
		}
	} else {
		err = jpeg.Encode(file_out, canvas, &jpeg.Options{Quality})
		if err != nil {
			fmt.Println("压缩图片失败")
			return false
		}
	}

	return true
}

// 是否是图片
func isPictureFormat(path string) (string, string, string) {
	temp := strings.Split(path, ".")
	if len(temp) <= 1 {
		return "", "", ""
	}
	mapRule := make(map[string]int64)
	// 在这里可以添加其他格式
	mapRule["jpg"] = 1
	mapRule["JPG"] = 1
	mapRule["png"] = 1
	mapRule["PNG"] = 1
	mapRule["jpeg"] = 1
	mapRule["JPEG"] = 1
	if mapRule[temp[1]] == 1 {
		println(temp[1])
		return path, temp[1], temp[0]
	} else {
		return "", "", ""
	}
}

参考

代码参考了golang_image_compress，改为了适合自己博客使用，源程序的代码更通用，更推荐在它基础上修改。