manage errors (9403f844) · Commits · blackheaven / tinygraphs

app-backend/images.go

+11 −7

Original line number	Diff line number	Diff line
		@@ -59,14 +59,18 @@ func writeImageWithTemplate(w http.ResponseWriter, img *image.Image) {

		buffer := new(bytes.Buffer)
		if err := jpeg.Encode(buffer, *img, nil); err != nil {
		log.Fatalln("unable to encode image")
		log.Fatalln("unable to encode image.")
		}

		str := base64.StdEncoding.EncodeToString(buffer.Bytes())
		var tmpl, _ = template.New("image").Parse(ImageTemplate)

		if tmpl, err := template.New("image").Parse(ImageTemplate); err != nil {
		log.Fatalln("unable to parse image template.")
		} else {
		data := map[string]interface{}{"Image": str}
		tmpl.Execute(w, data)
		if err = tmpl.Execute(w, data); err != nil {
		log.Fatalln("unable to execute template.")
		}
		}
		}

		// writeImage encodes an image 'img' in jpeg format and writes it into ResponseWriter.
		@@ -74,12 +78,12 @@ func writeImage(w http.ResponseWriter, img *image.Image) {

		buffer := new(bytes.Buffer)
		if err := jpeg.Encode(buffer, *img, nil); err != nil {
		log.Fatalln("unable to encode image")
		log.Fatalln("unable to encode image.")
		}

		w.Header().Set("Content-Type", "image/jpeg")
		w.Header().Set("Content-Length", strconv.Itoa(len(buffer.Bytes())))
		if _, err := w.Write(buffer.Bytes()); err != nil {
		log.Fatalln("unable to write image")
		log.Fatalln("unable to write image.")
		}
		}