bj_power/bj_power_mes/internal/plc/plc.go

package plc

import (
	"context"
	"strconv"
	"strings"
	"time"

	"bj_power_mes/internal/config"
)

// Client PLC 对接抽象。
// 生产对接西门子 S7：写入工序码寄存器，读完成信号位；未收到完成信号不下发下一条。
// mock 模式下直接模拟：下发后置发送状态，可主动标记完成。
type Client interface {
	// SendProcessCode 下发工位组合（如 "1,3,5"），返回已下发工位码
	SendProcessCode(ctx context.Context, processCombination string) ([]int, error)
	// QueryDone 查询最近一次下发是否收到完成信号，完成则返回 true
	QueryDone(ctx context.Context) (bool, error)
	// AckDone 标记完成（mock 使用，也可由 PLC 信号轮询写入）
	AckDone(ctx context.Context, ok bool)
	// QueryDockPallet 查询某接驳台(DOCK01..21)当前是否有托盘（读 PLC 托盘传感器信号）。
	// 用于 AGV 到达目的地前确认接驳台空闲。暂未接真实 PLC 时 mock 返回 false（默认无托盘）。
	QueryDockPallet(ctx context.Context, dockCode string) (bool, error)
}

// Manager 管理 PLC 客户端实例
type Manager struct {
	client Client
	mock   bool
}

func NewManager(c config.PlcConf, mock bool) (*Manager, error) {
	m := &Manager{mock: mock}
	if mock {
		m.client = newMockClient()
	} else {
		m.client = newTcpClient(c)
	}
	return m, nil
}

func (m *Manager) Get() Client { return m.client }

func (m *Manager) IsMock() bool { return m.mock }

// SetMockDockPallet 仅在 mock 模式下生效：手动置位某接驳台是否有托盘（联调用）。
func (m *Manager) SetMockDockPallet(dockCode string, has bool) {
	if !m.mock {
		return
	}
	if mc, ok := m.client.(*mockClient); ok {
		mc.SetDockPallet(dockCode, has)
	}
}

// parseCombination 解析工位组合 "1,3,5" / "10,11,12" -> [1,3,5] / [10,11,12]
func parseCombination(s string) []int {
	var out []int
	for _, p := range strings.Split(s, ",") {
		p = strings.TrimSpace(p)
		if p == "" {
			continue
		}
		n, err := strconv.Atoi(p)
		if err != nil || n < 1 || n > 12 {
			continue
		}
		out = append(out, n)
	}
	return out
}

// Sleep 简化轮询间隔（供 logic 使用）
func Sleep(d time.Duration) {
	time.Sleep(d)
}