wangjia
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
package handler
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/wangjia/jiu/backend/internal/middleware"
|
|
"github.com/wangjia/jiu/backend/internal/model"
|
|
)
|
|
|
|
type PartnerHandler struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewPartnerHandler(db *gorm.DB) *PartnerHandler {
|
|
return &PartnerHandler{db: db}
|
|
}
|
|
|
|
func (h *PartnerHandler) List(c *gin.Context) {
|
|
shopID := middleware.GetShopID(c)
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
|
|
|
query := h.db.Model(&model.Partner{}).
|
|
Where("shop_id = ? AND deleted_at IS NULL", shopID)
|
|
|
|
if t := c.Query("type"); t != "" {
|
|
query = query.Where("FIND_IN_SET(?, type)", t)
|
|
}
|
|
if kw := c.Query("keyword"); kw != "" {
|
|
query = query.Where("name LIKE ? OR phone LIKE ?", "%"+kw+"%", "%"+kw+"%")
|
|
}
|
|
|
|
var total int64
|
|
query.Count(&total)
|
|
|
|
partners := make([]model.Partner, 0)
|
|
query.Offset((page - 1) * pageSize).Limit(pageSize).Order("id DESC").Find(&partners)
|
|
|
|
c.JSON(http.StatusOK, gin.H{"data": partners, "total": total, "page": page, "page_size": pageSize})
|
|
}
|
|
|
|
func (h *PartnerHandler) Create(c *gin.Context) {
|
|
shopID := middleware.GetShopID(c)
|
|
var p model.Partner
|
|
if err := c.ShouldBindJSON(&p); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
p.ShopID = shopID
|
|
// 编码未提供时自动生成：供应商 S001/S002…，客户 C001/C002…
|
|
if p.Code == "" {
|
|
prefix := "S"
|
|
if p.Type == "customer" {
|
|
prefix = "C"
|
|
}
|
|
var count int64
|
|
h.db.Model(&model.Partner{}).
|
|
Where("shop_id = ? AND type = ? AND deleted_at IS NULL", shopID, p.Type).
|
|
Count(&count)
|
|
p.Code = fmt.Sprintf("%s%03d", prefix, count+1)
|
|
}
|
|
if err := h.db.Create(&p).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusCreated, gin.H{"data": p})
|
|
}
|
|
|
|
func (h *PartnerHandler) Update(c *gin.Context) {
|
|
shopID := middleware.GetShopID(c)
|
|
var p model.Partner
|
|
if err := h.db.Where("id = ? AND shop_id = ? AND deleted_at IS NULL", c.Param("id"), shopID).
|
|
First(&p).Error; err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
|
return
|
|
}
|
|
if err := c.ShouldBindJSON(&p); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
p.ShopID = shopID
|
|
h.db.Save(&p)
|
|
c.JSON(http.StatusOK, gin.H{"data": p})
|
|
}
|
|
|
|
func (h *PartnerHandler) Delete(c *gin.Context) {
|
|
shopID := middleware.GetShopID(c)
|
|
now := timeNow()
|
|
result := h.db.Model(&model.Partner{}).
|
|
Where("id = ? AND shop_id = ? AND deleted_at IS NULL", c.Param("id"), shopID).
|
|
Update("deleted_at", now)
|
|
if result.RowsAffected == 0 {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"message": "deleted"})
|
|
}
|