Skip to content

Commit b55ab89

Browse files
committed
Fixing java template for import statement - these are new additions and required handling for EnumSet
1 parent 51d6168 commit b55ab89

File tree

3 files changed

+15
-2
lines changed

3 files changed

+15
-2
lines changed

Templates/Java/models_generated/BaseMethodBody.java.tt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ import com.google.gson.annotations.SerializedName;
1111
import com.google.gson.JsonObject;
1212
import com.microsoft.graph.serializer.ISerializer;
1313
import com.microsoft.graph.models.extensions.*;
14+
import java.util.EnumSet;
1415

1516
<#=CreateClassDef(BaseTypeBody(c))#>
1617

Templates/Java/requests_generated/BaseEntityRequestBuilder.java.tt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import com.microsoft.graph.core.IBaseClient;
99
import com.microsoft.graph.http.BaseRequestBuilder;
1010
import com.microsoft.graph.options.Option;
11+
import java.util.EnumSet;
1112

1213
<#=CreateClassDef(BaseTypeRequestBuilder(c), "BaseRequestBuilder", IBaseTypeRequestBuilder(c))#>
1314

@@ -225,10 +226,15 @@ if (c.AsOdcmClass() != null)
225226
foreach (var p in method.Parameters)
226227
{
227228
if(!(p.Type is OdcmPrimitiveType) && p.Type.GetTypeString() != "com.google.gson.JsonElement") {
229+
var typeString = p.Type.GetTypeString();
230+
if(p.Type.GetTypeString().StartsWith("EnumSet<")){
231+
typeString = typeString.Substring(typeString.IndexOf("<") + 1);
232+
typeString = typeString.Substring(0, typeString.Length-1);
233+
}
228234
sb.AppendFormat(importFormat,
229235
host.CurrentModel.NamespaceName(),
230236
getPackagePrefix(p),
231-
p.Type.GetTypeString());
237+
typeString);
232238
sb.Append("\n");
233239
}
234240
}

Templates/Java/requests_generated/IBaseEntityRequestBuilder.java.tt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
<#=CreatePackageDefForIBaseEntityRequestBuilder(host)#>
88
import com.microsoft.graph.http.IRequestBuilder;
99
import com.microsoft.graph.options.Option;
10+
import java.util.EnumSet;
1011

1112
<#=CreateInterfaceDef(IBaseTypeRequestBuilder(c), "IRequestBuilder")#>
1213
/**
@@ -171,10 +172,15 @@ if (c.AsOdcmClass() != null)
171172
foreach (var p in method.Parameters)
172173
{
173174
if(!(p.Type is OdcmPrimitiveType) && p.Type.GetTypeString() != "com.google.gson.JsonElement") {
175+
var typeString = p.Type.GetTypeString();
176+
if(p.Type.GetTypeString().StartsWith("EnumSet<")){
177+
typeString = typeString.Substring(typeString.IndexOf("<") + 1);
178+
typeString = typeString.Substring(0, typeString.Length-1);
179+
}
174180
sb.AppendFormat(importFormat,
175181
host.CurrentModel.NamespaceName(),
176182
getPackagePrefix(p),
177-
p.Type.GetTypeString());
183+
typeString);
178184
sb.Append("\n");
179185
}
180186
}

0 commit comments

Comments
 (0)