242 lines
6.9 KiB
Go
242 lines
6.9 KiB
Go
|
package processor
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"database/sql"
|
|||
|
|
"fmt"
|
|||
|
|
"log/slog"
|
|||
|
|
"regexp"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
momlib "bjhardman.cn/bjhardman/mom"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// EquipUploadService 每日定时把 OKUMA 设备的运行时长汇总写到 MOM 的 AUTO_EQUIP_UPLOAD 表。
|
|||
|
|
//
|
|||
|
|
// 数据源：dc_device_status_record（PostgreSQL，与 back_cover 共库）
|
|||
|
|
// 触发：每分钟一次 ticker，命中 EquipUploadHour 整点且当日未上传过则触发
|
|||
|
|
// 时长口径（与 OEE 计算保持一致）：
|
|||
|
|
// - WorkTime = RUN(及其变体) 状态时长（小时）
|
|||
|
|
// - OpenTime = 非 OFFLINE 状态时长（小时）
|
|||
|
|
// - WaitTime = OpenTime - WorkTime
|
|||
|
|
// - SumCost = WorkTime
|
|||
|
|
type EquipUploadService struct {
|
|||
|
|
db *sql.DB
|
|||
|
|
mom *momlib.Mom
|
|||
|
|
hour int // 每日触发整点（0-23），上传前一日数据
|
|||
|
|
deviceTypes []string // device_type 过滤；空表示所有 OKUMA 类
|
|||
|
|
stopCh chan struct{}
|
|||
|
|
lastRunDate string // 已上传的日期（YYYY-MM-DD），用于幂等
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewEquipUploadService 构造服务。db / mom 任一为空时 Start 仍可调用但是空跑。
|
|||
|
|
func NewEquipUploadService(db *sql.DB, mom *momlib.Mom, hour int, deviceTypes []string) *EquipUploadService {
|
|||
|
|
if hour < 0 || hour > 23 {
|
|||
|
|
hour = 10
|
|||
|
|
}
|
|||
|
|
return &EquipUploadService{
|
|||
|
|
db: db,
|
|||
|
|
mom: mom,
|
|||
|
|
hour: hour,
|
|||
|
|
deviceTypes: deviceTypes,
|
|||
|
|
stopCh: make(chan struct{}),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Start 启动后台循环。每分钟检查一次，命中 hour 整点且当日未跑则上传。
|
|||
|
|
func (s *EquipUploadService) Start(ctx context.Context) {
|
|||
|
|
if s.db == nil || s.mom == nil {
|
|||
|
|
slog.Info("equip_upload: skip (db or mom not configured)")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
go func() {
|
|||
|
|
t := time.NewTicker(1 * time.Minute)
|
|||
|
|
defer t.Stop()
|
|||
|
|
slog.Info("equip_upload: started", "hour", s.hour, "deviceTypes", s.deviceTypes)
|
|||
|
|
for {
|
|||
|
|
select {
|
|||
|
|
case <-s.stopCh:
|
|||
|
|
return
|
|||
|
|
case <-ctx.Done():
|
|||
|
|
return
|
|||
|
|
case now := <-t.C:
|
|||
|
|
s.tick(ctx, now)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *EquipUploadService) Stop() {
|
|||
|
|
close(s.stopCh)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *EquipUploadService) tick(ctx context.Context, now time.Time) {
|
|||
|
|
if now.Hour() != s.hour {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
today := now.Format("2006-01-02")
|
|||
|
|
if s.lastRunDate == today {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
yesterday := now.AddDate(0, 0, -1).Format("2006-01-02")
|
|||
|
|
if err := s.uploadDay(ctx, yesterday); err != nil {
|
|||
|
|
slog.Error("equip_upload: failed", "date", yesterday, "error", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
s.lastRunDate = today
|
|||
|
|
slog.Info("equip_upload: success", "date", yesterday)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UploadDay 对外暴露的手动触发入口（可用于补传）。
|
|||
|
|
func (s *EquipUploadService) UploadDay(ctx context.Context, date string) error {
|
|||
|
|
if s.db == nil || s.mom == nil {
|
|||
|
|
return fmt.Errorf("db or mom not configured")
|
|||
|
|
}
|
|||
|
|
return s.uploadDay(ctx, date)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// uploadDay 汇总 date(YYYY-MM-DD) 的设备状态时长并调 MOM
|
|||
|
|
func (s *EquipUploadService) uploadDay(ctx context.Context, date string) error {
|
|||
|
|
rows, err := s.queryDurations(ctx, date)
|
|||
|
|
if err != nil {
|
|||
|
|
return fmt.Errorf("query durations: %w", err)
|
|||
|
|
}
|
|||
|
|
if len(rows) == 0 {
|
|||
|
|
slog.Info("equip_upload: no data", "date", date)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// device_name → (status → seconds)
|
|||
|
|
agg := make(map[string]map[string]float64)
|
|||
|
|
for _, r := range rows {
|
|||
|
|
if agg[r.DeviceName] == nil {
|
|||
|
|
agg[r.DeviceName] = make(map[string]float64)
|
|||
|
|
}
|
|||
|
|
agg[r.DeviceName][r.Status] += r.DurationSec
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for deviceName, statusMap := range agg {
|
|||
|
|
work, open := computeWorkOpen(statusMap)
|
|||
|
|
wait := open - work
|
|||
|
|
if wait < 0 {
|
|||
|
|
wait = 0
|
|||
|
|
}
|
|||
|
|
equipID := extractEquipID(deviceName)
|
|||
|
|
// 每条记录一个独立 ID，便于排查
|
|||
|
|
uid := fmt.Sprintf("%s-%s-%d", date, deviceName, time.Now().UnixNano())
|
|||
|
|
err := s.mom.InsertEquipUpload(ctx, momlib.EquipUpload{
|
|||
|
|
ID: uid,
|
|||
|
|
// ResourceCode 由 SDK 从 MomConf 自动回填
|
|||
|
|
EquipId: equipID,
|
|||
|
|
WorkDate: date,
|
|||
|
|
OpenTime: hours(open),
|
|||
|
|
WorkTime: hours(work),
|
|||
|
|
WaitTime: hours(wait),
|
|||
|
|
SumCost: hours(work),
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
slog.Error("equip_upload: insert failed",
|
|||
|
|
"device", deviceName, "date", date, "error", err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
slog.Info("equip_upload: uploaded",
|
|||
|
|
"device", deviceName, "equipId", equipID, "date", date,
|
|||
|
|
"openH", hours(open), "workH", hours(work), "waitH", hours(wait))
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type statusRow struct {
|
|||
|
|
DeviceName string
|
|||
|
|
Status string
|
|||
|
|
DurationSec float64
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// queryDurations 查询指定日期 [date 00:00, date+1 00:00) 区间内每个设备每个状态的总秒数
|
|||
|
|
func (s *EquipUploadService) queryDurations(ctx context.Context, date string) ([]statusRow, error) {
|
|||
|
|
// 如果 deviceTypes 为空，只查 OKUMA 类（与 dashboard-server 保持一致）
|
|||
|
|
deviceTypes := s.deviceTypes
|
|||
|
|
if len(deviceTypes) == 0 {
|
|||
|
|
deviceTypes = []string{"osp-mc", "osp-lathe", "okuma_mc", "okuma_lathe"}
|
|||
|
|
}
|
|||
|
|
placeholders := make([]string, len(deviceTypes))
|
|||
|
|
args := []any{date, date}
|
|||
|
|
for i, dt := range deviceTypes {
|
|||
|
|
placeholders[i] = fmt.Sprintf("$%d", i+3)
|
|||
|
|
args = append(args, dt)
|
|||
|
|
}
|
|||
|
|
q := fmt.Sprintf(`
|
|||
|
|
SELECT r.device_name, r.status,
|
|||
|
|
EXTRACT(EPOCH FROM (
|
|||
|
|
LEAST(COALESCE(r.end_time, NOW()), ($1::date + INTERVAL '1 day')) -
|
|||
|
|
GREATEST(r.start_time, $2::date)
|
|||
|
|
)) AS duration_sec
|
|||
|
|
FROM dc_device_status_record r
|
|||
|
|
WHERE r.start_time < ($1::date + INTERVAL '1 day')
|
|||
|
|
AND COALESCE(r.end_time, NOW()) > $2::date
|
|||
|
|
AND EXISTS (
|
|||
|
|
SELECT 1 FROM dc_device_status d
|
|||
|
|
WHERE d.device_name = r.device_name
|
|||
|
|
AND d.device_type IN (%s)
|
|||
|
|
)
|
|||
|
|
`, strings.Join(placeholders, ","))
|
|||
|
|
|
|||
|
|
rs, err := s.db.QueryContext(ctx, q, args...)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer rs.Close()
|
|||
|
|
var out []statusRow
|
|||
|
|
for rs.Next() {
|
|||
|
|
var row statusRow
|
|||
|
|
if err := rs.Scan(&row.DeviceName, &row.Status, &row.DurationSec); err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
if row.DurationSec > 0 {
|
|||
|
|
out = append(out, row)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return out, rs.Err()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// computeWorkOpen 跟 dashboard-server/handlers/oee.go 中的 OEE 计算逻辑保持一致：
|
|||
|
|
// - work：RUN 状态（含 ONLINE 兜底——非 OSP 设备的连接级状态）
|
|||
|
|
// - open：非 OFFLINE 的所有时段（即设备开机时段）
|
|||
|
|
func computeWorkOpen(statusMap map[string]float64) (workSec, openSec float64) {
|
|||
|
|
for st, sec := range statusMap {
|
|||
|
|
if st == "OFFLINE" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
openSec += sec
|
|||
|
|
if st == "RUN" || strings.Contains(st, "RUN") || st == "ONLINE" {
|
|||
|
|
workSec += sec
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func hours(sec float64) float64 {
|
|||
|
|
h := sec / 3600.0
|
|||
|
|
// 保留 2 位小数避免 SDK / DM 端浮点抖动
|
|||
|
|
return float64(int(h*100+0.5)) / 100.0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var equipIDDigits = regexp.MustCompile(`(\d+)`)
|
|||
|
|
|
|||
|
|
// extractEquipID 从 device_name 提取尾部数字作为 MOM EquipId（float64）。
|
|||
|
|
// 例如 "CNC1"→1，"OKUMA-MC-2"→2；提取不到则返回 0。
|
|||
|
|
func extractEquipID(deviceName string) float64 {
|
|||
|
|
matches := equipIDDigits.FindAllString(deviceName, -1)
|
|||
|
|
if len(matches) == 0 {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
// 取最后一段数字，更贴近"序号"语义
|
|||
|
|
v, err := strconv.Atoi(matches[len(matches)-1])
|
|||
|
|
if err != nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return float64(v)
|
|||
|
|
}
|