Method getUsersFromUserAttributes

backend/plugin/db/tidb/role.go:73–103

getUsersFromUserAttributes reads users from information_schema.user_attributes, returns the list of users with format "' '@' '".

(ctx context.Context)

// getUsersFromUserAttributes reads users from information_schema.user_attributes, returns the list of users with format "'<user>'@'<host>'".
func (d *Driver) getUsersFromUserAttributes(ctx context.Context) ([]string, error) {
	var users []string
	query := `
	SELECT
		user,
		host
	FROM information_schema.user_attributes
	WHERE user <> ''
		AND user NOT LIKE 'mysql.%'
	`
	roleRows, err := d.db.QueryContext(ctx, query)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to query information_schema.user_attributes")
	}
	defer roleRows.Close()
	for roleRows.Next() {
		var user string
		var host string
		if err := roleRows.Scan(
			&user,
			&host,
		); err != nil {
			return nil, err
		}
		users = append(users, fmt.Sprintf("'%s'@'%s'", user, host))
	}
	if err := roleRows.Err(); err != nil {
		return nil, errors.Wrapf(err, "failed to iterate information_schema.user_attributes")
	}
	return users, nil
}

// getUsersFromMySQLUser reads users from mysql.user, returns the list of users with format "'<user>'@'<host>'".
func (d *Driver) getUsersFromMySQLUser(ctx context.Context) ([]string, error) {