com.codingapi.springboot.permission.jdbc.AbsWrapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of springboot-starter-data-permission Show documentation

springboot-starter-data-permission project for Spring Boot

There is a newer version: 1.4.0

Show newest version

package com.codingapi.springboot.permission.jdbc;

import java.sql.SQLException;
import java.sql.Wrapper;

/**
 * @author lorne
 * @since 1.0.0
 */
public abstract class AbsWrapper implements Wrapper {

    private final Object delegate;

    public AbsWrapper(Object delegate) {
        this.delegate = delegate;
    }

    @Override
    public  T unwrap(Class iface) throws SQLException {
        final Object result;
        if (iface.isAssignableFrom(getClass())) {
            // if the proxy directly implements the interface or extends it, return the proxy
            result = this;
        } else if (iface.isAssignableFrom(delegate.getClass())) {
            // if the proxied object directly implements the interface or extends it, return
            // the proxied object
            result = unwrapProxy();
        } else if (Wrapper.class.isAssignableFrom(delegate.getClass())) {
            // if the proxied object implements the wrapper interface, then
            // return the result of it's unwrap method.
            result = ((Wrapper) unwrapProxy()).unwrap(iface);
        } else {
          /*
             This line of code can only be reached when the underlying object does not implement the wrapper
             interface.  This would mean that either the JDBC driver or the wrapper of the underlying object
             does not implement the JDBC 4.0 API.
           */
            throw new SQLException("Can not unwrap to " + iface.getName());
        }
        return iface.cast(result);
    }

    @Override
    public boolean isWrapperFor(Class iface) throws SQLException {
        if (iface.isAssignableFrom(getClass())) {
            // if the proxy directly proxy the interface or extends it, return true
            return true;
        } else if (iface.isAssignableFrom(delegate.getClass())) {
            // if the proxied object directly implements the interface or extends it, return true
            return true;
        } else if (Wrapper.class.isAssignableFrom(delegate.getClass())) {
            // if the proxied object implements the wrapper interface, then
            // return the result of it's isWrapperFor method.
            return ((Wrapper) unwrapProxy()).isWrapperFor(iface);
        }
        return false;
    }


    @Override
    public int hashCode() {
        return delegate.hashCode();
    }

    public Object unwrapProxy() {
        return delegate;
    }

}