Examples with MetadataSupervisorManager org.apache.druid.metadata.MetadataSupervisorManager used on opensource projects

Example 1 with MetadataSupervisorManager

use of org.apache.druid.metadata.MetadataSupervisorManager in project druid by druid-io.

the class MaterializedViewSupervisorTest method setUp.

@Before
public void setUp() {
    TestDerbyConnector derbyConnector = derbyConnectorRule.getConnector();
    derbyConnector.createDataSourceTable();
    derbyConnector.createSegmentTable();
    taskStorage = EasyMock.createMock(TaskStorage.class);
    taskMaster = EasyMock.createMock(TaskMaster.class);
    indexerMetadataStorageCoordinator = new IndexerSQLMetadataStorageCoordinator(objectMapper, derbyConnectorRule.metadataTablesConfigSupplier().get(), derbyConnector);
    metadataSupervisorManager = EasyMock.createMock(MetadataSupervisorManager.class);
    sqlSegmentsMetadataManager = EasyMock.createMock(SqlSegmentsMetadataManager.class);
    taskQueue = EasyMock.createMock(TaskQueue.class);
    taskQueue.start();
    objectMapper.registerSubtypes(new NamedType(HashBasedNumberedShardSpec.class, "hashed"));
    spec = new MaterializedViewSupervisorSpec("base", new DimensionsSpec(Collections.singletonList(new StringDimensionSchema("dim"))), new AggregatorFactory[] { new LongSumAggregatorFactory("m1", "m1") }, HadoopTuningConfig.makeDefaultTuningConfig(), null, null, null, null, null, false, objectMapper, taskMaster, taskStorage, metadataSupervisorManager, sqlSegmentsMetadataManager, indexerMetadataStorageCoordinator, new MaterializedViewTaskConfig(), EasyMock.createMock(AuthorizerMapper.class), EasyMock.createMock(ChatHandlerProvider.class), new SupervisorStateManagerConfig());
    derivativeDatasourceName = spec.getDataSourceName();
    supervisor = (MaterializedViewSupervisor) spec.createSupervisor();
}

Example 2 with MetadataSupervisorManager

use of org.apache.druid.metadata.MetadataSupervisorManager in project druid by druid-io.

the class SeekableStreamSupervisor method changeTaskCountInIOConfig.

private void changeTaskCountInIOConfig(int desiredActiveTaskCount) {
    ioConfig.setTaskCount(desiredActiveTaskCount);
    try {
        Optional<SupervisorManager> supervisorManager = taskMaster.getSupervisorManager();
        if (supervisorManager.isPresent()) {
            MetadataSupervisorManager metadataSupervisorManager = supervisorManager.get().getMetadataSupervisorManager();
            metadataSupervisorManager.insert(dataSource, spec);
        } else {
            log.error("supervisorManager is null in taskMaster, skipping scale action for dataSource [%s].", dataSource);
        }
    } catch (Exception e) {
        log.error(e, "Failed to sync taskCount to MetaStorage for dataSource [%s].", dataSource);
    }
}