X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=r.go;h=6ffa1cc077ac6607d8c8ac2329237407a61d7441;hb=75bdd5bb11f2798bd9a540e1de9f43f7f590e9b3;hp=9a4420d7b6968f8649c44d9ec3481227adf6896c;hpb=e3bf526c33e545694fafdf4e67ad7a92183d52c3;p=gorecfile.git

diff --git a/r.go b/r.go
index 9a4420d..6ffa1cc 100644
--- a/r.go
+++ b/r.go
@@ -1,6 +1,6 @@
 /*
 recfile -- GNU recutils'es recfiles parser on pure Go
-Copyright (C) 2020 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -38,11 +38,6 @@ func NewReader(r io.Reader) *Reader {
 	return &Reader{bufio.NewScanner(r)}
 }
 
-type Field struct {
-	Name  string
-	Value string
-}
-
 // Get next record. Each record is just a collection of fields. io.EOF
 // is returned if there is nothing to read more.
 func (r *Reader) Next() ([]Field, error) {
@@ -128,3 +123,29 @@ func (r *Reader) Next() ([]Field, error) {
 	}
 	return fields, nil
 }
+
+// Same as Next(), but with unique keys and last value.
+func (r *Reader) NextMap() (map[string]string, error) {
+	fields, err := r.Next()
+	if err != nil {
+		return nil, err
+	}
+	m := make(map[string]string, len(fields))
+	for _, f := range fields {
+		m[f.Name] = f.Value
+	}
+	return m, nil
+}
+
+// Same as Next(), but with unique keys and slices of values.
+func (r *Reader) NextMapWithSlice() (map[string][]string, error) {
+	fields, err := r.Next()
+	if err != nil {
+		return nil, err
+	}
+	m := make(map[string][]string)
+	for _, f := range fields {
+		m[f.Name] = append(m[f.Name], f.Value)
+	}
+	return m, nil
+}