1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

func CheckImageFile(path, style string) (string, error) {

    f, err := os.Open(path)

    if err != nil {

        fmt.Errorf("打开文件失败 %s", err.Error())

    }

    switch strings.ToUpper(style) {

    case "JPG", "JPEG":

        _, err = jpeg.Decode(f)

    case "PNG":

        _, err = png.Decode(f)

    case "GIF":

        _, err = gif.Decode(f)

    }

    if err != nil {

        fmt.Errorf("校验文件类型失败 %s", err.Error())

        return "", err

    }

    return "", nil

}