Examples with ComplexShardingStrategyConfiguration io.shardingjdbc.core.api.config.strategy.ComplexShardingStrategyConfiguration used on opensource projects

Example 1 with ComplexShardingStrategyConfiguration

use of io.shardingjdbc.core.api.config.strategy.ComplexShardingStrategyConfiguration in project sharding-jdbc by shardingjdbc.

the class InsertStatementParserTest method createShardingRuleWithGenerateKeyColumns.

private ShardingRule createShardingRuleWithGenerateKeyColumns() {
    DataSource dataSource = mock(DataSource.class);
    Connection connection = mock(Connection.class);
    DatabaseMetaData databaseMetaData = mock(DatabaseMetaData.class);
    try {
        when(dataSource.getConnection()).thenReturn(connection);
        when(connection.getMetaData()).thenReturn(databaseMetaData);
        when(databaseMetaData.getDatabaseProductName()).thenReturn("H2");
    } catch (final SQLException ex) {
        throw new RuntimeException(ex);
    }
    final ShardingRuleConfiguration shardingRuleConfig = new ShardingRuleConfiguration();
    TableRuleConfiguration tableRuleConfig = new TableRuleConfiguration();
    tableRuleConfig.setLogicTable("TABLE_XXX");
    tableRuleConfig.setActualDataNodes("ds.table_${0..2}");
    tableRuleConfig.setTableShardingStrategyConfig(new ComplexShardingStrategyConfiguration("field1", new TestComplexKeysShardingAlgorithm()));
    tableRuleConfig.setKeyGeneratorColumnName("field2");
    shardingRuleConfig.getTableRuleConfigs().add(tableRuleConfig);
    shardingRuleConfig.setDefaultKeyGenerator(new IncrementKeyGenerator());
    return new ShardingRule(shardingRuleConfig, Lists.newArrayList("ds"));
}

Example 2 with ComplexShardingStrategyConfiguration

use of io.shardingjdbc.core.api.config.strategy.ComplexShardingStrategyConfiguration in project sharding-jdbc by shardingjdbc.

the class ShardingStrategyTest method assertDoShardingForMultipleKeys.

@Test
public void assertDoShardingForMultipleKeys() {
    ComplexShardingStrategy strategy = new ComplexShardingStrategy(new ComplexShardingStrategyConfiguration("column", new TestComplexKeysShardingAlgorithm()));
    assertThat(strategy.doSharding(targets, Collections.<ShardingValue>singletonList(new PreciseShardingValue<>("logicTable", "column", "1"))), is((Collection<String>) Sets.newHashSet("1", "2", "3")));
}

Example 3 with ComplexShardingStrategyConfiguration

use of io.shardingjdbc.core.api.config.strategy.ComplexShardingStrategyConfiguration in project sharding-jdbc by shardingjdbc.

the class YamlShardingStrategyConfiguration method build.

/**
 * Build sharding strategy configuration.
 *
 * @return sharding strategy configuration
 */
public ShardingStrategyConfiguration build() {
    int shardingStrategyConfigCount = 0;
    ShardingStrategyConfiguration result = null;
    if (null != standard) {
        shardingStrategyConfigCount++;
        if (null == standard.getRangeAlgorithmClassName()) {
            result = new StandardShardingStrategyConfiguration(standard.getShardingColumn(), ShardingAlgorithmFactory.newInstance(standard.getPreciseAlgorithmClassName(), PreciseShardingAlgorithm.class));
        } else {
            result = new StandardShardingStrategyConfiguration(standard.getShardingColumn(), ShardingAlgorithmFactory.newInstance(standard.getPreciseAlgorithmClassName(), PreciseShardingAlgorithm.class), ShardingAlgorithmFactory.newInstance(standard.getRangeAlgorithmClassName(), RangeShardingAlgorithm.class));
        }
    }
    if (null != complex) {
        shardingStrategyConfigCount++;
        result = new ComplexShardingStrategyConfiguration(complex.getShardingColumns(), ShardingAlgorithmFactory.newInstance(complex.getAlgorithmClassName(), ComplexKeysShardingAlgorithm.class));
    }
    if (null != inline) {
        shardingStrategyConfigCount++;
        result = new InlineShardingStrategyConfiguration(inline.getShardingColumn(), inline.getAlgorithmExpression());
    }
    if (null != hint) {
        shardingStrategyConfigCount++;
        result = new HintShardingStrategyConfiguration(ShardingAlgorithmFactory.newInstance(hint.getAlgorithmClassName(), HintShardingAlgorithm.class));
    }
    if (null != none) {
        shardingStrategyConfigCount++;
        result = new NoneShardingStrategyConfiguration();
    }
    Preconditions.checkArgument(shardingStrategyConfigCount <= 1, "Only allowed 0 or 1 sharding strategy configuration.");
    return result;
}

Example 4 with ComplexShardingStrategyConfiguration

use of io.shardingjdbc.core.api.config.strategy.ComplexShardingStrategyConfiguration in project sharding-jdbc by shardingjdbc.

the class ShardingRuleConfigurationConverterTest method assertToJsonForComplexStrategy.

@Test
public void assertToJsonForComplexStrategy() {
    ComplexShardingStrategyConfiguration actual = new ComplexShardingStrategyConfiguration("order_id,item_id", new TestComplexKeysShardingAlgorithm());
    assertThat(ShardingRuleConfigurationConverter.toJson(getCommonShardingRuleConfig(actual)), is(getJsonForComplexStrategy()));
}

Example 5 with ComplexShardingStrategyConfiguration

use of io.shardingjdbc.core.api.config.strategy.ComplexShardingStrategyConfiguration in project sharding-jdbc by shardingjdbc.

the class ShardingStrategyConfigurationGsonTypeAdapter method write.

@Override
public void write(final JsonWriter out, final ShardingStrategyConfiguration value) throws IOException {
    out.beginObject();
    if (value instanceof StandardShardingStrategyConfiguration) {
        out.name("type").value(ShardingStrategyType.STANDARD.name());
        StandardShardingStrategyConfiguration shardingStrategyConfig = (StandardShardingStrategyConfiguration) value;
        out.name("shardingColumn").value(shardingStrategyConfig.getShardingColumn());
        out.name("preciseAlgorithmClassName").value(shardingStrategyConfig.getPreciseShardingAlgorithm().getClass().getName());
        if (null != shardingStrategyConfig.getRangeShardingAlgorithm()) {
            out.name("rangeAlgorithmClassName").value(shardingStrategyConfig.getRangeShardingAlgorithm().getClass().getName());
        }
    } else if (value instanceof ComplexShardingStrategyConfiguration) {
        out.name("type").value(ShardingStrategyType.COMPLEX.name());
        ComplexShardingStrategyConfiguration shardingStrategyConfig = (ComplexShardingStrategyConfiguration) value;
        out.name("shardingColumns").value(shardingStrategyConfig.getShardingColumns());
        out.name("algorithmClassName").value(shardingStrategyConfig.getShardingAlgorithm().getClass().getName());
    } else if (value instanceof InlineShardingStrategyConfiguration) {
        out.name("type").value(ShardingStrategyType.INLINE.name());
        InlineShardingStrategyConfiguration shardingStrategyConfig = (InlineShardingStrategyConfiguration) value;
        out.name("shardingColumn").value(shardingStrategyConfig.getShardingColumn());
        out.name("algorithmExpression").value(shardingStrategyConfig.getAlgorithmExpression());
    } else if (value instanceof HintShardingStrategyConfiguration) {
        out.name("type").value(ShardingStrategyType.HINT.name());
        HintShardingStrategyConfiguration shardingStrategyConfig = (HintShardingStrategyConfiguration) value;
        out.name("algorithmClassName").value(shardingStrategyConfig.getShardingAlgorithm().getClass().getName());
    } else if (value instanceof NoneShardingStrategyConfiguration) {
        out.name("type").value(ShardingStrategyType.NONE.name());
    }
    out.endObject();
}