42 lines
829 B
Go
42 lines
829 B
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"goweb/context"
|
|
"goweb/ent"
|
|
"goweb/ent/user"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
func LoadUser(orm *ent.Client) echo.MiddlewareFunc {
|
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
return func(c echo.Context) error {
|
|
userID, err := strconv.Atoi(c.Param("user"))
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusNotFound, "Not found")
|
|
}
|
|
|
|
u, err := orm.User.
|
|
Query().
|
|
Where(user.ID(userID)).
|
|
Only(c.Request().Context())
|
|
|
|
switch err.(type) {
|
|
case nil:
|
|
case *ent.NotFoundError:
|
|
return echo.NewHTTPError(http.StatusNotFound, "Not found")
|
|
default:
|
|
c.Logger().Error(err)
|
|
return echo.NewHTTPError(http.StatusInternalServerError, "Internal server error")
|
|
}
|
|
|
|
c.Set(context.UserKey, u)
|
|
|
|
return next(c)
|
|
}
|
|
}
|
|
}
|