1package main
 2
 3import (
 4	"bytes"
 5	"fmt"
 6	"image"
 7	"image/jpeg"
 8	"image/png"
 9)
10
11func dataToImage(data []byte, imageExtension string) (image.Image, error) {
12	reader := bytes.NewReader(data)
13	//img, err := png.Decode(reader)
14	var img image.Image
15	var err error
16	switch imageExtension {
17	case "png":
18		img, err = png.Decode(reader)
19	case "jpg", "jpeg":
20		img, err = jpeg.Decode(reader)
21	default:
22		img = nil
23	}
24	if err != nil {
25		fmt.Println(err)
26		return img, err
27	}
28	return img, err
29}