Search in sources :

Example 11 with StorableKey

use of com.hortonworks.registries.storage.StorableKey in project streamline by hortonworks.

the class StreamCatalogService method getTopologyTestRunCaseSource.

public TopologyTestRunCaseSource getTopologyTestRunCaseSource(Long testcaseId, Long id) {
    TopologyTestRunCaseSource testCaseSource = new TopologyTestRunCaseSource();
    testCaseSource.setId(id);
    TopologyTestRunCaseSource retrieved = dao.get(new StorableKey(TopologyTestRunCaseSource.NAMESPACE, testCaseSource.getPrimaryKey()));
    if (retrieved == null || !retrieved.getTestCaseId().equals(testcaseId)) {
        return null;
    }
    return retrieved;
}
Also used : StorableKey(com.hortonworks.registries.storage.StorableKey) TopologyTestRunCaseSource(com.hortonworks.streamline.streams.catalog.TopologyTestRunCaseSource)

Example 12 with StorableKey

use of com.hortonworks.registries.storage.StorableKey in project streamline by hortonworks.

the class StreamCatalogService method getTopologyTestRunHistory.

public TopologyTestRunHistory getTopologyTestRunHistory(Long id) {
    TopologyTestRunHistory history = new TopologyTestRunHistory();
    history.setId(id);
    return dao.get(new StorableKey(TopologyTestRunHistory.NAMESPACE, history.getPrimaryKey()));
}
Also used : StorableKey(com.hortonworks.registries.storage.StorableKey) TopologyTestRunHistory(com.hortonworks.streamline.streams.catalog.TopologyTestRunHistory)

Example 13 with StorableKey

use of com.hortonworks.registries.storage.StorableKey in project streamline by hortonworks.

the class StreamCatalogService method removeTopologyState.

public TopologyState removeTopologyState(Long topologyId) {
    TopologyState state = new TopologyState();
    state.setTopologyId(topologyId);
    return dao.remove(new StorableKey(TOPOLOGY_STATE_NAMESPACE, state.getPrimaryKey()));
}
Also used : TopologyState(com.hortonworks.streamline.streams.catalog.topology.state.TopologyState) StorableKey(com.hortonworks.registries.storage.StorableKey)

Example 14 with StorableKey

use of com.hortonworks.registries.storage.StorableKey in project streamline by hortonworks.

the class StreamCatalogService method removeStreamInfo.

public TopologyStream removeStreamInfo(Long topologyId, Long streamId, Long versionId) {
    TopologyStream topologyStream = getStreamInfo(topologyId, streamId, versionId);
    if (topologyStream != null) {
        setReconfigureTarget(topologyStream);
        topologyStream = dao.remove(new StorableKey(STREAMINFO_NAMESPACE, topologyStream.getPrimaryKey()));
        topologyStream.setVersionTimestamp(updateVersionTimestamp(versionId).getTimestamp());
    }
    return topologyStream;
}
Also used : StorableKey(com.hortonworks.registries.storage.StorableKey) TopologyStream(com.hortonworks.streamline.streams.catalog.TopologyStream)

Example 15 with StorableKey

use of com.hortonworks.registries.storage.StorableKey in project streamline by hortonworks.

the class StreamCatalogServiceTest method testListTopologiesProvidesVersionedTimestamp_STREAMLINE_526.

@Test
public void testListTopologiesProvidesVersionedTimestamp_STREAMLINE_526() {
    List<Topology> topologies = new ArrayList<>();
    topologies.add(createTopology(1L));
    topologies.add(createTopology(2L));
    topologies.add(createTopology(3L));
    List<TopologyVersion> versions = topologies.stream().map(x -> createTopologyVersionInfo(x.getId(), x.getId())).collect(Collectors.toList());
    new Expectations() {

        {
            dao.find(withEqual(new Topology().getNameSpace()), withAny(new ArrayList<>()));
            result = topologies;
            dao.find(withEqual(new TopologyVersion().getNameSpace()), withAny(new ArrayList<>()));
            result = versions;
            dao.get(withEqual(new StorableKey(versions.get(0).getNameSpace(), versions.get(0).getPrimaryKey())));
            result = versions.get(0);
            dao.get(withEqual(new StorableKey(versions.get(1).getNameSpace(), versions.get(1).getPrimaryKey())));
            result = versions.get(1);
            dao.get(withEqual(new StorableKey(versions.get(2).getNameSpace(), versions.get(2).getPrimaryKey())));
            result = versions.get(2);
        }
    };
    Collection<Topology> result = streamCatalogService.listTopologies();
    assertTrue(result.size() > 0);
    assertFalse(result.stream().anyMatch(x -> x.getVersionTimestamp() == null));
}
Also used : Topology(com.hortonworks.streamline.streams.catalog.Topology) MLModelRegistryClient(com.hortonworks.streamline.registries.model.client.MLModelRegistryClient) Arrays(java.util.Arrays) Expectations(mockit.Expectations) FileStorage(com.hortonworks.registries.common.util.FileStorage) RunWith(org.junit.runner.RunWith) ArrayList(java.util.ArrayList) Map(java.util.Map) Projection(com.hortonworks.streamline.streams.catalog.Projection) JMockit(mockit.integration.junit4.JMockit) TopologyVersion(com.hortonworks.streamline.streams.catalog.TopologyVersion) Tested(mockit.Tested) Collection(java.util.Collection) Assert.assertTrue(org.junit.Assert.assertTrue) Test(org.junit.Test) Collectors(java.util.stream.Collectors) List(java.util.List) TreeMap(java.util.TreeMap) Assert.assertFalse(org.junit.Assert.assertFalse) Injectable(mockit.Injectable) Assert(org.junit.Assert) StorableKey(com.hortonworks.registries.storage.StorableKey) StorageManager(com.hortonworks.registries.storage.StorageManager) Collections(java.util.Collections) Assert.assertEquals(org.junit.Assert.assertEquals) Expectations(mockit.Expectations) TopologyVersion(com.hortonworks.streamline.streams.catalog.TopologyVersion) StorableKey(com.hortonworks.registries.storage.StorableKey) ArrayList(java.util.ArrayList) Topology(com.hortonworks.streamline.streams.catalog.Topology) Test(org.junit.Test)

Aggregations

StorableKey (com.hortonworks.registries.storage.StorableKey)72 Schema (com.hortonworks.registries.common.Schema)6 OrderByField (com.hortonworks.registries.storage.OrderByField)6 PrimaryKey (com.hortonworks.registries.storage.PrimaryKey)6 HashMap (java.util.HashMap)6 Test (org.junit.Test)5 QueryParam (com.hortonworks.registries.common.QueryParam)4 StorageManager (com.hortonworks.registries.storage.StorageManager)4 SchemaNotFoundException (com.hortonworks.registries.schemaregistry.errors.SchemaNotFoundException)3 Storable (com.hortonworks.registries.storage.Storable)3 StorageException (com.hortonworks.registries.storage.exception.StorageException)3 Role (com.hortonworks.streamline.streams.security.catalog.Role)3 UserRole (com.hortonworks.streamline.streams.security.catalog.UserRole)3 CacheBuilder (com.google.common.cache.CacheBuilder)2 SchemaVersionInfoCache (com.hortonworks.registries.schemaregistry.cache.SchemaVersionInfoCache)2 CacheBackedStorageManager (com.hortonworks.registries.storage.CacheBackedStorageManager)2 StorageWriter (com.hortonworks.registries.storage.cache.writer.StorageWriter)2 AlreadyExistsException (com.hortonworks.registries.storage.exception.AlreadyExistsException)2 IllegalQueryParameterException (com.hortonworks.registries.storage.exception.IllegalQueryParameterException)2 MySqlSelectQuery (com.hortonworks.registries.storage.impl.jdbc.provider.mysql.query.MySqlSelectQuery)2