Создание мультитенантной архитектуры приложений с использованием Vaadin, Spring, jOOQ и PostreSQL

С помощью этого трюка наследования мы гарантируем, что фактическая структура таблиц во всех арендаторах синхронизируется, и, как мы увидим позже, мы можем выполнять запросы по всем арендаторам из главной схемы.

Скрипты liquibase в примере github создают двух арендаторов (tenant_1 и tenant_2) и по одному администратору каждый. имя пользователя admin для простоты совпадает с именем арендатора, пароль просто «admin». Поэтому, если вы попытаетесь запустить пример, вы можете войти в систему как tenant_1 / tenant_1 / admin.

Теперь у нас есть базовая схема базы данных. Скрипты liquibase нуждались в некоторой настройке для запуска не только на postgres, но и на hsqldb. Это нужно по крайней мере для модульного тестирования, но мы также хотим иметь возможность генерировать наши jooq mapping и dsl без необходимости запуска сервера postgresql на нашем сервере сборки.

При запуске сборки maven создается основная схема на временном экземпляре hsqldb и впоследствии запускается генератор кода jooq. Идея для этого пришла от  http://stanislas.github.io/2014/08/23/liquibase-and-jooq.html

Это все основные вещи, которые нам нужны, прежде чем мы теперь сможем заняться мультитенантной деятельностью. Я хотел иметь четкое разделение арендаторов с точки зрения подключения. Таким образом, у каждого арендатора есть собственный пользователь базы данных с ограниченным доступом и собственный пул соединений. У этого есть некоторые недостатки, поскольку мы, вероятно, могли бы получить довольно много пулов соединений, но преимущество в том, что независимо от того, что вы пытаетесь сделать с соединением арендатора, которое вы получите от весны, вы никогда не сможете прикоснуться к другому арендатору, чем тот, для которого вы аутентифицированы. TenantDataSource — это простая прокси-оболочка, которая направляет вас в соответствующий пул соединений на основе текущего объекта аутентификации Spring. Вот фрагмент:

@Component@Qualifier("no-tx")
public class MultiTenantDataSource implements DataSource {

    @Autowired

    private TenantAuthentication authentication;


    @Autowired

    private TenantDao tenantDao;


    @Autowired

    private TenantHelper tenantHelper;

        

    private Map<String, DataSource> dataSourceMap = new ConcurrentHashMap<>();
    @Override

    public Connection getConnection() throws SQLException {

        DataSource ds = getDataSource();

        return ds.getConnection();

    }

    private DataSource getDataSource() {

        String tenantName = authentication.getTenant(); // will throw if not authenticated

        return  dataSourceMap.computeIfAbsent(tenantName, (key)->{
            Optional<ITenant> tenantOptional = tenantDao.findbyName(tenantName);

            return tenantOptional.map(tenant -> {

                HikariConfig config = tenantHelper.toHikariConfig(tenant);

                return new HikariDataSource(config);

            }).orElseThrow(() -> new IllegalStateException("This should never happen"));

         });
    }

@Bean(name ="currentToken")
TenantAuthentication currentToken() {

    return ProxyFactory.getProxy(TenantAuthentication.class, new MethodInterceptor() {
        @Override

        public Object invoke(MethodInvocation invocation) throws Throwable {

            SecurityContext securityContext = SecurityContextHolder.getContext();

            TenantAuthenticationToken authentication = (TenantAuthenticationToken)securityContext.getAuthentication();

            if (authentication == null) {

                 throw new AuthenticationCredentialsNotFoundException("No auth..");

            }

            return invocation.getMethod().invoke(authentication, invocation.getArguments());

        }

    });

}

@Bean

public DSLContext dsl(){

    return ProxyFactory.getProxy(DSLContext.class, new MethodInterceptor() {

        Map<String, DSLContext> contextMap = new ConcurrentHashMap<>();

        @Override

        public Object invoke(MethodInvocation invocation) throws Throwable {

            String tenant = authentication.getTenant(); // will throw if not authenticated

            DSLContext ctx = contextMap.computeIfAbsent(tenant, (key) ->{ 

                Settings settings = new Settings().withRenderMapping(new RenderMapping().withSchemata(new MappedSchema().withInput("master").withOutput(key)));

                DefaultConfiguration configuration = new DefaultConfiguration();

                configuration.setSQLDialect(SQLDialect.POSTGRES_9_4);

                configuration.setSettings(settings);

                configuration.setConnectionProvider(connectionProvider());

                configuration.setExecuteListenerProvider(new DefaultExecuteListenerProvider(new ExceptionTranslator()));

                return new DefaultDSLContext(configuration);

            });

            return invocation.getMethod().invoke(ctx, invocation.getArguments());

        }

    });

}

@Component

public class UserDao {

    /**

     * the dsl context.

     *

     * the context we inject here is actually a proxy bound to the current tenant schema

     * backed by a datasource that has only access to this tenant schema

     */

    @Autowired

    DSLContext dsl;


    public List<IUser> findAll() {

        return dsl.selectFrom(USER).fetchInto(User.class);

    }

}

<sql>
    create view "master"."v_user" as
    select n.nspname as tenant, u.user_name, u.password_hash 

    from master.user u left join pg_class p on u.TABLEOID = p.oid 

    left join pg_catalog.pg_namespace n on n.OID =p.relnamespace 

    where u.active=true;
</sql>