modified: serv_nginx/api_bb/internal/handlers/avatar.go

modified:   serv_nginx/api_bb/internal/handlers/user.go
	modified:   serv_nginx/api_bb/internal/routes/routes.go
	modified:   serv_nginx/api_bb/internal/service/avatar_service.go
set all avatars manipulating into avatar.go and remove from user.go
This commit is contained in:
2025-10-16 13:14:27 +05:00
parent e1ccf83569
commit d9535ac053
4 changed files with 17 additions and 84 deletions
+11 -4
View File
@@ -3,6 +3,7 @@ package handlers
import (
"net/http"
"strings"
"time"
"api_bb/internal/service"
@@ -35,7 +36,6 @@ func (h *AvatarHandler) Routes() chi.Router {
zap.String("GET", "/{filename}"),
)
// r.Get("/avatar/{filename}", h.ServeAvatar)
r.Post("/upload", h.UploadAvatar)
r.Delete("/delete", h.DeleteAvatar)
r.Get("/{filename}", h.GetAvatar)
@@ -181,7 +181,7 @@ func (h *AvatarHandler) DeleteAvatar(w http.ResponseWriter, r *http.Request) {
})
}
// GET /v1/user/avatar/avatar_22_1760417314.png
// GET /v1/user/avatars/{filename}
func (h *AvatarHandler) GetAvatar(w http.ResponseWriter, r *http.Request) {
filename := chi.URLParam(r, "filename")
@@ -199,13 +199,20 @@ func (h *AvatarHandler) GetAvatar(w http.ResponseWriter, r *http.Request) {
)
}()
// Валидация имени файла
if filename == "" || strings.Contains(filename, "..") || strings.Contains(filename, "/") {
h.logger.Warn("GetAvatar: invalid filename", zap.String("filename", filename))
utils.RespondWithError(w, http.StatusBadRequest, "Invalid filename")
return
}
h.logger.Info("GetAvatar: handling get avatar request",
zap.String("method", r.Method),
zap.String("filename", filename),
zap.String("remote_addr", r.RemoteAddr),
)
// Вариант 1: Используем ServeAvatarFile (более эффективно для больших файлов)
// Используем ServeAvatarFile для обслуживания файла
h.logger.Debug("GetAvatar: calling avatarService.ServeAvatarFile",
zap.String("filename", filename),
)
@@ -233,7 +240,7 @@ func (h *AvatarHandler) GetAvatar(w http.ResponseWriter, r *http.Request) {
return
}
// Устанавливаем заголовки
// Устанавливаем заголовки для кэширования
h.logger.Debug("GetAvatar: setting response headers",
zap.String("content_type", contentType),
)
+1 -76
View File
@@ -4,12 +4,8 @@ package handlers
import (
"bytes"
"encoding/json"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"strings"
"time"
"api_bb/internal/models"
@@ -39,8 +35,7 @@ func (h *UserHandler) Routes() chi.Router {
r.Get("/profile", h.GetProfile)
r.Post("/editProfile", h.UpdateProfile)
r.Get("/avatars/{filename}", h.ServeAvatar)
// Убрали маршрут для обслуживания аватаров - теперь это делает AvatarHandler
return r
}
@@ -180,73 +175,3 @@ func (h *UserHandler) UpdateProfile(w http.ResponseWriter, r *http.Request) {
"user": toUserResponse(updatedUser),
})
}
func (h *UserHandler) ServeAvatar(w http.ResponseWriter, r *http.Request) {
filename := chi.URLParam(r, "filename")
h.logger.Info("handling serve avatar request",
zap.String("method", r.Method),
zap.String("filename", filename),
zap.String("remote_addr", r.RemoteAddr),
)
// Валидация
if filename == "" || strings.Contains(filename, "..") || strings.Contains(filename, "/") {
utils.RespondWithError(w, http.StatusBadRequest, "Invalid filename")
return
}
avatarPath := filepath.Join("./uploads/avatars", filename)
// Проверяем существование файла
fileInfo, err := os.Stat(avatarPath)
if os.IsNotExist(err) {
h.logger.Warn("avatar file not found", zap.String("path", avatarPath))
utils.RespondWithError(w, http.StatusNotFound, "Avatar not found")
return
}
if err != nil {
h.logger.Error("failed to access file", zap.Error(err))
utils.RespondWithError(w, http.StatusInternalServerError, "File access error")
return
}
// Открываем файл
file, err := os.Open(avatarPath)
if err != nil {
h.logger.Error("failed to open file", zap.Error(err))
utils.RespondWithError(w, http.StatusInternalServerError, "File open error")
return
}
defer file.Close()
// Определяем Content-Type
ext := strings.ToLower(filepath.Ext(filename))
contentType := "application/octet-stream"
switch ext {
case ".png": contentType = "image/png"
case ".jpg", ".jpeg": contentType = "image/jpeg"
case ".gif": contentType = "image/gif"
case ".webp": contentType = "image/webp"
}
// ✅ Устанавливаем ВСЕ заголовки
w.Header().Set("Content-Type", contentType)
w.Header().Set("Content-Length", fmt.Sprintf("%d", fileInfo.Size()))
w.Header().Set("Cache-Control", "public, max-age=31536000") // 1 год
w.Header().Set("Last-Modified", fileInfo.ModTime().Format(http.TimeFormat))
// Копируем файл в ResponseWriter
_, err = io.Copy(w, file)
if err != nil {
h.logger.Error("failed to send file", zap.Error(err))
// Не отправляем ошибку - соединение уже испорчено
return
}
h.logger.Info("avatar served successfully",
zap.String("filename", filename),
zap.String("content_type", contentType),
zap.Int64("file_size", fileInfo.Size()),
)
}
+2 -1
View File
@@ -56,8 +56,9 @@ func SetupRouter(db *gorm.DB, config *config.Config) http.Handler {
r.Use(middleware.AuthMiddleware(jwtService, userRepo))
r.Use(middleware.RequireAuth)
// Все операции с аватарами теперь через AvatarHandler
r.Mount("/avatars", allHandler.AvatarHandler().Routes())
// Профиль пользователя
r.Mount("/", allHandler.UserHandler().Routes())
// Здесь будут другие защищенные маршруты пользователя
@@ -19,7 +19,7 @@ type AvatarService interface {
UploadAvatar(userID uint, file multipart.File, header *multipart.FileHeader) (string, error)
DeleteAvatar(userID uint) error
GetAvatarPath(userID uint) (string, error)
GetAvatarFile(filename string) ([]byte, string, error) // НОВЫЙ МЕТОД
GetAvatarFile(filename string) ([]byte, string, error)
ServeAvatarFile(w io.Writer, filename string) (string, error)
}