chore: import upstream snapshot with attribution

testing/go/benchmark/section.go

@@ -0,0 +1,71 @@
+// Copyright 2024 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"strconv"
+	"strings"
+)
+
+// Section represents a benchmark section.
+type Section struct {
+	Test string
+	Time float64 // This is in number of iterations per second
+}
+
+// SectionResults creates a section for each test.
+func SectionResults(fileData string) map[string]Section {
+	sections := make(map[string]Section)
+	for {
+		headerStartIdx := strings.Index(fileData, `----`)
+		if headerStartIdx == -1 {
+			break
+		}
+		headerEndIdx := strings.Index(fileData[headerStartIdx+4:], `----`) + headerStartIdx + 4
+		if headerEndIdx == headerStartIdx+4 {
+			break
+		}
+		headerFull := fileData[headerStartIdx : headerEndIdx+4]
+		endingHeaderIdx := strings.LastIndex(fileData, headerFull)
+		if endingHeaderIdx == -1 {
+			break
+		}
+		section := Section{
+			Test: headerFull[4 : len(headerFull)-4],
+			Time: -1,
+		}
+		sectionText := strings.TrimSpace(fileData[len(headerFull):endingHeaderIdx])
+		fileData = fileData[endingHeaderIdx+len(headerFull):]
+		for _, line := range strings.Split(sectionText, "\n") {
+			if strings.Contains(line, `queries:`) {
+				parenIdx := strings.Index(line, `(`)
+				perSecIdx := strings.Index(line, ` per sec.)`)
+				if parenIdx != -1 && perSecIdx != -1 {
+					timeString := line[parenIdx+1 : perSecIdx]
+					parsedTime, err := strconv.ParseFloat(timeString, 64)
+					if err == nil {
+						section.Time = parsedTime
+					}
+				}
+				break
+			}
+		}
+		if section.Time == -1 {
+			continue
+		}
+		sections[section.Test] = section
+	}
+	return sections
+}