first commit
This commit is contained in:
202
internal/router/manage/customer/customer.go
Normal file
202
internal/router/manage/customer/customer.go
Normal file
@@ -0,0 +1,202 @@
|
||||
package customer
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
"management/internal/db/model/form"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/global"
|
||||
"management/internal/global/html"
|
||||
"management/internal/global/know"
|
||||
"management/internal/middleware/manage/auth"
|
||||
"management/internal/pkg/snowflake"
|
||||
"management/internal/router/manage/util"
|
||||
categoryservice "management/internal/service/category"
|
||||
"management/internal/tpl"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
func List(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
cc, _ := categoryservice.GetParentCategorySelectLetter(ctx, know.CustomerCategory)
|
||||
ss, _ := categoryservice.GetParentCategorySelectLetter(ctx, know.CustomerSource)
|
||||
tpl.HTML(w, r, "customer/list.tmpl", map[string]any{
|
||||
"Statuses": html.NewSelectControls(global.Statuses, 0),
|
||||
"Categories": html.NewSelectStringControls(cc, "0"),
|
||||
"Sources": html.NewSelectStringControls(ss, "0"),
|
||||
})
|
||||
}
|
||||
|
||||
func PostList(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
category := util.ConvertInt32(r.PostFormValue("category"), 9999)
|
||||
if category == 0 {
|
||||
category = 9999
|
||||
}
|
||||
source := util.ConvertInt32(r.PostFormValue("source"), 9999)
|
||||
if source == 0 {
|
||||
source = 9999
|
||||
}
|
||||
|
||||
title := strings.TrimSpace(r.PostFormValue("title"))
|
||||
var search string
|
||||
if len(title) > 0 {
|
||||
search = "%" + title + "%"
|
||||
if strings.HasSuffix(title, ":") {
|
||||
search = title[:len(title)-1] + "%"
|
||||
}
|
||||
}
|
||||
arg := &db.ListCustomerConditionParam{
|
||||
IsTitle: len(search) > 0,
|
||||
Title: search,
|
||||
Status: util.ConvertInt16(r.PostFormValue("status"), 9999),
|
||||
Category: category,
|
||||
Source: source,
|
||||
PageID: util.ConvertInt32(r.PostFormValue("page"), 1),
|
||||
PageSize: util.ConvertInt32(r.PostFormValue("rows"), 10),
|
||||
}
|
||||
arg.TimeBegin, arg.TimeEnd = util.DefaultStartTimeAndEndTime(r.PostFormValue("timeBegin"), r.PostFormValue("timeEnd"))
|
||||
res, total, err := db.Engine.ListCustomerCondition(ctx, arg)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
data := tpl.ResponseList{
|
||||
Code: 0,
|
||||
Message: "ok",
|
||||
Count: total,
|
||||
Data: res,
|
||||
}
|
||||
tpl.JSON(w, data)
|
||||
}
|
||||
|
||||
func Add(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
cc, _ := categoryservice.GetParentCategorySelectLetter(ctx, know.CustomerCategory)
|
||||
ss, _ := categoryservice.GetParentCategorySelectLetter(ctx, know.CustomerSource)
|
||||
tpl.HTML(w, r, "customer/edit.tmpl", map[string]any{
|
||||
"Item": &form.CustomerForm{},
|
||||
"Statuses": html.NewSelectControls(global.Statuses, 0),
|
||||
"Categories": html.NewSelectStringControls(cc, "0"),
|
||||
"Sources": html.NewSelectStringControls(ss, "0"),
|
||||
})
|
||||
}
|
||||
|
||||
func Edit(w http.ResponseWriter, r *http.Request) {
|
||||
vars := r.URL.Query()
|
||||
id := util.ConvertInt[int64](vars.Get("id"), 0)
|
||||
customer := &form.CustomerForm{}
|
||||
ctx := r.Context()
|
||||
if id > 0 {
|
||||
if cus, err := db.Engine.GetCustomer(ctx, id); err == nil {
|
||||
customer = customer.ToForm(cus)
|
||||
if u, err := db.Engine.GetSysUser(ctx, cus.CreatedBy); err == nil {
|
||||
customer.CreatedBy = u.Username
|
||||
}
|
||||
if u, err := db.Engine.GetSysUser(ctx, cus.UpdatedBy); err == nil {
|
||||
customer.UpdatedBy = u.Username
|
||||
}
|
||||
}
|
||||
}
|
||||
cc, _ := categoryservice.GetParentCategorySelectLetter(ctx, know.CustomerCategory)
|
||||
ss, _ := categoryservice.GetParentCategorySelectLetter(ctx, know.CustomerSource)
|
||||
tpl.HTML(w, r, "customer/edit.tmpl", map[string]any{
|
||||
"Item": customer,
|
||||
"Statuses": html.NewSelectControls(global.Statuses, customer.Status),
|
||||
"Categories": html.NewSelectStringControls(cc, strconv.Itoa(int(customer.Category))),
|
||||
"Sources": html.NewSelectStringControls(ss, strconv.Itoa(int(customer.Source))),
|
||||
})
|
||||
}
|
||||
|
||||
func Save(w http.ResponseWriter, r *http.Request) {
|
||||
data := &form.CustomerForm{}
|
||||
if err := form.BindForm(r, data); err != nil {
|
||||
tpl.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
ctx := r.Context()
|
||||
authUser := auth.AuthUser(ctx)
|
||||
if data.ID > 0 {
|
||||
arg := &db.UpdateCustomerParams{
|
||||
ID: data.ID,
|
||||
Name: pgtype.Text{
|
||||
String: data.Name,
|
||||
Valid: true,
|
||||
},
|
||||
Category: pgtype.Int4{
|
||||
Int32: data.Category,
|
||||
Valid: true,
|
||||
},
|
||||
Source: pgtype.Int4{
|
||||
Int32: data.Source,
|
||||
Valid: true,
|
||||
},
|
||||
Address: pgtype.Text{
|
||||
String: data.Address,
|
||||
Valid: true,
|
||||
},
|
||||
ContactName: pgtype.Text{
|
||||
String: data.ContactName,
|
||||
Valid: true,
|
||||
},
|
||||
ContactPhone: pgtype.Text{
|
||||
String: data.ContactPhone,
|
||||
Valid: true,
|
||||
},
|
||||
Status: pgtype.Int2{
|
||||
Int16: data.Status,
|
||||
Valid: true,
|
||||
},
|
||||
UpdatedBy: pgtype.Int4{
|
||||
Int32: authUser.ID,
|
||||
Valid: true,
|
||||
},
|
||||
}
|
||||
_, err := db.Engine.UpdateCustomer(ctx, arg)
|
||||
if err != nil {
|
||||
tpl.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
tpl.JSONOK(w, "更新成功")
|
||||
} else {
|
||||
arg := &db.CreateCustomerParams{
|
||||
ID: snowflake.GetId(),
|
||||
Name: data.Name,
|
||||
Category: data.Category,
|
||||
Source: data.Source,
|
||||
Address: data.Address,
|
||||
ContactName: data.ContactName,
|
||||
ContactPhone: data.ContactPhone,
|
||||
CreatedBy: authUser.ID,
|
||||
}
|
||||
_, err := db.Engine.CreateCustomer(ctx, arg)
|
||||
if err != nil {
|
||||
tpl.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
tpl.JSONOK(w, "添加成功")
|
||||
}
|
||||
}
|
||||
|
||||
func XmSelect(w http.ResponseWriter, r *http.Request) {
|
||||
all, err := db.Engine.AllCustomers(r.Context())
|
||||
if err != nil {
|
||||
tpl.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
var res []*dto.XmSelectStrDto
|
||||
for _, v := range all {
|
||||
res = append(res, &dto.XmSelectStrDto{
|
||||
Name: v.Name,
|
||||
Value: strconv.FormatInt(v.ID, 10),
|
||||
})
|
||||
}
|
||||
tpl.JSON(w, res)
|
||||
}
|
||||
Reference in New Issue
Block a user