305d494902
* switch to using team service * trying to fix tests * more tests to fix * add missing teamtest package
35 lines
1011 B
Go
35 lines
1011 B
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
"github.com/grafana/grafana/pkg/services/team"
|
|
)
|
|
|
|
type TeamGuardianStoreImpl struct {
|
|
sqlStore sqlstore.Store
|
|
teamService team.Service
|
|
}
|
|
|
|
func ProvideTeamGuardianStore(sqlStore sqlstore.Store, teamService team.Service) *TeamGuardianStoreImpl {
|
|
return &TeamGuardianStoreImpl{sqlStore: sqlStore, teamService: teamService}
|
|
}
|
|
|
|
func (t *TeamGuardianStoreImpl) GetTeamMembers(ctx context.Context, query models.GetTeamMembersQuery) ([]*models.TeamMemberDTO, error) {
|
|
if err := t.teamService.GetTeamMembers(ctx, &query); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return query.Result, nil
|
|
}
|
|
|
|
func (t *TeamGuardianStoreImpl) DeleteByUser(ctx context.Context, userID int64) error {
|
|
return t.sqlStore.WithTransactionalDbSession(ctx, func(sess *sqlstore.DBSession) error {
|
|
var rawSQL = "DELETE FROM team_member WHERE user_id = ?"
|
|
_, err := sess.Exec(rawSQL, userID)
|
|
return err
|
|
})
|
|
}
|