io.github.h800572003.generate.ResultSetResolver Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sql-generate Show documentation

The newest version!

package io.github.h800572003.generate;

import java.math.BigDecimal;
import java.sql.*;
import java.util.HashMap;
import java.util.Map;

/**
 * ResultSet Resolver
 */
public class ResultSetResolver {

    private Map, String> map = new HashMap<>();

    public void addRule(Class className, String ruleMethod) {
        map.put(className, ruleMethod);
    }
    public ResultSetResolver(){
        this.addRule(String.class,"getString");

        this.addRule(BigDecimal.class,"getBigDecimal");

        this.addRule(Integer.class,"getInt");
        this.addRule(int.class,"getInt");

        this.addRule(Long.class,"getLong");
        this.addRule(long.class,"getLong");

        this.addRule(Float.class,"getFloat");
        this.addRule(float.class,"getFloat");

        this.addRule(Double.class,"getDouble");
        this.addRule(double.class,"getDouble");

        this.addRule(boolean.class,"getBoolean");

        this.addRule(byte.class,"getByte");
        this.addRule(short.class,"getShort");

        this.addRule(Time.class,"getTime");
        this.addRule(Timestamp.class,"getTimestamp");
        this.addRule(Date.class,"getDate");
        this.addRule(byte[].class,"getBytes");

        this.addRule(Blob.class,"getBlob");
        this.addRule(Clob.class,"getClob");
    }

    public String getMethod(Class className) {
        return map.getOrDefault(className, "getString");
    }

}