Add RepeatedString

infiniteloopcloud · Oct 6, 2022 · 09a3a4b · 09a3a4b
1 parent bd5a4a0
commit 09a3a4b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -25,6 +25,7 @@ These are actually environment variables.
 Supported overwrite:
 
 - `TimeTime` -> `time.Time`
+- `RepeatedString` -> `[]string`
 
 #### Example
 

diff --git a/cmd/protoc-gen-go/internal_gengo/main.go b/cmd/protoc-gen-go/internal_gengo/main.go
@@ -927,6 +927,10 @@ func goTypeOverride(goType string) string {
 		case "TimeTime":
 			return "time.Time"
 		}
+
+		if strings.Contains(goType, "RepeatedString") {
+			return strings.ReplaceAll(goType, "RepeatedString", "[]string")
+		}
 	}
 	return goType
 }
diff --git a/cmd/protoc-gen-go/internal_gengo/test_data/test.proto b/cmd/protoc-gen-go/internal_gengo/test_data/test.proto
@@ -6,13 +6,15 @@ package proto;
 
 message Test {
   TimeTime created_at = 1;
-  map<uint64, string> map_field = 2;
+  map<uint64, RepeatedString> map_field = 2;
   string test = 3;
   String other = 43;
 }
 
 message TimeTime {}
 
+message RepeatedString {}
+
 message String {
   string string = 2;
 }