package db

import (
	cortypes "gitlink.org.cn/cloudream/jcs-pub/coordinator/types"
)

type HubLocationDB struct {
	*DB
}

func (db *DB) HubLocation() *HubLocationDB {
	return &HubLocationDB{DB: db}
}

func (*HubLocationDB) GetByHubID(ctx SQLContext, id cortypes.HubID) ([]cortypes.HubLocation, error) {
	var ret []cortypes.HubLocation
	err := ctx.Where("HubID = ?", id).Find(&ret).Error
	return ret, err
}

func (*HubLocationDB) GetAll(ctx SQLContext) ([]cortypes.HubLocation, error) {
	var ret []cortypes.HubLocation
	err := ctx.Find(&ret).Error
	return ret, err
}