tiennm99/zfoo
1
0
Fork 0
mirror of https://github.com/tiennm99/zfoo.git synced 2026-09-02 14:21:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf[protobuf]: list box type of protobuf

Browse Source
This commit is contained in:
godotg
2023-12-03 16:03:56 +08:00
parent d5c4badff6
commit e76dd3ab81
6 changed files with 161 additions and 192 deletions
Download Patch File Download Diff File Expand all files Collapse all files
protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java
+153 -3
View File
@@ -14,8 +14,10 @@
package com.zfoo.protocol.serializer.protobuf;
import com.zfoo.protocol.collection.CollectionUtils;
import com.zfoo.protocol.serializer.protobuf.builder.JavaBuilder;
import com.zfoo.protocol.serializer.protobuf.parser.TypeProtobuf;
import com.zfoo.protocol.serializer.protobuf.wire.MapField;
import com.zfoo.protocol.serializer.protobuf.wire.Option;
import com.zfoo.protocol.serializer.protobuf.wire.PbField;
import com.zfoo.protocol.serializer.protobuf.wire.ProtoMessage;
import com.zfoo.protocol.serializer.protobuf.parser.Proto;
import com.zfoo.protocol.serializer.protobuf.parser.ProtoParser;
@@ -25,6 +27,9 @@ import com.zfoo.protocol.util.StringUtils;
import java.io.File;
import java.util.*;
import static com.zfoo.protocol.util.FileUtils.LS;
import static com.zfoo.protocol.util.StringUtils.TAB;
public class GeneratePbUtils {
public static void create(PbBuildOption buildOption) {
@@ -88,7 +93,6 @@ public class GeneratePbUtils {
private static void generateDtoMessage(PbBuildOption buildOption, Proto proto, Map<String, Proto> protos) {
JavaBuilder builder = new JavaBuilder();
String msgPath = buildOption.getOutputPath() + File.separator;
Map<String, String> msgComments = new HashMap<>();
@@ -100,10 +104,156 @@ public class GeneratePbUtils {
msg.getComments().forEach(c -> mc.append(c));
}
msgComments.put(msg.getName(), mc.toString());
var code = builder.buildMessage(proto, msg, 1, null, protos);
var code = buildMessage(proto, msg, 1, null, protos);
var filePath = msgPath + File.separator + msg.getName() + ".java";
FileUtils.writeStringToFile(new File(filePath), code, false);
}
}
// -------------------------------------------------------------------------------------------------------------
public static String getJavaType(PbField field) {
String type = field.getType();
if (field instanceof MapField) {
var mapField = (MapField) field;
type = StringUtils.format("Map<{}, {}>", getBoxJavaType(mapField.getKey().value()), getBoxJavaType(mapField.getValue()));
return type;
}
return getJavaType(type);
}
public static String getJavaType(String type) {
var typeProtobuf = TypeProtobuf.typeOfProtobuf(type);
if (typeProtobuf == null) {
return type;
}
var javaType = typeProtobuf.javaType();
return javaType.getTypeString();
}
private static String getBoxJavaType(PbField field) {
return getBoxJavaType(field.getType());
}
private static String getBoxJavaType(String type) {
var typeProtobuf = TypeProtobuf.typeOfProtobuf(type);
if (typeProtobuf == null) {
return type;
}
var javaType = typeProtobuf.javaType();
return javaType.getBoxedType();
}
private static void addImport(List<String> imps, String imp) {
if (imps == null) {
return;
}
if (!imps.contains(imp)) {
imps.add(imp);
}
}
private static void buildMsgImps(ProtoMessage msg, List<PbField> tmp, List<String> imps) {
var fields = msg.getFields();
if (CollectionUtils.isNotEmpty(fields)) {
for (var field : fields) {
getJavaType(field);
tmp.add(field);
}
}
for (int i = 0; i < tmp.size(); i++) {
if (tmp.get(i) instanceof MapField) {
addImport(imps, Map.class.getName());
} else if (tmp.get(i).getCardinality() == PbField.Cardinality.REPEATED) {
addImport(imps, List.class.getName());
}
}
}
private static void buildDocComment(StringBuilder builder, ProtoMessage msg) {
if (CollectionUtils.isEmpty(msg.getComments())) {
return;
}
builder.append("/**").append(LS);
msg.getComments().forEach(it -> builder.append(StringUtils.format(" * {}", it)).append(LS));
builder.append(" */").append(LS);
}
private static void buildFieldComment(StringBuilder builder, PbField pbField) {
if (CollectionUtils.isEmpty(pbField.getComments())) {
return;
}
pbField.getComments().forEach(it -> builder.append(TAB).append(StringUtils.format("// {}", it)).append(LS));
}
private static String getJavaPackage(Proto proto) {
if (CollectionUtils.isEmpty(proto.getOptions())) {
return StringUtils.EMPTY;
}
for (Option option : proto.getOptions()) {
if ("java_package".equalsIgnoreCase(option.getName())) {
return option.getValue();
}
}
return StringUtils.EMPTY;
}
public static String buildMessage(Proto proto, ProtoMessage msg, int indent, Map<String, String> defineMsgs, Map<String, Proto> protos) {
var level = Math.max(indent, 1);
var tmp = new ArrayList<PbField>();
var imps = new ArrayList<String>();
var builder = new StringBuilder();
buildMsgImps(msg, tmp, imps);
List<PbField> fields = new ArrayList<>();
tmp.stream().sorted(Comparator.comparingInt(PbField::getTag))
.forEach(fields::add);
imps.stream().sorted(Comparator.naturalOrder())
.forEach(it -> builder.append(StringUtils.format("import {};", it)).append(LS));
buildDocComment(builder, msg);
builder.append(StringUtils.format("public class {} {", msg.getName())).append(LS);
int size = fields.size();
var builderMethod = new StringBuilder();
for (int i = 0; i < size; i++) {
PbField f = fields.get(i);
buildFieldComment(builder, f);
String type = getJavaType(f);
String name = f.getName();
if (f.getCardinality() == PbField.Cardinality.REPEATED) {
String boxedTypeName = getBoxJavaType(f);
type = "List<" + boxedTypeName + ">";
}
builder.append(TAB).append(StringUtils.format("private {} {};", type, name)).append(LS);
String getMethod;
if (!"bool".equalsIgnoreCase(f.getType())) {
getMethod = StringUtils.format("get{}", StringUtils.capitalize(f.getName()));
} else {
getMethod = StringUtils.format("is{}", StringUtils.capitalize(f.getName()));
}
builderMethod.append(TAB).append(StringUtils.format("public {} {}() {", type, getMethod)).append(LS);
builderMethod.append(TAB + TAB).append(StringUtils.format("return {};", f.getName())).append(LS);
builderMethod.append(TAB).append("}").append(LS);
String setMethod = StringUtils.format("set{}", StringUtils.capitalize(f.getName()));
builderMethod.append(TAB).append(StringUtils.format("public void {}({} {}) {", setMethod, type, f.getName())).append(LS);
builderMethod.append(TAB + TAB).append(StringUtils.format("this.{} = {};", f.getName(), f.getName())).append(LS);
builderMethod.append(TAB).append("}").append(LS);
}
builder.append(LS).append(builderMethod);
builder.append("}");
return builder.toString();
}
}
protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/builder/JavaBuilder.java
-177
View File
@@ -1,177 +0,0 @@
/*
* Copyright 2021 The edap Project
*
* 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.protobuf.builder;
import com.zfoo.protocol.collection.CollectionUtils;
import com.zfoo.protocol.serializer.protobuf.parser.Proto;
import com.zfoo.protocol.serializer.protobuf.parser.TypeJava;
import com.zfoo.protocol.serializer.protobuf.parser.TypeProtobuf;
import com.zfoo.protocol.serializer.protobuf.wire.MapField;
import com.zfoo.protocol.serializer.protobuf.wire.Option;
import com.zfoo.protocol.serializer.protobuf.wire.PbField;
import com.zfoo.protocol.serializer.protobuf.wire.ProtoMessage;
import com.zfoo.protocol.util.StringUtils;
import java.util.*;
import static com.zfoo.protocol.util.FileUtils.LS;
import static com.zfoo.protocol.util.StringUtils.TAB;
public class JavaBuilder {
public String getJavaType(PbField field) {
String type = field.getType();
if (field instanceof MapField) {
var mapField = (MapField) field;
type = StringUtils.format("Map<{}, {}>", getBoxJavaType(mapField.getKey().value()), getBoxJavaType(mapField.getValue()));
return type;
}
return getJavaType(type);
}
public String getJavaType(String type) {
var typeProtobuf = TypeProtobuf.typeOfProtobuf(type);
if (typeProtobuf == null) {
return type;
}
var javaType = typeProtobuf.javaType();
return javaType.getTypeString();
}
private String getBoxJavaType(PbField field) {
return getBoxJavaType(field.getType());
}
private String getBoxJavaType(String type) {
var typeProtobuf = TypeProtobuf.typeOfProtobuf(type);
if (typeProtobuf == null) {
return type;
}
var javaType = typeProtobuf.javaType();
return javaType.getBoxedType();
}
private void addImport(List<String> imps, String imp) {
if (imps == null) {
return;
}
if (!imps.contains(imp)) {
imps.add(imp);
}
}
private void buildMsgImps(ProtoMessage msg, List<PbField> tmp, List<String> imps) {
var fields = msg.getFields();
if (CollectionUtils.isNotEmpty(fields)) {
for (var field : fields) {
getJavaType(field);
tmp.add(field);
}
}
for (int i = 0; i < tmp.size(); i++) {
if (tmp.get(i) instanceof MapField) {
addImport(imps, Map.class.getName());
} else if (tmp.get(i).getCardinality() == PbField.Cardinality.REPEATED) {
addImport(imps, List.class.getName());
}
}
}
private void buildDocComment(StringBuilder builder, ProtoMessage msg) {
if (CollectionUtils.isEmpty(msg.getComments())) {
return;
}
builder.append("/**").append(LS);
msg.getComments().forEach(it -> builder.append(StringUtils.format(" * {}", it)).append(LS));
builder.append(" */").append(LS);
}
private void buildFieldComment(StringBuilder builder, PbField pbField) {
if (CollectionUtils.isEmpty(pbField.getComments())) {
return;
}
pbField.getComments().forEach(it -> builder.append(TAB).append(StringUtils.format("// {}", it)).append(LS));
}
private String getJavaPackage(Proto proto) {
if (CollectionUtils.isEmpty(proto.getOptions())) {
return StringUtils.EMPTY;
}
for (Option option : proto.getOptions()) {
if ("java_package".equalsIgnoreCase(option.getName())) {
return option.getValue();
}
}
return StringUtils.EMPTY;
}
public String buildMessage(Proto proto, ProtoMessage msg, int indent, Map<String, String> defineMsgs, Map<String, Proto> protos) {
var level = Math.max(indent, 1);
var tmp = new ArrayList<PbField>();
var imps = new ArrayList<String>();
var builder = new StringBuilder();
buildMsgImps(msg, tmp, imps);
List<PbField> fields = new ArrayList<>();
tmp.stream().sorted(Comparator.comparingInt(PbField::getTag))
.forEach(fields::add);
imps.stream().sorted(Comparator.naturalOrder())
.forEach(it -> builder.append(StringUtils.format("import {};", it)).append(LS));
buildDocComment(builder, msg);
builder.append(StringUtils.format("public class {} {", msg.getName())).append(LS);
int size = fields.size();
var builderMethod = new StringBuilder();
for (int i = 0; i < size; i++) {
PbField f = fields.get(i);
buildFieldComment(builder, f);
String type = getJavaType(f);
String name = f.getName();
if (f.getCardinality() == PbField.Cardinality.REPEATED) {
String boxedTypeName = getBoxJavaType(f);
type = "List<" + boxedTypeName + ">";
}
builder.append(TAB).append(StringUtils.format("private {} {};", type, name)).append(LS);
String getMethod;
if (!"bool".equalsIgnoreCase(f.getType())) {
getMethod = StringUtils.format("get{}", StringUtils.capitalize(f.getName()));
} else {
getMethod = StringUtils.format("is{}", StringUtils.capitalize(f.getName()));
}
builderMethod.append(TAB).append(StringUtils.format("public {} {}() {", type, getMethod)).append(LS);
builderMethod.append(TAB + TAB).append(StringUtils.format("return {};", f.getName())).append(LS);
builderMethod.append(TAB).append("}").append(LS);
String setMethod = StringUtils.format("set{}", StringUtils.capitalize(f.getName()));
builderMethod.append(TAB).append(StringUtils.format("public void {}({} {}) {", setMethod, type, f.getName())).append(LS);
builderMethod.append(TAB + TAB).append(StringUtils.format("this.{} = {};", f.getName(), f.getName())).append(LS);
builderMethod.append(TAB).append("}").append(LS);
}
builder.append(LS).append(builderMethod);
builder.append("}");
return builder.toString();
}
}
protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/parser/Proto.java
+2 -1
View File
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2020 The zfoo Authors
* Copyright 2021 The edap Project
*
* 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
*
protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/parser/ProtoParser.java
+2 -1
View File
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2020 The zfoo Authors
* Copyright 2021 The edap Project
*
* 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
*
protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/parser/TypeJava.java
+2 -6
View File
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2020 The zfoo Authors
* Copyright 2021 The edap Project
*
* 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
*
@@ -11,11 +12,6 @@
*/
package com.zfoo.protocol.serializer.protobuf.parser;
/**
* @author godotg
*/
import com.zfoo.protocol.collection.ArrayUtils;
/**
protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/parser/TypeProtobuf.java
+2 -4
View File
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2020 The zfoo Authors
* Copyright 2021 The edap Project
*
* 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
*
@@ -15,9 +16,6 @@ package com.zfoo.protocol.serializer.protobuf.parser;
import java.util.HashMap;
import java.util.Map;
/**
* @author godotg
*/
public enum TypeProtobuf {
FLOAT("float", TypeJava.FLOAT),
DOUBLE("double", TypeJava.DOUBLE),