Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.hazelcast.client.protocol.generator.CodeGenerationUtils Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2018, Hazelcast, Inc. All Rights Reserved.
*
* 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.hazelcast.client.protocol.generator;
import javax.lang.model.element.TypeElement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
@SuppressWarnings("checkstyle:methodcount")
public final class CodeGenerationUtils {
/**
* Package prefix for the codec module.
*/
public static final String CODEC_PACKAGE = "com.hazelcast.client.impl.protocol.codec.";
/**
* Fully qualified classname for Hazelcast Data class.
*/
public static final String DATA_FULL_NAME = "com.hazelcast.nio.serialization.Data";
/**
* The multiplier is used to convert a version string to a comparable integer number
*/
public static final int MAJOR_VERSION_MULTIPLIER = 1000;
private static final int BYTE_BIT_COUNT = 8;
@SuppressWarnings("checkstyle:whitespacearound")
private static final Map JAVA_TO_PYTHON_TYPES = new HashMap() {
{
put(DATA_FULL_NAME, "Data");
put("java.lang.String", "str");
put("java.lang.Integer", "int");
put("boolean", "bool");
put("java.util.List", "list");
put("java.util.Set", "set");
put("java.util.Collection", "collection");
put("java.util.Map", "dictionary");
put("java.util.Map.Entry", "tuple");
put("com.hazelcast.nio.Address", "Address");
put("com.hazelcast.client.impl.client.DistributedObjectInfo", "DistributedObjectInfo");
put("com.hazelcast.core.Member", "Member");
put("com.hazelcast.cluster.client.MemberAttributeChange", "MemberAttributeChange");
put("com.hazelcast.map.impl.SimpleEntryView", "SimpleEntryView");
}
};
@SuppressWarnings({"checkstyle:executablestatementcount", "checkstyle:whitespacearound"})
private static final Map JAVA_TO_GO_TYPES = new HashMap() {
{
put(DATA_FULL_NAME, "Data");
put("java.lang.String", "string");
put("byte", "uint8");
put("java.lang.Integer", "int32");
put("int", "int32");
put("boolean", "bool");
put("java.util.List", "[]");
put("java.util.Map", "map");
put("java.util.Map.Entry", "Pair");
put("java.lang.Long", "int64");
put("java.util.UUID", "Uuid");
put("long", "int64");
put("com.hazelcast.nio.Address", "Address");
put("com.hazelcast.client.impl.client.DistributedObjectInfo", "DistributedObjectInfo");
put("com.hazelcast.core.Member", "Member");
put("com.hazelcast.cluster.client.MemberAttributeChange", "MemberAttributeChange");
put("com.hazelcast.map.impl.SimpleEntryView", "DataEntryView");
put("byte[]", "[]byte");
put("long[]", "[]int64");
}
};
private static final List GO_POINTER_TYPES = Arrays
.asList("uint8", "int32", "int32", "bool", "[]", "map", "int64", "int64", "[]byte", "[]int64");
private static final Map JAVA_TO_NODE_TYPES = new HashMap() {
{
put(DATA_FULL_NAME, "data");
put("java.lang.String", "string");
put("java.lang.Integer", "int32");
put("boolean", "boolean");
put("int", "int32");
put("com.hazelcast.nio.Address", "Address");
put("java.util.List", "list");
put("java.util.Set", "set");
}
};
private static final Map JAVA_TO_TS_TYPES = new HashMap() {
{
put(DATA_FULL_NAME, "Data");
put("java.lang.String", "string");
put("java.lang.Integer", "number");
put("boolean", "boolean");
put("int", "number");
put("com.hazelcast.nio.Address", "Address");
put("com.hazelcast.core.Member", "Member");
put("com.hazelcast.client.impl.client.DistributedObjectInfo", "any");
put("java.util.List", "Array");
put("java.util.Collection", "Array");
put("java.util.Set", "Array");
put("long", "any");
put("java.lang.Long", "any");
put("java.util.Map.Entry", "");
put("java.util.Map", "Array");
put("byte", "number");
put("java.util.UUID", "UUID");
put("long[]", "any[]");
}
};
@SuppressWarnings("checkstyle:whitespacearound")
private static final Map JAVA_TO_CSHARP_TYPES = new HashMap() {
{
put(DATA_FULL_NAME, "IData");
put("java.lang.String", "string");
put("java.lang.Integer", "int");
put("java.lang.Long", "long");
put("boolean", "bool");
put("java.util.UUID", "Guid");
put("java.util.List", "IList");
put("java.util.Set", "ISet");
put("java.util.Collection", "ICollection");
put("java.util.Map", "IDictionary");
put("java.util.Map.Entry", "KeyValuePair");
put("com.hazelcast.nio.Address", "Address");
put("com.hazelcast.client.impl.client.DistributedObjectInfo", "DistributedObjectInfo");
put("com.hazelcast.core.Member", "Core.IMember");
put("com.hazelcast.cluster.client.MemberAttributeChange", "Hazelcast.Client.Request.Cluster.MemberAttributeChange");
put("com.hazelcast.map.impl.SimpleEntryView", "Hazelcast.Map.SimpleEntryView");
}
};
@SuppressWarnings({"checkstyle:whitespacearound", "checkstyle:executablestatementcount"})
private static final Map JAVA_TO_CPP_TYPES = new HashMap() {
{
put("java.lang.Integer", "int32_t");
put("int", "int32_t");
put("boolean", "bool");
put("java.lang.Boolean", "bool");
put("short", "int16_t");
put("char", "int8_t");
put("byte", "uint8_t");
put("long", "int64_t");
put(DATA_FULL_NAME, "serialization::pimpl::Data");
put("java.lang.String", "std::string");
put("byte[]", "std::vector");
put("java.util.List", "std::vector");
put("java.util.Set", "std::vector");
put("java.util.Collection", "std::vector");
put("java.util.Map", "std::vector PYTHON_RESERVED_WORDS = Arrays
.asList("and", "del", "from", "not", "while", "as", "elif", "global", "or", "with", "assert", "else", "if", "pass",
"yield", "break", "except", "import", "print", "class", "exec", "in", "raise", "continue", "finally", "is",
"return", "def", "for", "lambda", "try");
private static final List NON_NULLABLE_TYPES = Arrays
.asList("int", "long", "short", "byte", "boolean", "java.util.UUID");
private static final List PRIMITIVE_TYPES = Arrays.asList("int", "long", "short", "byte", "boolean");
private CodeGenerationUtils() {
}
public static String capitalizeFirstLetter(String input) {
return input.substring(0, 1).toUpperCase() + input.substring(1);
}
public static String getPackageNameFromQualifiedName(String qualifiedClassName) {
return qualifiedClassName.substring(0, qualifiedClassName.lastIndexOf("."));
}
public static String mergeIds(short classId, short methodId) {
return Integer.toHexString((classId << BYTE_BIT_COUNT) + methodId);
}
public static String addHexPrefix(String s) {
switch (s.length()) {
case 3:
return "0x0" + s;
case 2:
return "0x00" + s;
case 1:
return "0x000" + s;
default:
return "0x" + s;
}
}
public static String getArrayType(String type) {
int end = type.indexOf("[]");
return type.substring(0, end).trim();
}
public static String getGenericType(String type) {
int beg = type.indexOf("<");
int end = type.lastIndexOf(">");
return type.substring(beg + 1, end).trim();
}
public static String getSimpleType(String type) {
int beg = type.indexOf("<");
return type.substring(0, beg);
}
public static String getFirstGenericParameterType(String type) {
int beg = type.indexOf("<");
int end = type.indexOf(",");
return beg > 0 && end > 0 ? type.substring(beg + 1, end).trim() : "";
}
public static String getSecondGenericParameterType(String type) {
int beg = type.indexOf(",");
int end = type.lastIndexOf(">");
return beg > 0 && end > 0 ? type.substring(beg + 1, end).trim() : "";
}
public static boolean isPrimitive(String type) {
return PRIMITIVE_TYPES.contains(type);
}
public static boolean isNonNullable(String type) {
return NON_NULLABLE_TYPES.contains(type);
}
public static boolean isGeneric(String type) {
return type.contains("<");
}
public static String getTypeCategory(String type) {
int endIndex = type.indexOf('<');
String t = endIndex > 0 ? type.substring(0, endIndex) : type;
if (CodecModel.CUSTOM_CODEC_MAP.containsKey(t)) {
return "CUSTOM";
} else if (t.equals("java.util.Map.Entry")) {
return "MAPENTRY";
} else if (t.equals("java.util.List") || t.equals("java.util.Set") || t.equals("java.util.Collection")) {
return "COLLECTION";
} else if (type.endsWith("[]")) {
return "ARRAY";
}
return "OTHER";
}
public static String getTypeCodec(String type) {
int endIndex = type.indexOf('<');
String t = endIndex > 0 ? type.substring(0, endIndex) : type;
TypeElement typeElement = CodecModel.CUSTOM_CODEC_MAP.get(t);
return typeElement != null ? typeElement.toString() : "";
}
public static String getConvertedType(String type) {
if (type.startsWith("java.util.List<") || type.startsWith("java.util.Set<") || type.startsWith("java.util.Collection<")) {
return type.replaceAll("java.util.*<(.*)>", "java.util.Collection<$1>");
}
return type;
}
public static String modifyForGoTypes(String type) {
if (type.equals("String")) {
type = "string";
} else if (type.startsWith("Long") || type.startsWith("long")) {
type = "int64";
} else if (type.startsWith("Int") || type.startsWith("int")) {
type = "int32";
} else if (type.startsWith("UUID")) {
type = "Uuid";
}
return type;
}
public static String getDescription(String parameterName, String commentString) {
String result = "";
if (parameterName == null || commentString == null) {
return result;
}
int start = commentString.indexOf("@param");
if (start == -1) {
return result;
}
String paramString = commentString.substring(start);
String[] paramStrings = paramString.split("@param");
for (String parameterString : paramStrings) {
/**
* Example such string is
* key key of the entry
*/
String trimmedParameterString = parameterString.trim();
if (trimmedParameterString.length() > parameterName.length() && trimmedParameterString.startsWith(parameterName)) {
int returnIndex = trimmedParameterString.indexOf("@return");
if (returnIndex == -1) {
result = trimmedParameterString.substring(parameterName.length(), trimmedParameterString.length());
} else {
result = trimmedParameterString.substring(parameterName.length(), returnIndex);
}
// replace any new line with getGenericTypeParameters(String parameters) {
List paramList = new ArrayList();
int balanced = 0;
StringBuilder current = new StringBuilder();
for (int i = 0; i < parameters.length(); i++) {
char c = parameters.charAt(i);
if (balanced == 0 && c == ',') {
paramList.add(current.toString().trim());
current = new StringBuilder();
continue;
} else if (c == '<') {
balanced++;
} else if (c == '>') {
balanced--;
}
current.append(c);
}
paramList.add(current.toString());
return paramList;
}
public static String getGoPointerType(String type) {
String goType = getGoType(type);
if (!GO_POINTER_TYPES.contains(goType)) {
if (goType.contains("[]")) {
return "[]*" + goType.substring(2);
}
return "*" + goType;
}
return goType;
}
public static String getPythonType(String type) {
return getLanguageType(Lang.PY, type, JAVA_TO_PYTHON_TYPES);
}
public static String getGoType(String type) {
return getLanguageType(Lang.GO, type, JAVA_TO_GO_TYPES);
}
public static String getCSharpType(String type) {
return getLanguageType(Lang.CS, type, JAVA_TO_CSHARP_TYPES);
}
public static String getCppType(String type) {
return getLanguageType(Lang.CPP, type, JAVA_TO_CPP_TYPES);
}
public static String getTsType(String type) {
return getLanguageType(Lang.NODE, type, JAVA_TO_TS_TYPES);
}
@SuppressWarnings({"checkstyle:npathcomplexity", "checkstyle:whitespacearound", "checkstyle:cyclomaticcomplexity"})
public static String getLanguageType(Lang language, String type, Map languageMapping) {
type = type.trim();
if (isGeneric(type)) {
String simpleType = getLanguageType(language, getSimpleType(type), languageMapping);
String genericParameters = getGenericType(type);
List typeParameters = getGenericTypeParameters(genericParameters);
StringBuilder builder = new StringBuilder();
if (language == Lang.NODE && type.startsWith("java.util.Map.Entry<")) {
builder.append('[');
} else if (language != Lang.GO) {
builder.append(simpleType).append('<');
} else {
builder.append(simpleType);
}
if ((simpleType.equals("Pair") || simpleType.equals("DataEntryView")) && language == Lang.GO) {
return builder.toString();
}
Iterator iterator = typeParameters.iterator();
while (iterator.hasNext()) {
builder.append(getLanguageType(language, iterator.next(), languageMapping));
if (iterator.hasNext()) {
builder.append(", ");
}
}
if (language == Lang.CPP && type.startsWith("java.util.Map<")) {
builder.append(" > >");
} else if (language == Lang.NODE && type.startsWith("java.util.Map.Entry<")) {
builder.append("]");
} else if (language != Lang.GO) {
builder.append(">");
}
String result = builder.toString();
if (result.equals("EntryView")) {
result = "map::DataEntryView";
}
return result;
}
String convertedType = languageMapping.get(type);
return convertedType == null ? type : convertedType;
}
public static boolean shouldGenerateForCpp(String codecName) {
return !(codecName.equals("MapReduce") || codecName.equals("Cache") || codecName.equals("Ringbuffer") || codecName
.equals("EnterpriseMap") || codecName.equals("XATransaction"));
}
public static String convertToSnakeCase(String camelCase) {
return camelCase.replaceAll("(.)(\\p{Upper})", "$1_$2").toLowerCase();
}
public static String convertSpecialNodeName(String name) {
if (name.equals("function")) {
return "arr";
}
if (name.equals("arguments")) {
return "args";
}
return name;
}
public static String escape(String str, Lang lang) {
switch (lang) {
case PY:
return PYTHON_RESERVED_WORDS.contains(str) ? str + "_" : str;
default:
//TODO add other lang reserved words
return str;
}
}
/**
* @param messageSince The version of the message. E.g. 1.2
* @return Int representation of the version. Calculated as: minor + (major * 1000).
* E.g. 1002 for "1.2"
*/
public static int versionAsInt(String messageSince) {
String[] versions = messageSince.split("\\.");
return Integer.parseInt(versions[1]) + Integer.parseInt(versions[0]) * MAJOR_VERSION_MULTIPLIER;
}
/**
* @param version protocol version as int. E.g. 1002
* @return The protocol version as string, 1.2 for 1002
*/
public static String versionAsString(int version) {
return String.format("%d.%d", version / MAJOR_VERSION_MULTIPLIER, version % MAJOR_VERSION_MULTIPLIER);
}
/**
* @param version protocol version as int. E.g. 1002
* @return The protocol version as string, 1_2 for 1002
*/
public static String versionAsClassName(int version) {
return String.format("%d_%d", version / MAJOR_VERSION_MULTIPLIER, version % MAJOR_VERSION_MULTIPLIER);
}
}
