more API on map & array

Author: Chang Hua Ou

simpleyaml.go

@@ -82,6 +82,21 @@ func (y *Yaml) Array() ([]interface{}, error) {
 	return nil, errors.New("type assertion to []interface{} failed")
 }
 
+func (y *Yaml) IsArray() bool {
+	_, err := y.Array()
+
+	return err == nil
+}
+
+// return the size of array
+func (y *Yaml)GetArraySize() (int, error ) {
+	a, err := y.Array()
+	if err != nil {
+		return 0, err
+	}
+	return len( a ), nil
+}
+
 // GetIndex returns a pointer to a new `Yaml` object.
 // for `index` in its `array` representation
 //
@@ -135,3 +150,25 @@ func (y *Yaml) Map() (map[interface{}]interface{}, error) {
 	}
 	return nil, errors.New("type assertion to map[interface]interface{} failed")
 }
+
+// Check if it is a map
+func (y *Yaml)IsMap() bool {
+	_, err := y.Map()
+	return err == nil
+}
+
+// Get all the keys of the map
+func (y *Yaml)GetMapKeys() ([]string,error) {
+	m, err := y.Map()
+
+	if err != nil {
+		return nil, err
+	}
+	keys := make([]string, 0 )
+	for k, _ := range m {
+		if s,ok := k.(string) ; ok {
+			keys = append( keys, s )
+		}
+	}
+	return keys, nil
+}

simpleyaml_test.go

@@ -138,3 +138,22 @@ func TestArray(t *testing.T) {
 		t.Fatal("emails length not equal 2")
 	}
 }
+
+
+func TestMap(t *testing.T) {
+	y, err := NewYaml( data )
+	if err != nil {
+		t.Fatal("init yaml failed")
+	}
+	if !y.IsMap() {
+		t.Fatal( "map check failed")
+	}
+
+	keys, err := y.GetMapKeys()
+	if err != nil {
+		t.Fatal( "get keys from map is failed" )
+	}
+	if len( keys ) != 6 {
+		t.Fatal( "fail to check number of keys")
+	}
+}