This commit is contained in:
vipg
2025-11-14 17:48:50 +08:00
parent 43fd114804
commit 7322fee989
2 changed files with 184 additions and 284 deletions

View File

@@ -1,78 +1,97 @@
package logic package service
import ( import (
"database/sql" "database/sql"
"errors" "futures_trade_record/db"
"fmt" "futures_trade_record/logger"
"net/http"
"time" "time"
"github.com/gin-gonic/gin"
"github.com/google/uuid" "github.com/google/uuid"
"go.uber.org/zap" "go.uber.org/zap"
) )
// Variety 主表结构体 // Variety 品种主表结构体
type Variety struct { type Variety struct {
ID uuid.UUID ID string `json:"id"`
Deleted bool Deleted bool `json:"deleted"`
CreatedAt time.Time CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time UpdatedAt time.Time `json:"updated_at"`
} }
// VarietyName 子表结构体(名称) // VarietyName 品种名称子表结构体
type VarietyName struct { type VarietyName struct {
ID uuid.UUID ID string `json:"id"`
VarietyID uuid.UUID VarietyID string `json:"variety_id"`
Name string Name string `json:"name" binding:"required"`
Deleted bool Deleted bool `json:"deleted"`
CreatedAt time.Time CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time UpdatedAt time.Time `json:"updated_at"`
} }
// VarietyCode 子表结构体(代码) // VarietyCode 品种代码子表结构体
type VarietyCode struct { type VarietyCode struct {
ID uuid.UUID ID string `json:"id"`
VarietyID uuid.UUID VarietyID string `json:"variety_id"`
Code string Code string `json:"code" binding:"required"`
Deleted bool Deleted bool `json:"deleted"`
CreatedAt time.Time CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time UpdatedAt time.Time `json:"updated_at"`
} }
// VarietyTick 子表结构体(跳点) // VarietyTick 品种跳点子表结构体
type VarietyTick struct { type VarietyTick struct {
ID uuid.UUID ID string `json:"id"`
VarietyID uuid.UUID VarietyID string `json:"variety_id"`
Tick float64 Tick float64 `json:"tick"`
Deleted bool Deleted bool `json:"deleted"`
CreatedAt time.Time CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time UpdatedAt time.Time `json:"updated_at"`
} }
// VarietyTickPrice 子表结构体(跳点价格) // VarietyTickPrice 品种跳点价格子表结构体
type VarietyTickPrice struct { type VarietyTickPrice struct {
ID uuid.UUID ID string `json:"id"`
VarietyID uuid.UUID VarietyID string `json:"variety_id"`
Price float64 Price float64 `json:"price"`
Deleted bool Deleted bool `json:"deleted"`
CreatedAt time.Time CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time UpdatedAt time.Time `json:"updated_at"`
} }
// VarietyInfo 联合查询视图结构体 // CreateVarietyRequest 创建品种请求参数
type VarietyInfo struct { type CreateVarietyRequest struct {
VarietyID uuid.UUID Name string `json:"name" binding:"required"`
Name string Code string `json:"code" binding:"required"`
Code string Tick float64 `json:"tick"`
Tick float64 TickPrice float64 `json:"tick_price"`
TickPrice float64
} }
// CreateVariety 创建品种主记录及关联信息 // CreateVarietyHandler 创建品种接口处理函数
func CreateVariety(name, code string, tick, tickPrice float64) (uuid.UUID, error) { func CreateVarietyHandler(c *gin.Context) {
tx, err := DB.Begin() var req CreateVarietyRequest
if err := c.ShouldBindJSON(&req); err != nil {
logger.ZapLogger.Error("请求参数绑定失败", zap.Error(err))
c.JSON(http.StatusBadRequest, gin.H{
"code": http.StatusBadRequest,
"message": "无效的请求参数: " + err.Error(),
})
return
}
// 生成主表ID
varietyID := uuid.New().String()
// 开启数据库事务
tx, err := db.DB.Begin()
if err != nil { if err != nil {
zap.L().Error("创建事务失败", zap.Error(err)) logger.ZapLogger.Error("开启事务失败", zap.Error(err))
return uuid.Nil, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "服务器内部错误",
})
return
} }
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
@@ -80,277 +99,159 @@ func CreateVariety(name, code string, tick, tickPrice float64) (uuid.UUID, error
} }
}() }()
// 1. 创建主表记录 // 插入主表
var varietyID uuid.UUID now := time.Now()
err = tx.QueryRow(`
INSERT INTO variety (deleted)
VALUES ($1)
RETURNING id
`, false).Scan(&varietyID)
if err != nil {
tx.Rollback()
zap.L().Error("创建品种主表失败", zap.Error(err))
return uuid.Nil, err
}
// 2. 创建名称记录
_, err = tx.Exec(` _, err = tx.Exec(`
INSERT INTO variety_name (variety_id, name, deleted) INSERT INTO variety (id, deleted, created_at, updated_at)
VALUES ($1, $2, $3) VALUES ($1, $2, $3, $4)
`, varietyID, name, false) `, varietyID, false, now, now)
if err != nil { if err != nil {
tx.Rollback() tx.Rollback()
zap.L().Error("创建品种名称记录失败", zap.Error(err)) logger.ZapLogger.Error("插入品种主表失败", zap.Error(err))
return uuid.Nil, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "创建品种失败",
})
return
} }
// 3. 创建代码记录 // 插入名称子表
_, err = tx.Exec(` _, err = tx.Exec(`
INSERT INTO variety_code (variety_id, code, deleted) INSERT INTO variety_name (id, variety_id, name, deleted, created_at, updated_at)
VALUES ($1, $2, $3) VALUES ($1, $2, $3, $4, $5, $6)
`, varietyID, code, false) `, uuid.New().String(), varietyID, req.Name, false, now, now)
if err != nil { if err != nil {
tx.Rollback() tx.Rollback()
zap.L().Error("创建品种代码记录失败", zap.Error(err)) logger.ZapLogger.Error("插入品种名称表失败", zap.Error(err))
return uuid.Nil, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "创建品种失败",
})
return
} }
// 4. 创建跳点记录 // 插入代码子表
_, err = tx.Exec(` _, err = tx.Exec(`
INSERT INTO variety_tick (variety_id, tick, deleted) INSERT INTO variety_code (id, variety_id, code, deleted, created_at, updated_at)
VALUES ($1, $2, $3) VALUES ($1, $2, $3, $4, $5, $6)
`, varietyID, tick, false) `, uuid.New().String(), varietyID, req.Code, false, now, now)
if err != nil { if err != nil {
tx.Rollback() tx.Rollback()
zap.L().Error("创建跳点记录失败", zap.Error(err)) logger.ZapLogger.Error("插入品种代码表失败", zap.Error(err))
return uuid.Nil, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "创建品种失败",
})
return
} }
// 5. 创建跳点价格记录 // 插入跳点子表
_, err = tx.Exec(` _, err = tx.Exec(`
INSERT INTO variety_tick_price (variety_id, price, deleted) INSERT INTO variety_tick (id, variety_id, tick, deleted, created_at, updated_at)
VALUES ($1, $2, $3) VALUES ($1, $2, $3, $4, $5, $6)
`, varietyID, tickPrice, false) `, uuid.New().String(), varietyID, req.Tick, false, now, now)
if err != nil { if err != nil {
tx.Rollback() tx.Rollback()
zap.L().Error("创建跳点价格记录失败", zap.Error(err)) logger.ZapLogger.Error("插入品种跳点表失败", zap.Error(err))
return uuid.Nil, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "创建品种失败",
})
return
} }
// 插入跳点价格子表
_, err = tx.Exec(`
INSERT INTO variety_tick_price (id, variety_id, price, deleted, created_at, updated_at)
VALUES ($1, $2, $3, $4, $5, $6)
`, uuid.New().String(), varietyID, req.TickPrice, false, now, now)
if err != nil {
tx.Rollback()
logger.ZapLogger.Error("插入品种跳点价格表失败", zap.Error(err))
c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "创建品种失败",
})
return
}
// 提交事务
if err := tx.Commit(); err != nil { if err := tx.Commit(); err != nil {
tx.Rollback() tx.Rollback()
zap.L().Error("提交事务失败", zap.Error(err)) logger.ZapLogger.Error("提交事务失败", zap.Error(err))
return uuid.Nil, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "创建品种失败",
})
return
} }
return varietyID, nil c.JSON(http.StatusOK, gin.H{
"code": http.StatusOK,
"message": "品种创建成功",
"data": gin.H{
"variety_id": varietyID,
},
})
} }
// GetVarietyByID 查询品种详情通过ID // GetVarietyListHandler 获取品种列表接口
func GetVarietyByID(id uuid.UUID) (*VarietyInfo, error) { func GetVarietyListHandler(c *gin.Context) {
var info VarietyInfo rows, err := db.DB.Query(`
err := DB.QueryRow(` SELECT variety_id, name, code, tick, tick_price
SELECT
variety_id, name, code, tick, tick_price
FROM variety_info_view FROM variety_info_view
WHERE variety_id = $1
`, id).Scan(
&info.VarietyID,
&info.Name,
&info.Code,
&info.Tick,
&info.TickPrice,
)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, fmt.Errorf("品种不存在: %s", id)
}
zap.L().Error("查询品种失败", zap.String("id", id.String()), zap.Error(err))
return nil, err
}
return &info, nil
}
// GetVarietyByName 按名称查询品种
func GetVarietyByName(name string) (*VarietyInfo, error) {
var info VarietyInfo
err := DB.QueryRow(`
SELECT
variety_id, name, code, tick, tick_price
FROM variety_info_view
WHERE name = $1 AND name IS NOT NULL
`, name).Scan(
&info.VarietyID,
&info.Name,
&info.Code,
&info.Tick,
&info.TickPrice,
)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, fmt.Errorf("未找到名称为 %s 的品种", name)
}
zap.L().Error("按名称查询品种失败", zap.String("name", name), zap.Error(err))
return nil, err
}
return &info, nil
}
// UpdateVarietyName 更新品种名称
func UpdateVarietyName(varietyID uuid.UUID, newName string) error {
result, err := DB.Exec(`
UPDATE variety_name
SET name = $1
WHERE variety_id = $2 AND deleted = false
`, newName, varietyID)
if err != nil {
zap.L().Error("更新品种名称失败", zap.String("id", varietyID.String()), zap.Error(err))
return err
}
rowsAffected, err := result.RowsAffected()
if err != nil {
return err
}
if rowsAffected == 0 {
return fmt.Errorf("未找到可更新的品种名称记录: %s", varietyID)
}
return nil
}
// UpdateVarietyTick 更新跳点值
func UpdateVarietyTick(varietyID uuid.UUID, newTick float64) error {
result, err := DB.Exec(`
UPDATE variety_tick
SET tick = $1
WHERE variety_id = $2 AND deleted = false
`, newTick, varietyID)
if err != nil {
zap.L().Error("更新跳点值失败", zap.String("id", varietyID.String()), zap.Error(err))
return err
}
rowsAffected, err := result.RowsAffected()
if err != nil {
return err
}
if rowsAffected == 0 {
return fmt.Errorf("未找到可更新的跳点记录: %s", varietyID)
}
return nil
}
// DeleteVariety 逻辑删除品种(软删除)
func DeleteVariety(varietyID uuid.UUID) error {
tx, err := DB.Begin()
if err != nil {
zap.L().Error("创建事务失败", zap.Error(err))
return err
}
defer func() {
if r := recover(); r != nil {
tx.Rollback()
}
}()
// 1. 软删除主表
_, err = tx.Exec(`
UPDATE variety
SET deleted = true
WHERE id = $1
`, varietyID)
if err != nil {
tx.Rollback()
zap.L().Error("删除品种主表失败", zap.String("id", varietyID.String()), zap.Error(err))
return err
}
// 2. 软删除关联子表(实际可依赖外键级联删除,这里显式处理确保一致性)
tables := []string{
"variety_name",
"variety_code",
"variety_tick",
"variety_tick_price",
}
for _, table := range tables {
_, err = tx.Exec(fmt.Sprintf(`
UPDATE %s
SET deleted = true
WHERE variety_id = $1
`, table), varietyID)
if err != nil {
tx.Rollback()
zap.L().Error("删除子表记录失败",
zap.String("table", table),
zap.String("id", varietyID.String()),
zap.Error(err))
return err
}
}
if err := tx.Commit(); err != nil {
tx.Rollback()
zap.L().Error("提交删除事务失败", zap.Error(err))
return err
}
return nil
}
// ListVarieties 分页查询品种列表
func ListVarieties(page, pageSize int) ([]VarietyInfo, int, error) {
if page < 1 {
page = 1
}
offset := (page - 1) * pageSize
// 查询总数
var total int
err := DB.QueryRow(`
SELECT COUNT(DISTINCT variety_id)
FROM variety_info_view
WHERE variety_id IS NOT NULL
`).Scan(&total)
if err != nil {
zap.L().Error("查询品种总数失败", zap.Error(err))
return nil, 0, err
}
// 查询分页数据
rows, err := DB.Query(`
SELECT
variety_id, name, code, tick, tick_price
FROM variety_info_view
WHERE variety_id IS NOT NULL
ORDER BY variety_id DESC ORDER BY variety_id DESC
LIMIT $1 OFFSET $2 `)
`, pageSize, offset)
if err != nil { if err != nil {
zap.L().Error("查询品种列表失败", zap.Error(err)) logger.ZapLogger.Error("查询品种列表失败", zap.Error(err))
return nil, 0, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "获取品种列表失败",
})
return
} }
defer rows.Close() defer rows.Close()
var varieties []VarietyInfo var varieties []gin.H
for rows.Next() { for rows.Next() {
var info VarietyInfo var (
err := rows.Scan( varietyID string
&info.VarietyID, name sql.NullString
&info.Name, code sql.NullString
&info.Code, tick sql.NullFloat64
&info.Tick, price sql.NullFloat64
&info.TickPrice,
) )
if err != nil {
zap.L().Error("解析品种记录失败", zap.Error(err)) if err := rows.Scan(&varietyID, &name, &code, &tick, &price); err != nil {
return nil, 0, err logger.ZapLogger.Error("扫描品种数据失败", zap.Error(err))
c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "获取品种列表失败",
})
return
} }
varieties = append(varieties, info)
varieties = append(varieties, gin.H{
"variety_id": varietyID,
"name": name.String,
"code": code.String,
"tick": tick.Float64,
"tick_price": price.Float64,
})
} }
if err = rows.Err(); err != nil { if err := rows.Err(); err != nil {
zap.L().Error("遍历品种记录失败", zap.Error(err)) logger.ZapLogger.Error("行迭代错误", zap.Error(err))
return nil, 0, err c.JSON(http.StatusInternalServerError, gin.H{
"code": http.StatusInternalServerError,
"message": "获取品种列表失败",
})
return
} }
return varieties, total, nil c.JSON(http.StatusOK, gin.H{
"code": http.StatusOK,
"message": "success",
"data": varieties,
})
} }

View File

@@ -1 +0,0 @@
package service