feat: support explictly null set in model(poc)

shenwen.yin · shenwen.yin · commit cca0cbd3bd1f · 2025-01-14T14:44:08.000+08:00
diff --git a/src/main/java/com/smartx/tower/App.java b/src/main/java/com/smartx/tower/App.java
@@ -0,0 +1,42 @@
+package com.smartx.tower;
+
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import com.smartx.tower.ApiClient;
+import com.smartx.tower.ApiException;
+import com.smartx.tower.ClientUtil;
+import com.smartx.tower.JSON;
+import com.smartx.tower.TaskUtil;
+import com.smartx.tower.api.BrickTopoApi;
+import com.smartx.tower.api.ContentLibraryVmTemplateApi;
+import com.smartx.tower.api.MetricsApi;
+import com.smartx.tower.api.VmApi;
+import com.smartx.tower.api.VmDiskApi;
+import com.smartx.tower.model.*;
+import java.io.IOException;
+import java.util.List;
+import java.util.ArrayList;
+import java.util.stream.Collectors;
+import java.util.Objects;
+
+public class App {
+    public static void main(String[] args) throws ApiException, IOException {
+        test();
+    }
+
+    public static void test() throws ApiException, IOException {
+        VmUpdateHostOptionsParams params = new VmUpdateHostOptionsParams();
+        VmUpdateHostOptionsParamsData data = new VmUpdateHostOptionsParamsData();
+        data.hostname(null).presentDnsServers();
+        params.setData(data);
+        Gson gson = new JSON().getGson();
+        System.out.println(gson.toJson(data));
+        System.out.println(gson.toJson(params));
+        VmUpdateHostOptionsParamsData data2 = gson.fromJson("{\"dns_servers\":[\"114.114.114.114\"]}",
+                VmUpdateHostOptionsParamsData.class);
+        VmUpdateHostOptionsParams params2 = gson.fromJson("{\"data\":{\"dns_servers\":[\"114.114.114.114\"]}}",
+                VmUpdateHostOptionsParams.class);
+        System.out.println(data2);
+        System.out.println(params2);
+    }
+}
diff --git a/src/main/java/com/smartx/tower/ConditionalNullable.java b/src/main/java/com/smartx/tower/ConditionalNullable.java
@@ -0,0 +1,78 @@
+package com.smartx.tower;
+
+import java.lang.reflect.Field;
+import java.util.HashSet;
+import java.util.Set;
+
+import com.google.gson.Gson;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.TypeAdapter;
+import com.google.gson.TypeAdapterFactory;
+import com.google.gson.annotations.SerializedName;
+import com.google.gson.reflect.TypeToken;
+import com.google.gson.stream.JsonReader;
+import com.google.gson.stream.JsonWriter;
+
+public class ConditionalNullable {
+    private final static String IS_PRESENT = "_isPresent_";
+
+    public static class ConditionalNullablePojo {
+        protected Set<String> _isPresent_ = new HashSet<String>();
+    }
+
+    public static class ConditionalNullablePojoAdapterFactory implements TypeAdapterFactory {
+        @Override
+        public <T> TypeAdapter<T> create(Gson gson, TypeToken<T> type) {
+            Class<? super T> rawType = type.getRawType();
+            if (!ConditionalNullablePojo.class.isAssignableFrom(rawType)) {
+                return null;
+            }
+            Set<String> jsonField = new HashSet<>();
+            Field[] fields = rawType.getDeclaredFields();
+            for (int i = 0; i < fields.length; i++) {
+                Field field = fields[i];
+                if (field.isAnnotationPresent(SerializedName.class)) {
+                    jsonField.add(field.getAnnotation(SerializedName.class).value());
+                }
+            }
+            return new TypeAdapter<T>() {
+                @Override
+                public void write(JsonWriter out, T value) throws java.io.IOException {
+                    if (value == null) {
+                        out.nullValue();
+                        return;
+                    }
+                    ConditionalNullablePojo pojo = (ConditionalNullablePojo) value;
+                    TypeAdapter<T> delegateAdapter = gson.getDelegateAdapter(ConditionalNullablePojoAdapterFactory.this,
+                            type);
+                    TypeAdapter<JsonElement> elementAdapter = gson.getAdapter(JsonElement.class);
+                    JsonObject jobj = delegateAdapter.toJsonTree(value).getAsJsonObject();
+                    for (String fieldName : jsonField) {
+                        if (jobj.has(fieldName) && jobj.get(fieldName).isJsonNull()
+                                && !pojo._isPresent_.contains(fieldName)) {
+                            jobj.remove(fieldName);
+                        }
+                    }
+                    if (jobj.has(IS_PRESENT)) {
+                        jobj.remove(IS_PRESENT);
+                    }
+                    if (!out.getSerializeNulls()) {
+                        out.setSerializeNulls(true);
+                        elementAdapter.write(out, jobj);
+                        out.setSerializeNulls(false);
+                    } else {
+                        elementAdapter.write(out, jobj);
+                    }
+                }
+
+                public T read(JsonReader in) throws java.io.IOException {
+                    JsonObject jsonObject = gson.getAdapter(JsonElement.class).read(in).getAsJsonObject();
+                    T instance = gson.getDelegateAdapter(ConditionalNullablePojoAdapterFactory.this, type)
+                            .fromJsonTree(jsonObject);
+                    return instance;
+                }
+            };
+        }
+    }
+}
diff --git a/src/main/java/com/smartx/tower/JSON.java b/src/main/java/com/smartx/tower/JSON.java
@@ -39,8 +39,7 @@ public class JSON {
 
     @SuppressWarnings("unchecked")
     public static GsonBuilder createGson() {
-        GsonFireBuilder fireBuilder = new GsonFireBuilder()
-        ;
+        GsonFireBuilder fireBuilder = new GsonFireBuilder();
         GsonBuilder builder = fireBuilder.createGsonBuilder();
         return builder;
     }
@@ -54,10 +53,13 @@ private static String getDiscriminatorValue(JsonElement readElement, String disc
     }
 
     /**
-     * Returns the Java class that implements the OpenAPI schema for the specified discriminator value.
+     * Returns the Java class that implements the OpenAPI schema for the specified
+     * discriminator value.
      *
-     * @param classByDiscriminatorValue The map of discriminator values to Java classes.
-     * @param discriminatorValue The value of the OpenAPI discriminator in the input data.
+     * @param classByDiscriminatorValue The map of discriminator values to Java
+     *                                  classes.
+     * @param discriminatorValue        The value of the OpenAPI discriminator in
+     *                                  the input data.
      * @return The Java class that implements the OpenAPI schema
      */
     private static Class getClassByDiscriminator(Map classByDiscriminatorValue, String discriminatorValue) {
@@ -70,12 +72,13 @@ private static Class getClassByDiscriminator(Map classByDiscriminatorValue, Stri
 
     public JSON() {
         gson = createGson()
-            .registerTypeAdapter(Date.class, dateTypeAdapter)
-            .registerTypeAdapter(java.sql.Date.class, sqlDateTypeAdapter)
-            .registerTypeAdapter(OffsetDateTime.class, offsetDateTimeTypeAdapter)
-            .registerTypeAdapter(LocalDate.class, localDateTypeAdapter)
-            .registerTypeAdapter(byte[].class, byteArrayAdapter)
-            .create();
+                .registerTypeAdapter(Date.class, dateTypeAdapter)
+                .registerTypeAdapter(java.sql.Date.class, sqlDateTypeAdapter)
+                .registerTypeAdapter(OffsetDateTime.class, offsetDateTimeTypeAdapter)
+                .registerTypeAdapter(LocalDate.class, localDateTypeAdapter)
+                .registerTypeAdapter(byte[].class, byteArrayAdapter)
+                .registerTypeAdapterFactory(new ConditionalNullable.ConditionalNullablePojoAdapterFactory())
+                .create();
     }
 
     /**
@@ -103,7 +106,8 @@ public JSON setGson(Gson gson) {
      *
      * @param lenientOnJson Set it to true to ignore some syntax errors
      * @return JSON
-     * @see <a href="https://www.javadoc.io/doc/com.google.code.gson/gson/2.8.5/com/google/gson/stream/JsonReader.html">https://www.javadoc.io/doc/com.google.code.gson/gson/2.8.5/com/google/gson/stream/JsonReader.html</a>
+     * @see <a href=
+     *      "https://www.javadoc.io/doc/com.google.code.gson/gson/2.8.5/com/google/gson/stream/JsonReader.html">https://www.javadoc.io/doc/com.google.code.gson/gson/2.8.5/com/google/gson/stream/JsonReader.html</a>
      */
     public JSON setLenientOnJson(boolean lenientOnJson) {
         isLenientOnJson = lenientOnJson;
@@ -133,7 +137,8 @@ public <T> T deserialize(String body, Type returnType) {
         try {
             if (isLenientOnJson) {
                 JsonReader jsonReader = new JsonReader(new StringReader(body));
-                // see https://www.javadoc.io/doc/com.google.code.gson/gson/2.8.5/com/google/gson/stream/JsonReader.html
+                // see
+                // https://www.javadoc.io/doc/com.google.code.gson/gson/2.8.5/com/google/gson/stream/JsonReader.html
                 jsonReader.setLenient(true);
                 return gson.fromJson(jsonReader, returnType);
             } else {
@@ -215,7 +220,7 @@ public OffsetDateTime read(JsonReader in) throws IOException {
                 default:
                     String date = in.nextString();
                     if (date.endsWith("+0000")) {
-                        date = date.substring(0, date.length()-5) + "Z";
+                        date = date.substring(0, date.length() - 5) + "Z";
                     }
                     return OffsetDateTime.parse(date, formatter);
             }
@@ -282,7 +287,8 @@ public static class SqlDateTypeAdapter extends TypeAdapter<java.sql.Date> {
 
         private DateFormat dateFormat;
 
-        public SqlDateTypeAdapter() {}
+        public SqlDateTypeAdapter() {
+        }
 
         public SqlDateTypeAdapter(DateFormat dateFormat) {
             this.dateFormat = dateFormat;
@@ -335,7 +341,8 @@ public static class DateTypeAdapter extends TypeAdapter<Date> {
 
         private DateFormat dateFormat;
 
-        public DateTypeAdapter() {}
+        public DateTypeAdapter() {
+        }
 
         public DateTypeAdapter(DateFormat dateFormat) {
             this.dateFormat = dateFormat;
diff --git a/src/main/java/com/smartx/tower/model/VmUpdateHostOptionsParams.java b/src/main/java/com/smartx/tower/model/VmUpdateHostOptionsParams.java
@@ -7,6 +7,7 @@
 import com.google.gson.annotations.SerializedName;
 import com.google.gson.stream.JsonReader;
 import com.google.gson.stream.JsonWriter;
+import com.smartx.tower.ConditionalNullable;
 import com.smartx.tower.model.VmUpdateHostOptionsParamsData;
 import com.smartx.tower.model.VmWhereInput;
 import io.swagger.annotations.ApiModel;
@@ -17,7 +18,7 @@
  * VmUpdateHostOptionsParams
  */
 @javax.annotation.Generated(value = "org.openapitools.codegen.languages.JavaSmartxClientCodegen")
-public class VmUpdateHostOptionsParams {
+public class VmUpdateHostOptionsParams extends ConditionalNullable.ConditionalNullablePojo {
   public static final String SERIALIZED_NAME_DATA = "data";
   @SerializedName(SERIALIZED_NAME_DATA)
   private VmUpdateHostOptionsParamsData data;
@@ -26,55 +27,53 @@ public class VmUpdateHostOptionsParams {
   @SerializedName(SERIALIZED_NAME_WHERE)
   private VmWhereInput where;
 
-  public VmUpdateHostOptionsParams() { 
+  public VmUpdateHostOptionsParams() {
   }
 
   public VmUpdateHostOptionsParams data(VmUpdateHostOptionsParamsData data) {
-    
+
     this.data = data;
     return this;
   }
 
-   /**
+  /**
    * Get data
+   * 
    * @return data
-  **/
+   **/
   @javax.annotation.Nonnull
   @ApiModelProperty(required = true, value = "")
 
   public VmUpdateHostOptionsParamsData getData() {
     return data;
   }
 
-
   public void setData(VmUpdateHostOptionsParamsData data) {
     this.data = data;
   }
 
-
   public VmUpdateHostOptionsParams where(VmWhereInput where) {
-    
+
     this.where = where;
     return this;
   }
 
-   /**
+  /**
    * Get where
+   * 
    * @return where
-  **/
+   **/
   @javax.annotation.Nonnull
   @ApiModelProperty(required = true, value = "")
 
   public VmWhereInput getWhere() {
     return where;
   }
 
-
   public void setWhere(VmWhereInput where) {
     this.where = where;
   }
 
-
   @Override
   public boolean equals(Object o) {
     if (this == o) {
@@ -113,6 +112,4 @@ private String toIndentedString(Object o) {
     }
     return o.toString().replace("\n", "\n    ");
   }
-
 }
-
diff --git a/src/main/java/com/smartx/tower/model/VmUpdateHostOptionsParamsData.java b/src/main/java/com/smartx/tower/model/VmUpdateHostOptionsParamsData.java
