Adding caching to profile images to master (#5428)

* Adding caching to profile images

* Fixing break

* Adding back in failed read

* Fixing build break

* Fixing break
This commit is contained in:
Corey Hulen
2017-02-15 18:54:41 -05:00
committed by Joram Wilander
parent db2966b7cb
commit 1d6ea40022
2 changed files with 18 additions and 8 deletions

View File

@@ -590,26 +590,33 @@ func getAudits(c *Context, w http.ResponseWriter, r *http.Request) {
func getProfileImage(c *Context, w http.ResponseWriter, r *http.Request) { func getProfileImage(c *Context, w http.ResponseWriter, r *http.Request) {
params := mux.Vars(r) params := mux.Vars(r)
id := params["user_id"] id := params["user_id"]
readFailed := false
var etag string var etag string
if user, err := app.GetUser(id); err != nil { if users, err := app.GetUsersByIds([]string{id}, false); err != nil {
c.Err = err c.Err = err
return return
} else { } else {
if len(users) == 0 {
c.Err = model.NewLocAppError("getProfileImage", "store.sql_user.get_profiles.app_error", nil, "")
return
}
user := users[0]
etag = strconv.FormatInt(user.LastPictureUpdate, 10) etag = strconv.FormatInt(user.LastPictureUpdate, 10)
if HandleEtag(etag, "Profile Image", w, r) { if HandleEtag(etag, "Profile Image", w, r) {
return return
} }
var img []byte var img []byte
img, err = app.GetProfileImage(user) img, readFailed, err = app.GetProfileImage(user)
if err != nil { if err != nil {
c.Err = err c.Err = err
return return
} }
if c.Session.UserId == id { if readFailed {
w.Header().Set("Cache-Control", "max-age=300, public") // 5 mins w.Header().Set("Cache-Control", "max-age=300, public") // 5 mins
} else { } else {
w.Header().Set("Cache-Control", "max-age=86400, public") // 24 hrs w.Header().Set("Cache-Control", "max-age=86400, public") // 24 hrs

View File

@@ -667,25 +667,28 @@ func CreateProfileImage(username string, userId string) ([]byte, *model.AppError
} }
} }
func GetProfileImage(user *model.User) ([]byte, *model.AppError) { func GetProfileImage(user *model.User) ([]byte, bool, *model.AppError) {
var img []byte var img []byte
readFailed := false
if len(utils.Cfg.FileSettings.DriverName) == 0 { if len(utils.Cfg.FileSettings.DriverName) == 0 {
var err *model.AppError var err *model.AppError
if img, err = CreateProfileImage(user.Username, user.Id); err != nil { if img, err = CreateProfileImage(user.Username, user.Id); err != nil {
return nil, err return nil, false, err
} }
} else { } else {
path := "users/" + user.Id + "/profile.png" path := "users/" + user.Id + "/profile.png"
if data, err := ReadFile(path); err != nil { if data, err := ReadFile(path); err != nil {
readFailed = true
if img, err = CreateProfileImage(user.Username, user.Id); err != nil { if img, err = CreateProfileImage(user.Username, user.Id); err != nil {
return nil, err return nil, false, err
} }
if user.LastPictureUpdate == 0 { if user.LastPictureUpdate == 0 {
if err := WriteFile(img, path); err != nil { if err := WriteFile(img, path); err != nil {
return nil, err return nil, false, err
} }
} }
@@ -694,7 +697,7 @@ func GetProfileImage(user *model.User) ([]byte, *model.AppError) {
} }
} }
return img, nil return img, readFailed, nil
} }
func SetProfileImage(userId string, imageData *multipart.FileHeader) *model.AppError { func SetProfileImage(userId string, imageData *multipart.FileHeader) *model.AppError {