package dbutil

import (
	"gorm.io/gorm"

	"github.com/pingcap/errors"
	"github.com/tidbcloud/dataflow-service/internal/db/entity"
)

func GetUserWithClusterID(tx *gorm.DB, clusterID entity.ID) (entity.DBUser, error) {
	var user entity.DBUser
	err := entity.NewDBUserQuerySet(tx).
		ClusterIDEq(clusterID).
		One(&user)
	if err != nil {
		return user, errors.Annotate(TagNotFound(err), "GetUserWithClusterIDAndRoleName")
	}

	return user, nil
}