diff options
Diffstat (limited to 'internal/image')
| -rw-r--r-- | internal/image/handler.go | 5 | ||||
| -rw-r--r-- | internal/image/service.go | 1 |
2 files changed, 3 insertions, 3 deletions
diff --git a/internal/image/handler.go b/internal/image/handler.go index f41ed4d..71a5383 100644 --- a/internal/image/handler.go +++ b/internal/image/handler.go @@ -35,9 +35,8 @@ func (h *imageHandler) GetImage(w http.ResponseWriter, r *http.Request) { fileID := chi.URLParam(r, "fileName") fileMeta, err := h.service.GetFile(fileID) if err != nil { - w.WriteHeader(400) - log.Printf("error: %+v", err) - w.Write([]byte("WTF Incorrect Content Type")) + http.Error(w, "an error has occured", http.StatusBadRequest) + return } fileUrl := fmt.Sprintf("/f/%s", fileMeta.FilePath) data := struct { diff --git a/internal/image/service.go b/internal/image/service.go index 10f4148..2a60c67 100644 --- a/internal/image/service.go +++ b/internal/image/service.go @@ -78,6 +78,7 @@ func (is *Service) AddFile(extension string, postMeta *PostMeta, fileBytes []byt func (is *Service) GetFile(fileUrl string) (*PostMeta, error) { result, err := is.db.GetFile(fileUrl) if err != nil { + return &PostMeta{}, err } return result, err } |