Examples with WriteableConfiguration org.apache.nifi.minifi.c2.api.cache.WriteableConfiguration used on opensource projects

Example 1 with WriteableConfiguration

use of org.apache.nifi.minifi.c2.api.cache.WriteableConfiguration in project nifi-minifi by apache.

the class CacheConfigurationProviderTest method testGetConfiguration.

@Test
public void testGetConfiguration() throws ConfigurationProviderException {
    int version = 99;
    ConfigurationCacheFileInfo configurationCacheFileInfo = mock(ConfigurationCacheFileInfo.class);
    ConfigurationCacheFileInfo configurationCacheFileInfo2 = mock(ConfigurationCacheFileInfo.class);
    WriteableConfiguration configuration = mock(WriteableConfiguration.class);
    WriteableConfiguration configuration2 = mock(WriteableConfiguration.class);
    Map<String, List<String>> parameters = mock(Map.class);
    when(configConfigurationCache.getCacheFileInfo(TEST_CONTENT_TYPE, parameters)).thenReturn(configurationCacheFileInfo);
    when(configConfigurationCache.getCacheFileInfo(TEST_CONTENT_TYPE_2, parameters)).thenReturn(configurationCacheFileInfo2);
    when(configurationCacheFileInfo.getConfiguration(version)).thenReturn(configuration);
    when(configurationCacheFileInfo2.getConfiguration(version)).thenReturn(configuration2);
    assertEquals(configuration, cacheConfigurationProvider.getConfiguration(TEST_CONTENT_TYPE, version, parameters));
    assertEquals(configuration2, cacheConfigurationProvider.getConfiguration(TEST_CONTENT_TYPE_2, version, parameters));
}

Example 2 with WriteableConfiguration

use of org.apache.nifi.minifi.c2.api.cache.WriteableConfiguration in project nifi-minifi by apache.

the class DelegatingConfigurationProvider method getConfiguration.

@Override
public Configuration getConfiguration(String contentType, Integer version, Map<String, List<String>> parameters) throws ConfigurationProviderException {
    HttpURLConnection remoteC2ServerConnection = null;
    try {
        if (version == null) {
            remoteC2ServerConnection = getDelegateConnection(contentType, parameters);
            version = Integer.parseInt(remoteC2ServerConnection.getHeaderField("X-Content-Version"));
            if (logger.isDebugEnabled()) {
                logger.debug("Got current version " + version + " from upstream.");
            }
        }
        ConfigurationCacheFileInfo cacheFileInfo = configurationCache.getCacheFileInfo(contentType, parameters);
        WriteableConfiguration configuration = cacheFileInfo.getConfiguration(version);
        if (!configuration.exists()) {
            if (remoteC2ServerConnection == null) {
                remoteC2ServerConnection = getDelegateConnection(contentType, parameters);
            }
            try (InputStream inputStream = remoteC2ServerConnection.getInputStream();
                OutputStream outputStream = configuration.getOutputStream()) {
                IOUtils.copy(inputStream, outputStream);
            } catch (IOException e) {
                throw new ConfigurationProviderException("Unable to copy remote configuration to cache.", e);
            }
        }
        return configuration;
    } finally {
        if (remoteC2ServerConnection != null) {
            remoteC2ServerConnection.disconnect();
        }
    }
}

Example 3 with WriteableConfiguration

use of org.apache.nifi.minifi.c2.api.cache.WriteableConfiguration in project nifi-minifi by apache.

the class DelegatingConfigurationProviderTest method testGetConfigurationExistsWithVersion.

@Test
public void testGetConfigurationExistsWithVersion() throws ConfigurationProviderException {
    ConfigurationCacheFileInfo configurationCacheFileInfo = mock(ConfigurationCacheFileInfo.class);
    WriteableConfiguration configuration = mock(WriteableConfiguration.class);
    when(configurationCache.getCacheFileInfo(contentType, parameters)).thenReturn(configurationCacheFileInfo);
    when(configurationCacheFileInfo.getConfiguration(version)).thenReturn(configuration);
    when(configuration.exists()).thenReturn(true);
    assertEquals(configuration, delegatingConfigurationProvider.getConfiguration(contentType, version, parameters));
}

Example 4 with WriteableConfiguration

use of org.apache.nifi.minifi.c2.api.cache.WriteableConfiguration in project nifi-minifi by apache.

the class DelegatingConfigurationProviderTest method testGetConfigurationDoesntExistWithVersion.

@Test
public void testGetConfigurationDoesntExistWithVersion() throws ConfigurationProviderException, IOException {
    ConfigurationCacheFileInfo configurationCacheFileInfo = mock(ConfigurationCacheFileInfo.class);
    WriteableConfiguration configuration = mock(WriteableConfiguration.class);
    byte[] payload = "payload".getBytes(StandardCharsets.UTF_8);
    ByteArrayOutputStream output = new ByteArrayOutputStream();
    when(httpURLConnection.getInputStream()).thenReturn(new ByteArrayInputStream(payload));
    when(configuration.getOutputStream()).thenReturn(output);
    when(configurationCache.getCacheFileInfo(contentType, parameters)).thenReturn(configurationCacheFileInfo);
    when(configurationCacheFileInfo.getConfiguration(version)).thenReturn(configuration);
    when(configuration.exists()).thenReturn(false);
    assertEquals(configuration, delegatingConfigurationProvider.getConfiguration(contentType, version, parameters));
    assertArrayEquals(payload, output.toByteArray());
}

Example 5 with WriteableConfiguration

use of org.apache.nifi.minifi.c2.api.cache.WriteableConfiguration in project nifi-minifi by apache.

the class FileSystemConfigurationCacheTest method getConfigurationTest.

@Test
public void getConfigurationTest() throws IOException, ConfigurationProviderException {
    final String pathRoot = "files";
    final String pathPattern = "config";
    FileSystemConfigurationCache cache = new FileSystemConfigurationCache(pathRoot, pathPattern);
    Map<String, List<String>> parameters = new HashMap<>();
    ConfigurationCacheFileInfo info = cache.getCacheFileInfo("text/yaml", parameters);
    WriteableConfiguration configuration = info.getConfiguration(1);
    assertEquals("config.text.yaml.v1", configuration.getName());
    assertEquals("1", configuration.getVersion());
    assertTrue(configuration.exists());
}