GO语言实现批量压缩图片和水印

2019-11-10 09:49:33刘景俊

前段时间想做个图片站,就用手机照了很多相片,但是要一个个用PS去压缩修改尺寸太麻烦了。最后想到了用golang去实现,算是边学边练吧。其中用到了github.com/nfnt/resize这个第三方库,仅仅支持JPG图片格式。


package main
import (
    "fmt"
    "github.com/nfnt/resize"
    "image"
    "image/draw"
    "image/jpeg"
    "image/png"
    "io/ioutil"
    "log"
    "math/rand"
    "os"
    "runtime"
    "strconv"
    "strings"
    "time"
)
func main() {
    runtime.GOMAXPROCS(runtime.NumCPU())
    cmd("data/")
    fmt.Println("OK!")
}
// 执行操作
func cmd(path string) {
    files, _ := ioutil.ReadDir(path)
    for _, file := range files {
        if file.IsDir() {
            fmt.Println("目录" + file.Name())
            cmd(path + file.Name() + "/")
        } else {
            if strings.Contains(strings.ToLower(file.Name()), ".jpg") {
                // 随机名称
                to := path + random_name() + ".jpg"
                origin := path + file.Name()
                fmt.Println("正在处理" + origin + ">>>" + to)
                cmd_resize(origin, 2048, 0, to)
                defer os.Remove(origin)
            }
        }
    }
}
// 改变大小
func cmd_resize(file string, width uint, height uint, to string) {
    // 打开图片并解码
    file_origin, _ := os.Open(file)
    origin, _ := jpeg.Decode(file_origin)
    defer file_origin.Close()
    canvas := resize.Resize(width, height, origin, resize.Lanczos3)
    file_out, err := os.Create(to)
    if err != nil {