Golang の画像操作: 切断や画像の欠落を検出して修復する方法

import (
    "fmt"
    "os"
    "io"
)

func IsImageBroken(file string) bool {
    f, err := os.Open(file)
    if err != nil {
        fmt.Println(err)
        return true
    }
    defer f.Close()

    buf := make([]byte, 512)
    _, err = f.Read(buf)
    if err != nil && err != io.EOF {
        fmt.Println(err)
        return true
    }

    fileType := http.DetectContentType(buf)
    if fileType == "image/jpeg" || fileType == "image/png" {
        return false
    } else {
        return true
    }
}

func main() {
    file := "example.jpg"
    if IsImageBroken(file) {
        fmt.Println("图片断线")
    } else {
        fmt.Println("图片完整")
    }
}

import (
    "fmt"
    "image"
    "io/ioutil"
    "os"
)

func FixImageMissing(file string) error {
    f, err := ioutil.ReadFile(file)
    if err != nil {
        fmt.Println(err)
        return err
    }

    img, _, err := image.Decode(bytes.NewReader(f))
    if err != nil {
        fmt.Println(err)
        return err
    }

    out, err := os.Create(file)
    if err != nil {
        fmt.Println(err)
        return err
    }
    defer out.Close()

    err = jpeg.Encode(out, img, nil)
    if err != nil {
        fmt.Println(err)
        return err
    }

    return nil
}

func main() {
    file := "example.jpg"
    err := FixImageMissing(file)
    if err != nil {
        fmt.Println("修复失败")
    } else {
        fmt.Println("修复成功")
    }
}