Merge remote-tracking branch 'origin/master'
This commit is contained in:
218
svc/routes/brand_classification.go
Normal file
218
svc/routes/brand_classification.go
Normal file
@@ -0,0 +1,218 @@
|
||||
package routes
|
||||
|
||||
import (
|
||||
"bssapp-backend/db"
|
||||
"bssapp-backend/queries"
|
||||
"bssapp-backend/utils"
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
)
|
||||
|
||||
type BrandClassificationLookupResponse struct {
|
||||
Groups []queries.BrandGroupOption `json:"groups"`
|
||||
}
|
||||
|
||||
type BrandSyncResponse struct {
|
||||
Upserted int `json:"upserted"`
|
||||
Deleted int `json:"deleted"`
|
||||
Total int `json:"total"`
|
||||
}
|
||||
|
||||
type BrandSetGroupPayload struct {
|
||||
GroupID int `json:"group_id"`
|
||||
}
|
||||
|
||||
type BrandBulkItem struct {
|
||||
BrandCode string `json:"brand_code"`
|
||||
GroupID int `json:"group_id"`
|
||||
}
|
||||
|
||||
type BrandBulkPayload struct {
|
||||
Items []BrandBulkItem `json:"items"`
|
||||
}
|
||||
|
||||
func GetBrandClassificationLookupsHandler(pg *sql.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||
if err := queries.EnsureBrandClassificationTables(pg); err != nil {
|
||||
http.Error(w, "brand tables bootstrap error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
traceID := utils.TraceIDFromRequest(r)
|
||||
ctx := utils.ContextWithTraceID(r.Context(), traceID)
|
||||
|
||||
groups, err := queries.ListBrandGroups(ctx, pg)
|
||||
if err != nil {
|
||||
http.Error(w, "brand groups lookup error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
_ = json.NewEncoder(w).Encode(BrandClassificationLookupResponse{Groups: groups})
|
||||
}
|
||||
}
|
||||
|
||||
func ListBrandsHandler(pg *sql.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||
if err := queries.EnsureBrandClassificationTables(pg); err != nil {
|
||||
http.Error(w, "brand tables bootstrap error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
q := strings.TrimSpace(r.URL.Query().Get("q"))
|
||||
traceID := utils.TraceIDFromRequest(r)
|
||||
ctx := utils.ContextWithTraceID(r.Context(), traceID)
|
||||
|
||||
rows, err := queries.ListBrandsWithGroups(ctx, pg, q, 20000)
|
||||
if err != nil {
|
||||
http.Error(w, "brand list error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
_ = json.NewEncoder(w).Encode(rows)
|
||||
}
|
||||
}
|
||||
|
||||
func SyncBrandsFromMSSQLHandler(pg *sql.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||
|
||||
mssql := db.GetDB()
|
||||
if mssql == nil {
|
||||
http.Error(w, "mssql connection not available", http.StatusServiceUnavailable)
|
||||
return
|
||||
}
|
||||
|
||||
traceID := utils.TraceIDFromRequest(r)
|
||||
ctx := utils.ContextWithTraceID(r.Context(), traceID)
|
||||
|
||||
res, err := queries.SyncBrandsFromMSSQL(ctx, mssql, pg)
|
||||
if err != nil {
|
||||
log.Printf("brand sync error trace=%s err=%v", traceID, err)
|
||||
http.Error(w, "brand sync error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
_ = json.NewEncoder(w).Encode(BrandSyncResponse{
|
||||
Upserted: res.Upserted,
|
||||
Deleted: res.Deleted,
|
||||
Total: res.Total,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func SetBrandGroupHandler(pg *sql.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||
|
||||
if err := queries.EnsureBrandClassificationTables(pg); err != nil {
|
||||
http.Error(w, "brand tables bootstrap error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
brandCode := strings.TrimSpace(mux.Vars(r)["code"])
|
||||
if brandCode == "" {
|
||||
http.Error(w, "invalid brand_code", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
var payload BrandSetGroupPayload
|
||||
if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
|
||||
http.Error(w, "invalid payload", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if payload.GroupID < 0 || payload.GroupID > 3 {
|
||||
http.Error(w, "invalid group_id", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
traceID := utils.TraceIDFromRequest(r)
|
||||
ctx := utils.ContextWithTraceID(r.Context(), traceID)
|
||||
|
||||
tx, err := pg.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
http.Error(w, "pg transaction start error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
if err := queries.SetBrandGroup(ctx, tx, brandCode, payload.GroupID); err != nil {
|
||||
http.Error(w, "brand group save error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
if err := tx.Commit(); err != nil {
|
||||
http.Error(w, "pg transaction commit error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||||
"success": true,
|
||||
"brand_code": brandCode,
|
||||
"group_id": payload.GroupID,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func SetBrandGroupsBulkHandler(pg *sql.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||
|
||||
if err := queries.EnsureBrandClassificationTables(pg); err != nil {
|
||||
http.Error(w, "brand tables bootstrap error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
var payload BrandBulkPayload
|
||||
if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
|
||||
http.Error(w, "invalid payload", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if len(payload.Items) == 0 {
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{"success": true, "updated": 0})
|
||||
return
|
||||
}
|
||||
|
||||
traceID := utils.TraceIDFromRequest(r)
|
||||
ctx := utils.ContextWithTraceID(r.Context(), traceID)
|
||||
|
||||
tx, err := pg.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
http.Error(w, "pg transaction start error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
updated := 0
|
||||
for _, it := range payload.Items {
|
||||
code := strings.TrimSpace(it.BrandCode)
|
||||
if code == "" {
|
||||
continue
|
||||
}
|
||||
if it.GroupID < 0 || it.GroupID > 3 {
|
||||
http.Error(w, "invalid group_id", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if err := queries.SetBrandGroup(ctx, tx, code, it.GroupID); err != nil {
|
||||
http.Error(w, "brand group save error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
updated++
|
||||
}
|
||||
|
||||
if err := tx.Commit(); err != nil {
|
||||
http.Error(w, "pg transaction commit error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||||
"success": true,
|
||||
"updated": updated,
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user