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

Posted 2021-03-13 sandeepin

为了方便给自己的博客配图，用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，改为了适合自己博客使用，源程序的代码更通用，更推荐在它基础上修改。