com.landawn.abacus.type.EnumType Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of abacus-common Show documentation
Show all versions of abacus-common Show documentation
A general programming library in Java/Android. It's easy to learn and simple to use with concise and powerful APIs.
/*
* Copyright (C) 2015 HaiYang Li
*
* 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.landawn.abacus.type;
import java.io.IOException;
import java.lang.reflect.Field;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.EnumMap;
import java.util.HashMap;
import java.util.Map;
import com.landawn.abacus.annotation.JsonXmlField;
import com.landawn.abacus.annotation.Type.EnumBy;
import com.landawn.abacus.parser.SerializationConfig;
import com.landawn.abacus.util.BiMap;
import com.landawn.abacus.util.CharacterWriter;
import com.landawn.abacus.util.ClassUtil;
import com.landawn.abacus.util.N;
import com.landawn.abacus.util.Numbers;
import com.landawn.abacus.util.Strings;
/**
*
* @author Haiyang Li
* @param
* @since 0.8
*/
@SuppressWarnings("java:S2160")
public final class EnumType> extends SingleValueType {
public static final String ENUM = Enum.class.getSimpleName();
private static final String NULL = "null";
private final BiMap numberEnum = new BiMap<>();
private final Map enumJsonXmlNameMap;
private final Map jsonXmlNameEnumMap;
private final EnumBy enumBy;
private boolean hasNull = false;
EnumType(final String enumClassName) {
this(enumClassName, false);
}
EnumType(final String clsName, final boolean ordinal) {
super(ordinal ? clsName + "(true)" : clsName, (Class) getEnumClass(ClassUtil.forClass(clsName)));
enumJsonXmlNameMap = new EnumMap<>(typeClass);
jsonXmlNameEnumMap = new HashMap<>();
for (T enumConstant : typeClass.getEnumConstants()) {
numberEnum.put(enumConstant.ordinal(), enumConstant);
String jsonXmlName = getJsonXmlName(enumConstant);
enumJsonXmlNameMap.put(enumConstant, jsonXmlName);
jsonXmlNameEnumMap.put(jsonXmlName, enumConstant);
}
try {
hasNull = Enum.valueOf(typeClass, NULL) != null;
} catch (Exception e) {
// ignore;
}
enumBy = ordinal ? EnumBy.ORDINAL : EnumBy.NAME;
}
/**
*
*
* @return
*/
public EnumBy enumerated() {
return enumBy;
}
/**
*
*
* @return
*/
@Override
public boolean isSerializable() {
return true;
}
/**
*
*
* @return
*/
@Override
public boolean isImmutable() {
return true;
}
/**
*
*
* @param x
* @return
*/
@Override
public String stringOf(T x) {
return (jsonValueType == null) ? (x == null ? null : x.name()) : super.stringOf(x);
}
/**
*
*
* @param str
* @return
*/
@Override
public T valueOf(String str) {
if (jsonValueType == null) {
if (N.isNullOrEmpty(str) || (!hasNull && NULL.equals(str))) {
return null;
}
if (Strings.isAsciiDigtalInteger(str)) {
return valueOf(Numbers.toInt(str));
} else {
final T val = jsonXmlNameEnumMap.get(str);
if (val != null) {
return val;
} else {
return Enum.valueOf(typeClass, str);
}
}
} else {
return super.valueOf(str);
}
}
/**
*
*
* @param value
* @return
*/
public T valueOf(final int value) {
T result = numberEnum.get(value);
if ((result == null) && (value != 0)) {
throw new IllegalArgumentException("No " + typeClass.getName() + " for int value: " + value);
}
return result;
}
/**
*
*
* @param rs
* @param columnIndex
* @return
* @throws SQLException
*/
@Override
public T get(final ResultSet rs, final int columnIndex) throws SQLException {
if (jsonValueType == null) {
if (enumBy == EnumBy.ORDINAL) {
return valueOf(rs.getInt(columnIndex));
} else {
return valueOf(rs.getString(columnIndex));
}
} else {
return super.get(rs, columnIndex);
}
}
/**
*
*
* @param rs
* @param columnLabel
* @return
* @throws SQLException
*/
@Override
public T get(final ResultSet rs, final String columnLabel) throws SQLException {
if (jsonValueType == null) {
if (enumBy == EnumBy.ORDINAL) {
return valueOf(rs.getInt(columnLabel));
} else {
return valueOf(rs.getString(columnLabel));
}
} else {
return super.get(rs, columnLabel);
}
}
/**
*
*
* @param stmt
* @param columnIndex
* @param x
* @throws SQLException
*/
@Override
public void set(final PreparedStatement stmt, final int columnIndex, final T x) throws SQLException {
if (jsonValueType == null) {
if (enumBy == EnumBy.ORDINAL) {
stmt.setInt(columnIndex, (x == null) ? 0 : numberEnum.getByValue(x).intValue());
} else {
stmt.setString(columnIndex, (x == null) ? null : x.name());
}
} else {
super.set(stmt, columnIndex, x);
}
}
/**
*
*
* @param stmt
* @param parameterName
* @param x
* @throws SQLException
*/
@Override
public void set(final CallableStatement stmt, final String parameterName, final T x) throws SQLException {
if (jsonValueType == null) {
if (enumBy == EnumBy.ORDINAL) {
stmt.setInt(parameterName, (x == null) ? 0 : numberEnum.getByValue(x).intValue());
} else {
stmt.setString(parameterName, (x == null) ? null : x.name());
}
} else {
super.set(stmt, parameterName, x);
}
}
/**
*
*
* @param writer
* @param x
* @param config
* @throws IOException
*/
@SuppressWarnings("rawtypes")
@Override
public void writeCharacter(final CharacterWriter writer, final T x, final SerializationConfig config) throws IOException {
if (x == null) {
writer.write(NULL_CHAR_ARRAY);
} else {
if (jsonValueType == null) {
if (enumBy == EnumBy.ORDINAL) {
writer.writeInt(((Enum) x).ordinal());
} else {
final char ch = config == null ? 0 : config.getStringQuotation();
if (ch == 0) {
writer.writeCharacter(enumJsonXmlNameMap.get(x));
} else {
writer.write(ch);
writer.writeCharacter(enumJsonXmlNameMap.get(x));
writer.write(ch);
}
}
} else {
super.writeCharacter(writer, x, config);
}
}
}
private String getJsonXmlName(final T enumConstant) {
try {
final Field field = enumConstant.getClass().getField(((Enum) enumConstant).name());
if (field.isAnnotationPresent(JsonXmlField.class) && N.notNullOrEmpty(field.getAnnotation(JsonXmlField.class).name())) {
return field.getAnnotation(JsonXmlField.class).name();
}
try { //NOSONAR
if (field.isAnnotationPresent(com.alibaba.fastjson.annotation.JSONField.class)
&& N.notNullOrEmpty(field.getAnnotation(com.alibaba.fastjson.annotation.JSONField.class).name())) {
return field.getAnnotation(com.alibaba.fastjson.annotation.JSONField.class).name();
}
} catch (Throwable e) {
// ignore
}
try { //NOSONAR
if (field.isAnnotationPresent(com.alibaba.fastjson2.annotation.JSONField.class)
&& N.notNullOrEmpty(field.getAnnotation(com.alibaba.fastjson2.annotation.JSONField.class).name())) {
return field.getAnnotation(com.alibaba.fastjson2.annotation.JSONField.class).name();
}
} catch (Throwable e) {
// ignore
}
try { //NOSONAR
if (field.isAnnotationPresent(com.fasterxml.jackson.annotation.JsonProperty.class)
&& N.notNullOrEmpty(field.getAnnotation(com.fasterxml.jackson.annotation.JsonProperty.class).value())) {
return field.getAnnotation(com.fasterxml.jackson.annotation.JsonProperty.class).value();
}
} catch (Throwable e) {
// ignore
}
} catch (NoSuchFieldException | SecurityException e) {
// should never happen.
}
return enumConstant.name();
}
private static Class getEnumClass(final Class cls) {
return cls.isEnum() ? cls : cls.getEnclosingClass();
}
}