package com.nanometer.smartlab.entity.handler;

import com.nanometer.smartlab.entity.enumtype.ApproverFlag;
import com.nanometer.smartlab.entity.enumtype.WarningLevel;
import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class WarningLevelHandler extends BaseTypeHandler<WarningLevel> {

    @Override
    public WarningLevel getNullableResult(ResultSet rs, String index) throws SQLException {
        return WarningLevel.parse(rs.getInt(index));
    }

    @Override
    public WarningLevel getNullableResult(ResultSet rs, int index) throws SQLException {
        return WarningLevel.parse(rs.getInt(index));
    }

    @Override
    public WarningLevel getNullableResult(CallableStatement cs, int index) throws SQLException {
        return WarningLevel.parse(cs.getInt(index));
    }

    @Override
    public void setNonNullParameter(PreparedStatement ps, int index, WarningLevel val, JdbcType arg3) throws SQLException {
        ps.setInt(index, val.getLevelCode());
    }
}