zz

csvparser.go

@@ -8,86 +8,136 @@ import (
 
 type CsvParser struct {
 	enclosers       []string
-	delimiter rune
+	delimiters      string
+	mergeDelimiters bool
 	fields          []string
+	ignore          string
+	maxFieldIndex   int
 }
+func (p *csvParser) FieldCount() int {
+  return maxFieldIndex + 1
+}
+/*
+* delimiters: string with all delimiter chars
+* mergeDelimiters: if true, all successive delimiters are considered as one
+* enclosers: array of string, each string contains a enclorser tuple: "", [], {}...
+* lineFormat: format of the line (each fieldname or ignore)
+ */
+func (p *CsvParser) Initialize(delimiters string, mergeDelimiters bool, enclosers []string, lineFormat string, ignore string) error {
 
-func (p *CsvParser) Initialize(delimiter string, enclosers []string, lineFormat string) error {
+	if utf8.RuneCountInString(delimiters) == 0 {
-	if utf8.RuneCountInString(delimiter) != 1 {
+		return fmt.Errorf("delimiter shoud be at least one character")
-		return fmt.Errorf("delimiter shoud be one character")
 	}
-	p.enclosers = make([]string, 0)
+	p.delimiters = delimiters
+	p.mergeDelimiters = mergeDelimiters
 	for _, encloser := range enclosers {
 		if utf8.RuneCountInString(encloser) != 2 {
 			return fmt.Errorf("encolser should have to characters")
 		}
 	}
 	p.enclosers = enclosers
-	p.delimiter = []rune(delimiter)[0]
+
-	for _, pair := range enclosers {
+	p.ignore = ignore
-		if utf8.RuneCountInString(pair) != 2 {
-			return fmt.Errorf("encoloser should contain two characters: %s", pair)
-		}
-	}
 	// line format is in the form of: field1 field2 ignore ...
 	// if field name is ignore, it is parsed but not retained
-	p.fields = strings.Split(lineFormat, " ")
+	fields := strings.Split(lineFormat, " ")
 
+	p.fields = make([]string, 0)
+	for _, f := range fields {
+		if len(f) > 0 {
+			p.fields = append(p.fields, f)
+		}
+	}
+	p.maxFieldIndex = len(p.fields) - 1
 	return nil
 }
 
+type ParserState struct {
+	inField       bool // reading field value
+	delimiter     bool // last char was a delimiter
+	escape        bool // the previaus char was a escaper \\
+	enclosed      bool // the current field is enclosed
+	enclosedMode  bool // the current algo is for enclosed string, do not search delimiter
+	encloserStart rune // the current enclosed field opener char
+	encloserEnd   rune // the current enclosed field closer char
+}
+
+/*
+* si pas dans la lecture d'un champ
+ */
 func (p *CsvParser) Parse(line string) (map[string]string, error) {
-	inEnclosedField := false
-	currentEncloserEnd := ' '
-	escape := false
 	currentFieldIndex := 0
+	valueStart := -1
+
+	state := ParserState{inField: false, delimiter: true, escape: false, enclosed: false, enclosedMode: false, encloserStart: '?', encloserEnd: '?'}
 
 	ret := make(map[string]string)
-	valueStart := 0
+
-	indexMax := len(line) - 1
+	maxIndex := len(line) - 1
 	for index, r := range line {
-		if index == indexMax {
-			if currentFieldIndex < len(p.fields) {
-				//fmt.Println("start:", valueStart, "end:", index)
-				//fmt.Println("Found a field value for:", p.fields[currentFieldIndex], line[valueStart:index])
-				if inEnclosedField && r == currentEncloserEnd {
-					ret[p.fields[currentFieldIndex]] = line[valueStart:index]
-				} else {
-					ret[p.fields[currentFieldIndex]] = line[valueStart : index+1]
-				}
-
-				//fmt.Println("Index is:", index)
-			}
-		}
 		if r == '\\' {
-			escape = !escape
+			state.escape = true
-		} else if inEnclosedField {
+			continue
-			if r == currentEncloserEnd && !escape {
-				inEnclosedField = false
 		}
-		} else if r == p.delimiter {
+		// previous rune was \\: ignore the current rune
-
+		if state.escape {
-			if currentFieldIndex < len(p.fields) {
+			state.escape = false
-				//fmt.Println("start:", valueStart, "end:", index)
+			continue
-				//fmt.Println("Found a field value for:", p.fields[currentFieldIndex], line[valueStart:index])
-				ret[p.fields[currentFieldIndex]] = line[valueStart:index]
-				//fmt.Println("Index is:", index)
-				valueStart = index + 1
 		}
-			currentFieldIndex++
+		// reading en enclosed field: watch for end of this field (encloserEnd)
-		} else {
+		if state.enclosedMode {
-			for _, encloser := range p.enclosers {
+			// searching for end of encloser
-				runes := []rune(encloser)
+			if r == state.encloserEnd {
-				if r == runes[0] {
+				state.enclosedMode = false
-					// opening encloser
+			}
-					inEnclosedField = true
+			if index < maxIndex {
-					currentEncloserEnd = runes[1]
+				continue
-					valueStart++
+			}
+		}
+		isDelimiter := false
+		for _, d := range p.delimiters {
+			if r == d {
+				isDelimiter = true
 				break
 			}
 		}
-		}
+
+		// previous rune was a delimiter and mergeDelimiters is true, skip
+		if isDelimiter && state.delimiter && p.mergeDelimiters {
+			continue
 		}
 
+		// current rune is a delimiter, a value is present and the next char is the next value
+		if isDelimiter || index == maxIndex {
+			if index == maxIndex && valueStart == -1 {
+				valueStart = 0
+			}
+			state.delimiter = true
+			field := line[valueStart : index+1]
+			field = strings.TrimRight(field, p.delimiters)
+			if state.enclosed {
+				field = strings.TrimRight(strings.TrimLeft(field, string(state.encloserStart)), string(state.encloserEnd))
+			}
+			if currentFieldIndex <= p.maxFieldIndex && p.fields[currentFieldIndex] != p.ignore {
+				ret[p.fields[currentFieldIndex]] = field
+			}
+			currentFieldIndex++
+			state = ParserState{inField: false, delimiter: true, escape: false, enclosed: false, enclosedMode: false, encloserStart: '?', encloserEnd: '?'}
+			continue
+		}
+		if state.delimiter {
+			valueStart = index
+			for _, e := range p.enclosers {
+				runes := []rune(e)
+				if r == runes[0] {
+					state.enclosed = true
+					state.enclosedMode = true
+					state.encloserStart = runes[0]
+					state.encloserEnd = runes[1]
+				}
+				state.delimiter = false
+			}
+		}
+	}
 	return ret, nil
 }

csvparser_test.go

@@ -7,8 +7,9 @@ import (
 
 func TestCorrectLines(t *testing.T) {
 	var csvParser CsvParser
-	csvParser.Initialize(" ", []string{"\"\"", "[]"}, "firstname lastname complete_name")
 	line := ""
+	/*
+		csvParser.Initialize(" ", false, []string{"\"\"", "[]"}, "firstname lastname complete_name")
 		fmt.Println("parsing:", line)
 		event, err := csvParser.Parse(line)
 		if err != nil {
@@ -48,6 +49,14 @@ func TestCorrectLines(t *testing.T) {
 		}
 		fmt.Println("event is", event)
 
+		line = "John Doe \"John Doe\"\\\\"
+		fmt.Println("parsing:", line)
+		event, err = csvParser.Parse(line)
+		if err != nil {
+			t.Fatalf("Parsing of empty line failed %v %v", err, event)
+		}
+		fmt.Println("event is", event)
+
 		line = "John Doe I don't know him"
 		fmt.Println("parsing:", line)
 		event, err = csvParser.Parse(line)
@@ -55,4 +64,60 @@ func TestCorrectLines(t *testing.T) {
 			t.Fatalf("Parsing of empty line failed %v %v", err, event)
 		}
 		fmt.Println("event is", event)
+  */
+	csvParser.Initialize(" \t", true, []string{"\"\"", "[]"}, "ignore ignore ignore proxy ignore domain clientip ignore ignore apache-date request status bytes duration referer user-agent", "ignore")
+	line = "Aug  7 00:00:00 proxy-4 haproxy[17429]: www.yvelines.gouv.fr 66.249.64.10 - - [06/Aug/2024:23:59:59 +0200] \"GET /content/download/19274/117923/file/SE_EAU_20190325_LesJardines_78201900027_LetNotifCompletude+recepisse.pdf HTTP/1.1\" 301 1414 240 \"\" \"Mozilla/5.0 (Linux; Android 6.0.1; Nexus 5X Build/MMB29P) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.6478.182 Mobile Safari/537.36 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)\" \"-\" \"GB\" \"15169\""
+	fmt.Println("parsing:", line)
+	event, err := csvParser.Parse(line)
+	if err != nil {
+		t.Fatalf("Parsing of empty line failed %v %v", err, event)
+	}
+	fmt.Println("event is", event)
+  /*
+	csvParser.Initialize(" \t", true, []string{"\"\"", "[]"}, "ignore ignore ignore proxy ignore domain clientip ignore ignore apache-date request status bytes duration referer user-agent", "ignore")
+	line = "Aug  7 00:00:00 proxy-4 haproxy[17429]: www.yvelines.gouv.fr 66.249.64.10 - - [06/Aug/2024:23:59:59 +0200] \"GET /content/download/19274/117923/file/SE_EAU_20190325_LesJardines_78201900027_LetNotifCompletude+recepisse.pdf HTTP/1.1\" 301 1414 240 \"\" \"Mozilla/5.0 (Linux; Android 6.0.1; Nexus 5X Build/MMB29P) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.6478.182 Mobile Safari/537.36 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)\" \"-\" \"GB\" 15169"
+	fmt.Println("parsing:", line)
+	event, err = csvParser.Parse(line)
+	if err != nil {
+		t.Fatalf("Parsing of empty line failed %v %v", err, event)
+	}
+	fmt.Println("event is", event)
+
+	csvParser.Initialize(" \t", true, []string{"\"\"", "[]"}, "ignore ignore ignore proxy ignore domain clientip ignore ignore apache-date request status bytes duration referer user-agent", "ignore")
+	line = "Aug  7 00:00:00 proxy-4 haproxy[17429]: www.yvelines.gouv.fr 66.249.64.10 - - [06/Aug/2024:23:59:59 +0200] \"GET /content/download/19274/117923/file/SE_EAU_20190325_LesJardines_78201900027_LetNotifCompletude+recepisse.pdf HTTP/1.1\" 301 1414 240 \"\" \"Mozilla/5.0 (Linux; Android 6.0.1; Nexus 5X Build/MMB29P) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.6478.182 Mobile Safari/537.36 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)\" \"-\" \"GB\" \"15169"
+	fmt.Println("parsing:", line)
+	event, err = csvParser.Parse(line)
+	if err != nil {
+		t.Fatalf("Parsing of empty line failed %v %v", err, event)
+	}
+	fmt.Println("event is", event)
+
+	csvParser.Initialize(" ", true, []string{"\"\""}, "verb uri http-version", "ignore")
+  line = "GET /Actions-de-l-Etat/Vos-aides/Particuliers/Pass-culture?_escaped_fragment_=/particuliers/page/R65575 HTTP/1.1"
+	fmt.Println("parsing:", line)
+	event, err = csvParser.Parse(line)
+	if err != nil {
+		t.Fatalf("Parsing of empty line failed %v %v", err, event)
+	}
+	fmt.Println("event is", event)
+	*/
+	csvParser.Initialize(" ", true, []string{"\"\""}, "verb uri http-version", "ignore")
+  line = "GET /Actions-de-l-Etat/Vos-aides/Particuliers/Pass-culture?_escaped_fragment_=/particuliers/page/R65575 HTTP/1.1"
+	fmt.Println("parsing:", line)
+	event, err = csvParser.Parse(line)
+	if err != nil {
+		t.Fatalf("Parsing of empty line failed %v %v", err, event)
+	}
+	fmt.Println("event is", event)
+
+	csvParser.Initialize(" ", true, []string{"\"\""}, "verb uri http-version", "ignore")
+  line = "G"
+	fmt.Println("parsing:", line)
+	event, err = csvParser.Parse(line)
+	if err != nil {
+		t.Fatalf("Parsing of empty line failed %v %v", err, event)
+	}
+	fmt.Println("event is", event)
+
+
 }

go.mod

@@ -1,3 +1,4 @@
 module git.passke.org/laurentu/csv-parser
 
+toolchain go1.22.1
 go 1.22