Helper functions to read path elements.

functions.go

@@ -0,0 +1,35 @@
+package shuttle
+
+import (
+	"strconv"
+	"strings"
+)
+
+func ReadPathElement(rawPath, upstreamElement string) string {
+	var nextDocument bool
+	var document string
+	for len(rawPath) > 0 {
+		slashIndex := strings.Index(rawPath, "/")
+		if slashIndex >= 0 {
+			document, rawPath = rawPath[:slashIndex], rawPath[slashIndex+1:]
+		} else {
+			document = rawPath
+			rawPath = ""
+		}
+
+		if nextDocument {
+			return document
+		}
+
+		if document == upstreamElement {
+			nextDocument = true
+		}
+	}
+
+	return ""
+}
+func ReadNumericPathElement(rawPath, upstreamElement string) uint64 {
+	raw := ReadPathElement(rawPath, upstreamElement)
+	value, _ := strconv.ParseUint(raw, 10, 64)
+	return value
+}

functions_test.go

@@ -0,0 +1,25 @@
+package shuttle
+
+import "testing"
+
+func TestReadPathElement(t *testing.T) {
+	assertPathElement(t, "/path/users/value", "users", "value")
+	assertPathElement(t, "/path/users/value/", "users", "value")
+	assertPathElement(t, "/path/users/value/other/stuff", "users", "value")
+}
+func assertPathElement(t *testing.T, raw, element, expected string) {
+	Assert(t).That(ReadPathElement(raw, element)).Equals(expected)
+}
+
+func TestReadNumericPathElement(t *testing.T) {
+	assertNumericPathElement(t, "/path/users/123", "users", 123)
+	assertNumericPathElement(t, "/path/users/123/", "users", 123)
+	assertNumericPathElement(t, "/path/users/123/other/stuff", "users", 123)
+	assertNumericPathElement(t, "/path/users/-123/other/stuff", "users", 0)
+	assertNumericPathElement(t, "/path/users/abc/other/stuff", "users", 0)
+	assertNumericPathElement(t, "/path/users/abc123/other/stuff", "users", 0)
+	assertNumericPathElement(t, "/path/users/_123/other/stuff", "users", 0)
+}
+func assertNumericPathElement(t *testing.T, raw, element string, expected uint64) {
+	Assert(t).That(ReadNumericPathElement(raw, element)).Equals(expected)
+}