Convert repeated type even if it contains primitives

SuduIDE · Nov 1, 2023 · 1161372 · 1161372
1 parent 5bb4633
commit 1161372
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/generator/src/main/java/org/sudu/protogen/generator/type/RepeatedType.java b/generator/src/main/java/org/sudu/protogen/generator/type/RepeatedType.java
@@ -34,7 +34,7 @@ public RepeatedContainer getRepeatedType() {
 
     @Override
     public CodeBlock toGrpcTransformer(CodeBlock expr, Set<String> usedDefinitions) {
-        if (!(elementModel instanceof PrimitiveTypeModel)) {
+        if (!(elementModel instanceof PrimitiveTypeModel && repeatedType == RepeatedContainer.LIST)) {
             String nextDefinition = nextDefinition(usedDefinitions);
             CodeBlock lambdaParameter = CodeBlock.of(nextDefinition);
             Set<String> newDefinitions = new HashSet<>(usedDefinitions) {{ add(nextDefinition); }};
@@ -50,7 +50,7 @@ public CodeBlock toGrpcTransformer(CodeBlock expr, Set<String> usedDefinitions)
     @Override
     public CodeBlock fromGrpcTransformer(CodeBlock expr, Set<String> usedDefinitions) {
         expr = repeatedType.convertListToInstance(expr);
-        if (!(elementModel instanceof PrimitiveTypeModel)) {
+        if (!(elementModel instanceof PrimitiveTypeModel && repeatedType == RepeatedContainer.LIST)) {
             String nextDefinition = nextDefinition(usedDefinitions);
             CodeBlock lambdaParameter = CodeBlock.of(nextDefinition);
             Set<String> newDefinitions = new HashSet<>(usedDefinitions) {{ add(nextDefinition); }};

diff --git a/tests/src/test/proto/general.proto b/tests/src/test/proto/general.proto
@@ -60,7 +60,7 @@ message GrpcRepeatedTypes {
 }
 
 message GrpcSomeEvent {
-  repeated string ids = 1;
+  repeated string ids = 1 [(.protogen.repeated_container) = SET];
   option (.protogen.topic) = "SOME_TOPIC";
 }