7c1282711b
* INFRA-2873 : Boilerplate setup for team management revamp: * INFRA-2873 : Complete till add team flow * INFRA-2873 : Added unit tests and migration scripts * INFRA-2873 : Code review comments * INFRA-2873 : Add getter for team severity * INFRA-2873 : Second round of review
31 lines
879 B
Go
31 lines
879 B
Go
package teamUserSeverity
|
|
|
|
import (
|
|
"houston/model/teamSeverity"
|
|
"houston/model/teamUser"
|
|
)
|
|
|
|
type TeamUserSeverityEntity struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
TeamUser uint `gorm:"column:team_user;not null"`
|
|
TeamSeverity uint `gorm:"column:team_severity;not null"`
|
|
|
|
// Add foreign key constraints
|
|
TeamUserEntity teamUser.TeamUserEntity `gorm:"foreignKey:TeamUser"`
|
|
TeamSeverityEntity teamSeverity.TeamSeverityEntity `gorm:"foreignKey:TeamSeverity"`
|
|
}
|
|
|
|
func (TeamUserSeverityEntity) TableName() string {
|
|
return "team_user_severity"
|
|
}
|
|
|
|
func (entity TeamUserSeverityEntity) ToDTO() TeamUserSeverityDTO {
|
|
return TeamUserSeverityDTO{
|
|
ID: entity.ID,
|
|
TeamUser: entity.TeamUser,
|
|
TeamSeverity: entity.TeamSeverity,
|
|
TeamUserDTO: *entity.TeamUserEntity.ToDTO(),
|
|
TeamSeverityDTO: entity.TeamSeverityEntity.ToDTO(),
|
|
}
|
|
}
|