feat[protocol]: scala support

protocol/src/main/java/com/zfoo/protocol/serializer/CodeLanguage.java

@@ -22,6 +22,7 @@
 import com.zfoo.protocol.serializer.kotlin.CodeGenerateKotlin;
 import com.zfoo.protocol.serializer.lua.CodeGenerateLua;
 import com.zfoo.protocol.serializer.python.CodeGeneratePython;
+import com.zfoo.protocol.serializer.scala.CodeGenerateScala;
 import com.zfoo.protocol.serializer.typescript.CodeGenerateTypeScript;
 
 /**
@@ -34,11 +35,11 @@ public enum CodeLanguage {
      */
     Enhance(1, null),
 
-    Java(1<<1, CodeGenerateJava.class),
+    Java(1 << 1, CodeGenerateJava.class),
 
-    Kotlin(1<<2, CodeGenerateKotlin.class),
+    Kotlin(1 << 2, CodeGenerateKotlin.class),
 
-    Scala(1<<3, null),
+    Scala(1 << 3, CodeGenerateScala.class),
 
     Cpp(1 << 7, CodeGenerateCpp.class),
 

protocol/src/main/java/com/zfoo/protocol/serializer/scala/IScalaSerializer.java

@@ -0,0 +1,34 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.serializer.scala;
+
+import com.zfoo.protocol.model.Pair;
+import com.zfoo.protocol.registration.field.IFieldRegistration;
+
+import java.lang.reflect.Field;
+
+/**
+ * @author godotg
+ */
+public interface IScalaSerializer {
+    /**
+     * 获取属性的类型，默认值
+     */
+    Pair<String, String> field(Field field, IFieldRegistration fieldRegistration);
+
+    void writeObject(StringBuilder builder, String objectStr, int deep, Field field, IFieldRegistration fieldRegistration);
+
+    String readObject(StringBuilder builder, int deep, Field field, IFieldRegistration fieldRegistration);
+
+}

protocol/src/main/java/com/zfoo/protocol/serializer/scala/ScalaArraySerializer.java

@@ -0,0 +1,111 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.serializer.scala;
+
+import com.zfoo.protocol.generate.GenerateProtocolFile;
+import com.zfoo.protocol.model.Pair;
+import com.zfoo.protocol.registration.field.ArrayField;
+import com.zfoo.protocol.registration.field.IFieldRegistration;
+import com.zfoo.protocol.serializer.CodeLanguage;
+import com.zfoo.protocol.serializer.CutDownArraySerializer;
+import com.zfoo.protocol.util.StringUtils;
+
+import java.lang.reflect.Field;
+
+import static com.zfoo.protocol.util.FileUtils.LS;
+
+/**
+ * @author godotg
+ */
+public class ScalaArraySerializer implements IScalaSerializer {
+
+    @Override
+    public Pair<String, String> field(Field field, IFieldRegistration fieldRegistration) {
+        var type = StringUtils.format("Array[{}]", CodeGenerateScala.toScalaClassName(field.getType().getComponentType().getSimpleName()));
+        return new Pair<>(type, "_");
+    }
+
+    @Override
+    public void writeObject(StringBuilder builder, String objectStr, int deep, Field field, IFieldRegistration fieldRegistration) {
+        GenerateProtocolFile.addTab(builder, deep);
+        if (CutDownArraySerializer.getInstance().writeObject(builder, objectStr, field, fieldRegistration, CodeLanguage.Scala)) {
+            return;
+        }
+
+        ArrayField arrayField = (ArrayField) fieldRegistration;
+
+        builder.append(StringUtils.format("buffer.writeInt({}.length)", objectStr)).append(LS);
+        GenerateProtocolFile.addTab(builder, deep);
+        String length = "length" + GenerateProtocolFile.localVariableId++;
+        builder.append(StringUtils.format("val {} = {}.length", length, objectStr)).append(LS);
+
+        String i = "i" + GenerateProtocolFile.localVariableId++;
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("for ({} <- 0 until {}) {", i, length)).append(LS);
+        GenerateProtocolFile.addTab(builder, deep + 1);
+        String element = "element" + GenerateProtocolFile.localVariableId++;
+        builder.append(StringUtils.format("val {} = {}({})", element, objectStr, i)).append(LS);
+
+        CodeGenerateScala.scalaSerializer(arrayField.getArrayElementRegistration().serializer())
+                .writeObject(builder, element, deep + 1, field, arrayField.getArrayElementRegistration());
+
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append("}").append(LS);
+    }
+
+    @Override
+    public String readObject(StringBuilder builder, int deep, Field field, IFieldRegistration fieldRegistration) {
+        GenerateProtocolFile.addTab(builder, deep);
+        var cutDown = CutDownArraySerializer.getInstance().readObject(builder, field, fieldRegistration, CodeLanguage.Scala);
+        if (cutDown != null) {
+            return cutDown;
+        }
+
+
+        var arrayField = (ArrayField) fieldRegistration;
+        var result = "result" + GenerateProtocolFile.localVariableId++;
+
+        var typeName = CodeGenerateScala.toScalaClassName(arrayField.getType().getSimpleName());
+
+        var i = "index" + GenerateProtocolFile.localVariableId++;
+        var init = "init" + GenerateProtocolFile.localVariableId++;
+        var size = "size" + GenerateProtocolFile.localVariableId++;
+        builder.append(StringUtils.format("val {} = buffer.readInt", size)).append(LS);
+
+        GenerateProtocolFile.addTab(builder, deep);
+        var pair = CodeGenerateScala.scalaSerializer(arrayField.getArrayElementRegistration().serializer()).field(field, arrayField.getArrayElementRegistration());
+        var defaultValue = pair.getValue();
+        if (defaultValue.equals("null")) {
+            defaultValue = StringUtils.format("{}()", typeName);
+        }
+        builder.append(StringUtils.format("val {} = new mutable.ArrayBuffer[{}]()", result, typeName)).append(LS);
+
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("if ({} > 0) {", size)).append(LS);
+
+        GenerateProtocolFile.addTab(builder, deep + 1);
+        builder.append(StringUtils.format("for ({} <- 0 until {}) {", i, size)).append(LS);
+        var readObject = CodeGenerateScala.scalaSerializer(arrayField.getArrayElementRegistration().serializer())
+                .readObject(builder, deep + 2, field, arrayField.getArrayElementRegistration());
+        GenerateProtocolFile.addTab(builder, deep + 2);
+        builder.append(StringUtils.format("{}.addOne({})", result, readObject));
+        builder.append(LS);
+        GenerateProtocolFile.addTab(builder, deep + 1);
+        builder.append("}").append(LS);
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append("}").append(LS);
+
+        return result + ".toArray";
+    }
+}

protocol/src/main/java/com/zfoo/protocol/serializer/scala/ScalaBooleanSerializer.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.serializer.scala;
+
+import com.zfoo.protocol.generate.GenerateProtocolFile;
+import com.zfoo.protocol.model.Pair;
+import com.zfoo.protocol.registration.field.IFieldRegistration;
+import com.zfoo.protocol.util.StringUtils;
+
+import java.lang.reflect.Field;
+
+import static com.zfoo.protocol.util.FileUtils.LS;
+
+/**
+ * @author godotg
+ */
+public class ScalaBooleanSerializer implements IScalaSerializer {
+
+    @Override
+    public Pair<String, String> field(Field field, IFieldRegistration fieldRegistration) {
+        return new Pair<>("Boolean", "false");
+    }
+
+    @Override
+    public void writeObject(StringBuilder builder, String objectStr, int deep, Field field, IFieldRegistration fieldRegistration) {
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("buffer.writeBool({})", objectStr)).append(LS);
+    }
+
+    @Override
+    public String readObject(StringBuilder builder, int deep, Field field, IFieldRegistration fieldRegistration) {
+        String result = "result" + GenerateProtocolFile.localVariableId++;
+
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("val {} = buffer.readBool", result)).append(LS);
+        return result;
+    }
+}

protocol/src/main/java/com/zfoo/protocol/serializer/scala/ScalaByteSerializer.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.serializer.scala;
+
+import com.zfoo.protocol.generate.GenerateProtocolFile;
+import com.zfoo.protocol.model.Pair;
+import com.zfoo.protocol.registration.field.IFieldRegistration;
+import com.zfoo.protocol.util.StringUtils;
+
+import java.lang.reflect.Field;
+
+import static com.zfoo.protocol.util.FileUtils.LS;
+
+/**
+ * @author godotg
+ */
+public class ScalaByteSerializer implements IScalaSerializer {
+
+    @Override
+    public Pair<String, String> field(Field field, IFieldRegistration fieldRegistration) {
+        return new Pair<>("Byte", "0");
+    }
+
+    @Override
+    public void writeObject(StringBuilder builder, String objectStr, int deep, Field field, IFieldRegistration fieldRegistration) {
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("buffer.writeByte({})", objectStr)).append(LS);
+    }
+
+    @Override
+    public String readObject(StringBuilder builder, int deep, Field field, IFieldRegistration fieldRegistration) {
+        String result = "result" + GenerateProtocolFile.localVariableId++;
+
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("val {} = buffer.readByte", result)).append(LS);
+        return result;
+    }
+
+}

protocol/src/main/java/com/zfoo/protocol/serializer/scala/ScalaDoubleSerializer.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.serializer.scala;
+
+import com.zfoo.protocol.generate.GenerateProtocolFile;
+import com.zfoo.protocol.model.Pair;
+import com.zfoo.protocol.registration.field.IFieldRegistration;
+import com.zfoo.protocol.util.StringUtils;
+
+import java.lang.reflect.Field;
+
+import static com.zfoo.protocol.util.FileUtils.LS;
+
+/**
+ * @author godotg
+ */
+public class ScalaDoubleSerializer implements IScalaSerializer {
+
+    @Override
+    public Pair<String, String> field(Field field, IFieldRegistration fieldRegistration) {
+        return new Pair<>("Double", "0D");
+    }
+
+    @Override
+    public void writeObject(StringBuilder builder, String objectStr, int deep, Field field, IFieldRegistration fieldRegistration) {
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("buffer.writeDouble({})", objectStr)).append(LS);
+    }
+
+    @Override
+    public String readObject(StringBuilder builder, int deep, Field field, IFieldRegistration fieldRegistration) {
+        String result = "result" + GenerateProtocolFile.localVariableId++;
+
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("val {} = buffer.readDouble", result)).append(LS);
+        return result;
+    }
+
+}

protocol/src/main/java/com/zfoo/protocol/serializer/scala/ScalaFloatSerializer.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright (C) 2020 The zfoo Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and limitations under the License.
+ */
+
+package com.zfoo.protocol.serializer.scala;
+
+import com.zfoo.protocol.generate.GenerateProtocolFile;
+import com.zfoo.protocol.model.Pair;
+import com.zfoo.protocol.registration.field.IFieldRegistration;
+import com.zfoo.protocol.util.StringUtils;
+
+import java.lang.reflect.Field;
+
+import static com.zfoo.protocol.util.FileUtils.LS;
+
+/**
+ * @author godotg
+ */
+public class ScalaFloatSerializer implements IScalaSerializer {
+
+    @Override
+    public Pair<String, String> field(Field field, IFieldRegistration fieldRegistration) {
+        return new Pair<>("Float", "0f");
+    }
+
+    @Override
+    public void writeObject(StringBuilder builder, String objectStr, int deep, Field field, IFieldRegistration fieldRegistration) {
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("buffer.writeFloat({})", objectStr)).append(LS);
+    }
+
+    @Override
+    public String readObject(StringBuilder builder, int deep, Field field, IFieldRegistration fieldRegistration) {
+        String result = "result" + GenerateProtocolFile.localVariableId++;
+
+        GenerateProtocolFile.addTab(builder, deep);
+        builder.append(StringUtils.format("val {} = buffer.readFloat", result)).append(LS);
+        return result;
+    }
+
+}