68 lines
2.2 KiB
Go
68 lines
2.2 KiB
Go
|
package model
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ ProPlanProcessStepModel = (*customProPlanProcessStepModel)(nil)
|
||
|
|
|
||
|
|
type (
|
||
|
|
ProPlanProcessStepModel interface {
|
||
|
|
proPlanProcessStepModel
|
||
|
|
FindByPlanId(ctx context.Context, planId string) ([]*ProPlanProcessStep, error)
|
||
|
|
BulkInsert(ctx context.Context, data []*ProPlanProcessStep) error
|
||
|
|
UpdateStepStatus(ctx context.Context, planId, stepType string, status int64, operatorId string) error
|
||
|
|
}
|
||
|
|
|
||
|
|
customProPlanProcessStepModel struct {
|
||
|
|
*defaultProPlanProcessStepModel
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewProPlanProcessStepModel(conn sqlx.SqlConn) ProPlanProcessStepModel {
|
||
|
|
return &customProPlanProcessStepModel{
|
||
|
|
defaultProPlanProcessStepModel: newProPlanProcessStepModel(conn),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *customProPlanProcessStepModel) FindByPlanId(ctx context.Context, planId string) ([]*ProPlanProcessStep, error) {
|
||
|
|
var list []*ProPlanProcessStep
|
||
|
|
query := fmt.Sprintf("SELECT %s FROM %s WHERE plan_id = ? ORDER BY step_order", proPlanProcessStepRows, m.table)
|
||
|
|
if err := m.conn.QueryRowsCtx(ctx, &list, query, planId); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return list, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *customProPlanProcessStepModel) BulkInsert(ctx context.Context, data []*ProPlanProcessStep) error {
|
||
|
|
if len(data) == 0 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
values := make([]string, 0, len(data))
|
||
|
|
args := make([]interface{}, 0, len(data)*8)
|
||
|
|
for _, d := range data {
|
||
|
|
values = append(values, "(?, ?, ?, ?, ?, ?, ?, ?)")
|
||
|
|
args = append(args, d.StepId, d.PlanId, d.StepType, d.StepOrder, d.Status,
|
||
|
|
d.OperatorId, d.CompletedAt, d.Remark)
|
||
|
|
}
|
||
|
|
query := fmt.Sprintf("INSERT INTO %s (%s) VALUES %s", m.table, proPlanProcessStepRowsExpectAutoSet, strings.Join(values, ","))
|
||
|
|
_, err := m.conn.ExecCtx(ctx, query, args...)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *customProPlanProcessStepModel) UpdateStepStatus(ctx context.Context, planId, stepType string, status int64, operatorId string) error {
|
||
|
|
var completedAt sql.NullTime
|
||
|
|
if status == 2 {
|
||
|
|
completedAt = sql.NullTime{Time: time.Now(), Valid: true}
|
||
|
|
}
|
||
|
|
query := fmt.Sprintf("UPDATE %s SET status = ?, operator_id = ?, completed_at = ? WHERE plan_id = ? AND step_type = ?", m.table)
|
||
|
|
_, err := m.conn.ExecCtx(ctx, query, status, operatorId, completedAt, planId, stepType)
|
||
|
|
return err
|
||
|
|
}
|