io.github.jinghui70.rainbow.dbaccess.enumSupport.EnumMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rainbow-dbaccess Show documentation

Rainbow DBAccess

The newest version!

package io.github.jinghui70.rainbow.dbaccess.enumSupport;

import cn.hutool.core.util.EnumUtil;
import io.github.jinghui70.rainbow.dbaccess.fieldmapper.FieldMapper;

import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Objects;

public class EnumMapper> extends FieldMapper {

    private final Class enumClass;

    private final boolean isCode;

    private final boolean isOrdinal;

    public EnumMapper(Class enumClass) {
        this.enumClass = enumClass;
        isCode = CodeEnum.class.isAssignableFrom(enumClass);
        isOrdinal = OrdinalEnum.class.isAssignableFrom(enumClass);
    }

    @Override
    public T formDB(ResultSet rs, int index) throws SQLException {
        if (isOrdinal) {
            int value = rs.getInt(index);
            if (rs.wasNull()) return null;
            return EnumUtil.getEnumAt(enumClass, value);
        }
        String value = rs.getString(index);
        if (value == null) return null;
        if (isCode) {
            for (T t : enumClass.getEnumConstants()) {
                if (Objects.equals(value, ((CodeEnum) t).code()))
                    return t;
            }
            return null;
        }
        return Enum.valueOf(enumClass, value);
    }

    @Override
    public void saveToDB(PreparedStatement ps, int paramIndex, Object value) throws SQLException {
        if (value == null)
            ps.setObject(paramIndex, value);
        else if (isOrdinal)
            ps.setInt(paramIndex, ((Enum) value).ordinal());
        else if (isCode)
            ps.setString(paramIndex, ((CodeEnum) value).code());
        else
            ps.setString(paramIndex, ((Enum) value).name());
    }
}