5e35532dbb
- 添加 FloorOption 结构体字段支持尺寸、描述和变体信息 - 优化 loadFloorOptions 函数，实现按品牌和花色名称去重并附带多尺寸变体 - 区分木材、瓷砖、乙烯基和层压板产品，并分类返回 - 更新室内铺装纹理选项，完善图案描述和名称本地化 - 丰富Floor Options API响应数据，包含尺寸、变体、描述等字段 - 增强AI换地板功能，支持尺寸变体SKU，自动调整物理尺寸比例 - 完善地板替换的AI提示词，增加材质锁定、尺寸说明和纹理一致性要求 - 改进地板识别蒙版生成逻辑，确保精准分割地板区域 - Redis任务状态查询接口增加耗时统计字段，提供更细粒度进度信息 - 更新go.mod依赖，新增redis和gorm相关包支持
55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"floorvisualizer/internal/model"
|
|
"floorvisualizer/internal/repository"
|
|
)
|
|
|
|
func main() {
|
|
dsn := repository.DefaultDSN
|
|
if v := os.Getenv("DATABASE_URL"); v != "" {
|
|
dsn = v
|
|
}
|
|
dataDir := "data/products"
|
|
if v := os.Getenv("PRODUCT_DATA_DIR"); v != "" {
|
|
dataDir = v
|
|
}
|
|
|
|
db, err := repository.Connect(dsn)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer db.Close()
|
|
db.Exec("DELETE FROM products")
|
|
files, err := os.ReadDir(dataDir)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
total := 0
|
|
for _, f := range files {
|
|
if f.IsDir() || !strings.EqualFold(filepath.Ext(f.Name()), ".json") {
|
|
continue
|
|
}
|
|
data, err := os.ReadFile(filepath.Join(dataDir, f.Name()))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var prods []model.Product
|
|
if err := json.Unmarshal(data, &prods); err != nil {
|
|
panic(fmt.Errorf("parse %s: %w", f.Name(), err))
|
|
}
|
|
if err := repository.InsertProducts(db, prods); err != nil {
|
|
panic(err)
|
|
}
|
|
total += len(prods)
|
|
fmt.Printf(" %s: %d\n", f.Name(), len(prods))
|
|
}
|
|
fmt.Printf("\nDone! %d products\n", total)
|
|
}
|