JCS-pub/coordinator/internal/db/hub.go

package db

import (
	"time"

	jcstypes "gitlink.org.cn/cloudream/jcs-pub/common/types"
)

type HubDB struct {
	*DB
}

func (db *DB) Hub() *HubDB {
	return &HubDB{DB: db}
}

func (*HubDB) GetAllHubs(ctx SQLContext) ([]jcstypes.Hub, error) {
	var ret []jcstypes.Hub

	err := ctx.Table("Hub").Find(&ret).Error
	return ret, err
}

func (*HubDB) GetByID(ctx SQLContext, hubID jcstypes.HubID) (jcstypes.Hub, error) {
	var ret jcstypes.Hub
	err := ctx.Table("Hub").Where("HubID = ?", hubID).Find(&ret).Error

	return ret, err
}

func (*HubDB) BatchGetByID(ctx SQLContext, hubIDs []jcstypes.HubID) ([]jcstypes.Hub, error) {
	var ret []jcstypes.Hub
	err := ctx.Table("Hub").Where("HubID IN (?)", hubIDs).Find(&ret).Error

	return ret, err
}

// UpdateState 更新状态，并且设置上次上报时间为现在
func (*HubDB) UpdateState(ctx SQLContext, hubID jcstypes.HubID, state string) error {
	err := ctx.
		Model(&jcstypes.Hub{}).
		Where("HubID = ?", hubID).
		Updates(map[string]interface{}{
			"State":          state,
			"LastReportTime": time.Now(),
		}).Error
	return err
}