suyiiyii/yq
1
0
Fork 0
mirror of https://github.com/mikefarah/yq.git synced 2024-12-19 20:19:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

gofmt'ed

Browse Source
This commit is contained in:
mfarah 2015-09-28 12:00:38 +10:00
parent ccd95146fe
commit 9869a41122
2 changed files with 85 additions and 86 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
yaml.go
View File

@ -2,13 +2,13 @@ package main
import (
"fmt"
"gopkg.in/yaml.v2"
"log"
"io/ioutil"
"os"
"github.com/codegangsta/cli"
"strings"
"gopkg.in/yaml.v2"
"io/ioutil"
"log"
"os"
"strconv"
"strings"
)
func main() {
@ -64,7 +64,7 @@ func read_file(filename string) []byte {
func read_map(context map[interface{}]interface{}, head string, tail []string) interface{} {
value := context[head]
if (len(tail) > 0) {
if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)])
} else {
return value
@ -73,23 +73,22 @@ func read_map(context map[interface{}]interface{}, head string, tail []string) i
func recurse(value interface{}, head string, tail []string) interface{} {
switch value.(type) {
case []interface {}:
case []interface{}:
index, err := strconv.ParseInt(head, 10, 64)
if err != nil {
log.Fatalf("Error accessing array: %v", err)
}
return read_array(value.([]interface {}), index, tail)
return read_array(value.([]interface{}), index, tail)
default:
return read_map(value.(map[interface{}]interface{}), head, tail)
}
}
func read_array(array []interface {}, head int64, tail[]string) interface{} {
func read_array(array []interface{}, head int64, tail []string) interface{} {
value := array[head]
if (len(tail) > 0) {
if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)])
} else {
return value
}
}

10
yaml_test.go
View File

@ -1,13 +1,13 @@
package main
import (
"testing"
"gopkg.in/yaml.v2"
"fmt"
"gopkg.in/yaml.v2"
"os"
"testing"
)
var raw_data = `
var raw_data = `
a: Easy!
b:
c: 2
@ -28,14 +28,14 @@ func TestMain(m *testing.M) {
func TestRead_map_simple(t *testing.T) {
result := read_map(parsed_data, "b", []string{"c"})
if( result != 2) {
if result != 2 {
t.Error("Excpted 2 but got ", result)
}
}
func TestRead_map_array(t *testing.T) {
result := read_map(parsed_data, "b", []string{"d", "1"})
if( result != 4) {
if result != 4 {
t.Error("Excpted 4 but got ", result)
}
}