fix memory.Writer

memory/writer.go

@@ -39,8 +39,8 @@ func (w *Writer) WriteValue(val interface{}) error {
 	switch w.stack[last].(type) {
 	case *Node:
 		(*w.stack[last].(*Node))[w.curKey] = val
-	case []interface{}:
-		w.stack[last] = append(w.stack[last].([]interface{}), val)
+	case *[]interface{}:
+		*w.stack[last].(*[]interface{}) = append(*w.stack[last].(*[]interface{}), val)
 	default:
 		panic("Currupted stack")
 	}
@@ -60,9 +60,9 @@ func (w *Writer) WriteBeginNode(n_elems int) error {
 		w.stack = append(w.stack, &w.Node)
 		return nil
 	} else {
-		n := &Node{}
+		n := Node{}
 		err := w.WriteValue(n)
-		w.stack = append(w.stack, n)
+		w.stack = append(w.stack, &n)
 		return err
 	}
 }
@@ -86,7 +86,8 @@ func (w *Writer) WriteBeginArray(n_elems int) error {
 		return fmt.Errorf("Cannot start array")
 	}
 
-	w.stack = append(w.stack, []interface{}{})
+	slice := []interface{}{}
+	w.stack = append(w.stack, w.curKey, &slice)
 	return nil
 }
 
@@ -95,6 +96,9 @@ func (w *Writer) WriteEndArray() error {
 		return fmt.Errorf("Cannot end array")
 	}
 
-	w.stack = w.stack[:len(w.stack)-1]
-	return nil
+	slice := w.stack[len(w.stack)-1].(*[]interface{})
+	w.curKey = w.stack[len(w.stack)-2].(string)
+	w.stack = w.stack[:len(w.stack)-2]
+	err := w.WriteValue(*slice)
+	return err
 }