This commit is contained in:
vipg
2026-02-09 16:49:18 +08:00
parent 6640f09639
commit bdb1065217
6 changed files with 261 additions and 141 deletions

View File

@@ -0,0 +1,86 @@
package handler
import (
"encoding/json"
"net/http"
"common/types"
"user/internal/model"
"user/internal/service"
)
type Handler struct {
S *service.Service
}
func New(s *service.Service) *Handler {
return &Handler{S: s}
}
func (h *Handler) Register(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
writeJSON(w, http.StatusMethodNotAllowed, false, "method not allowed", nil)
return
}
var req model.RegisterReq
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
writeJSON(w, http.StatusBadRequest, false, "invalid json", nil)
return
}
userID, err := h.S.Register(req.Account, req.Password)
if err != nil {
switch err {
case service.ErrInvalidInput:
writeJSON(w, http.StatusBadRequest, false, "invalid account or password", nil)
case service.ErrAccountExists:
writeJSON(w, http.StatusConflict, false, "account exists", nil)
default:
writeJSON(w, http.StatusInternalServerError, false, "internal error", nil)
}
return
}
writeJSON(w, http.StatusCreated, true, "ok", map[string]string{"user_id": userID})
}
func (h *Handler) Login(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
writeJSON(w, http.StatusMethodNotAllowed, false, "method not allowed", nil)
return
}
var req model.LoginReq
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
writeJSON(w, http.StatusBadRequest, false, "invalid json", nil)
return
}
userID, err := h.S.Login(req.Account, req.Password)
if err != nil {
switch err {
case service.ErrInvalidInput:
writeJSON(w, http.StatusBadRequest, false, "invalid account or password", nil)
case service.ErrUnauthorized:
writeJSON(w, http.StatusUnauthorized, false, "unauthorized", nil)
default:
writeJSON(w, http.StatusInternalServerError, false, "internal error", nil)
}
return
}
writeJSON(w, http.StatusOK, true, "ok", map[string]string{"user_id": userID})
}
func (h *Handler) Healthz(w http.ResponseWriter, r *http.Request) {
writeJSON(w, http.StatusOK, true, "ok", nil)
}
func (h *Handler) Version(w http.ResponseWriter, r *http.Request) {
writeJSON(w, http.StatusOK, true, "ok", map[string]string{"version": "user-service v0.1.0"})
}
func (h *Handler) Root(w http.ResponseWriter, r *http.Request) {
writeJSON(w, http.StatusOK, true, "ok", map[string]string{"service": "user"})
}
func writeJSON(w http.ResponseWriter, code int, status bool, msg string, data interface{}) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(code)
json.NewEncoder(w).Encode(types.Response{Status: status, Message: msg, Data: data})
}