attribute.go

package saml2

import "github.com/russellhaering/gosaml2/types"

// Values is a convenience wrapper for a map of strings to Attributes, which
// can be used for easy access to the string values of Attribute lists.
type Values map[string]types.Attribute

// Get is a safe method (nil maps will not panic) for returning the first value
// for an attribute at a key, or the empty string if none exists.
func (vals Values) Get(k string) string {
	if vals == nil {
		return ""
	}
	if v, ok := vals[k]; ok && len(v.Values) > 0 {
		return string(v.Values[0].Value)
	}
	return ""
}

//GetSize returns the number of values for an attribute at a key.
//Returns '0' in case of error or if key is not found.
func (vals Values) GetSize(k string) int {
    if vals == nil {
        return 0
    }

	v, ok := vals[k]
    if ok {
        return len(v.Values)
    }

    return 0
}

//GetAll returns all the values for an attribute at a key.
//Returns an empty slice in case of error of if key is not found.
func (vals Values) GetAll(k string) []string {
    var av []string

    if vals == nil {
        return av
    }

    if v, ok := vals[k]; ok && len(v.Values) > 0 {
        for i := 0; i < len(v.Values); i++ {
            av = append(av, string(v.Values[i].Value))
        }
    }

    return av
}