grafana/pkg/services/teamguardian/database/database.go

package database

import (
	"context"

	"github.com/grafana/grafana/pkg/infra/db"
	"github.com/grafana/grafana/pkg/services/team"
)

type TeamGuardianStoreImpl struct {
	sqlStore    db.DB
	teamService team.Service
}

func ProvideTeamGuardianStore(sqlStore db.DB, teamService team.Service) *TeamGuardianStoreImpl {
	return &TeamGuardianStoreImpl{sqlStore: sqlStore, teamService: teamService}
}

func (t *TeamGuardianStoreImpl) GetTeamMembers(ctx context.Context, query team.GetTeamMembersQuery) ([]*team.TeamMemberDTO, error) {
	queryResult, err := t.teamService.GetTeamMembers(ctx, &query)
	if err != nil {
		return nil, err
	}

	return queryResult, nil
}

func (t *TeamGuardianStoreImpl) DeleteByUser(ctx context.Context, userID int64) error {
	return t.sqlStore.WithTransactionalDbSession(ctx, func(sess *db.Session) error {
		var rawSQL = "DELETE FROM team_member WHERE user_id = ?"
		_, err := sess.Exec(rawSQL, userID)
		return err
	})
}