net.binis.codegen.hibernate.NamedCodeEnumValueConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of code-generator-hibernate Show documentation
Binis Code Generation Hibernate Extension
The newest version!
package net.binis.codegen.hibernate;

/*-
 * #%L
 * code-generator-hibernate
 * %%
 * Copyright (C) 2021 - 2024 Binis Belev
 * %%
 * 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.
 * #L%
 */

import net.binis.codegen.objects.base.enumeration.CodeEnum;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.type.descriptor.ValueBinder;
import org.hibernate.type.descriptor.ValueExtractor;
import org.hibernate.type.descriptor.java.JavaType;
import org.hibernate.type.descriptor.jdbc.JdbcType;

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.Serializable;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Locale;

/**
 * BasicValueConverter handling the conversion of an enum based on
 * JPA {@link jakarta.persistence.EnumType#STRING} strategy (storing the name)
 *
 * @author Steve Ebersole
 */
public class NamedCodeEnumValueConverter implements CodeEnumValueConverter, Serializable {
    private final CodeEnumJavaType domainTypeDescriptor;
    private final JdbcType jdbcType;
    private final JavaType relationalTypeDescriptor;

    private transient ValueExtractor valueExtractor;
    private transient ValueBinder valueBinder;

    public NamedCodeEnumValueConverter(
            CodeEnumJavaType domainTypeDescriptor,
            JdbcType jdbcType,
            JavaType relationalTypeDescriptor) {
        this.domainTypeDescriptor = domainTypeDescriptor;
        this.jdbcType = jdbcType;
        this.relationalTypeDescriptor = relationalTypeDescriptor;

        this.valueExtractor = jdbcType.getExtractor(relationalTypeDescriptor);
        this.valueBinder = jdbcType.getBinder(relationalTypeDescriptor);
    }

    @Override
    public CodeEnumJavaType getDomainJavaType() {
        return domainTypeDescriptor;
    }

    @Override
    public JavaType getRelationalJavaType() {
        return relationalTypeDescriptor;
    }

    @Override
    public E toDomainValue(String relationalForm) {
        return domainTypeDescriptor.fromName(relationalForm);
    }

    @Override
    public String toRelationalValue(E domainForm) {
        return domainTypeDescriptor.toName(domainForm);
    }

    @Override
    public int getJdbcTypeCode() {
        return jdbcType.getJdbcTypeCode();
    }

    public int getDefaultSqlTypeCode() {
        return jdbcType.getDefaultSqlTypeCode();
    }

    @Override
    public String toSqlLiteral(Object value) {
        //noinspection rawtypes
        return String.format(Locale.ROOT, "'%s'", ((CodeEnum) value).name());
    }

    private void readObject(ObjectInputStream stream) throws ClassNotFoundException, IOException {
        stream.defaultReadObject();

        this.valueExtractor = jdbcType.getExtractor(relationalTypeDescriptor);
        this.valueBinder = jdbcType.getBinder(relationalTypeDescriptor);
    }

    @Override
    public void writeValue(
            PreparedStatement statement,
            E value,
            int position,
            SharedSessionContractImplementor session) throws SQLException {
        final String jdbcValue = value == null ? null : value.name();
        valueBinder.bind(statement, jdbcValue, position, session);
    }
}