open-JSD-7610/src/main/java/com/fr/plugin/xxxx/saml/SignDBAccessProvider.java

package com.fr.plugin.xxxx.saml;

import com.fr.db.fun.impl.AbstractDBAccessProvider;
import com.fr.plugin.xxxx.saml.dao.RoleUrlDao;
import com.fr.plugin.xxxx.saml.dao.SignReportDao;
import com.fr.plugin.xxxx.saml.dao.UserRoleDao;
import com.fr.plugin.xxxx.saml.entity.RoleUrlEntity;
import com.fr.plugin.xxxx.saml.entity.SignReportEntity;
import com.fr.plugin.xxxx.saml.entity.UserRoleEntity;
import com.fr.stable.db.accessor.DBAccessor;
import com.fr.stable.db.dao.BaseDAO;
import com.fr.stable.db.dao.DAOProvider;

/**
 * @Author fr.open
 * @Date 2020/12/3
 * @Description
 **/
public class SignDBAccessProvider extends AbstractDBAccessProvider {

    private static DBAccessor dbAccessor = null;

    public static DBAccessor getDbAccessor() {
        return dbAccessor;
    }

    @Override
    public DAOProvider[] registerDAO() {
        return new DAOProvider[]{
                new DAOProvider() {
                    @Override
                    public Class getEntityClass() {
                        return SignReportEntity.class;
                    }

                    @Override
                    public Class<? extends BaseDAO> getDAOClass() {
                        return SignReportDao.class;
                    }
                },
                new DAOProvider() {
                    @Override
                    public Class getEntityClass() {
                        return UserRoleEntity.class;
                    }

                    @Override
                    public Class<? extends BaseDAO> getDAOClass() {
                        return UserRoleDao.class;
                    }
                },
                new DAOProvider() {
                    @Override
                    public Class getEntityClass() {
                        return RoleUrlEntity.class;
                    }

                    @Override
                    public Class<? extends BaseDAO> getDAOClass() {
                        return RoleUrlDao.class;
                    }
                }
        };
    }

    @Override
    public void onDBAvailable(DBAccessor dbAccessor) {
        SignDBAccessProvider.dbAccessor = dbAccessor;
    }
}