Examples with ImmutableEntry org.apache.drill.common.collections.ImmutableEntry used on opensource projects

Example 1 with ImmutableEntry

use of org.apache.drill.common.collections.ImmutableEntry in project drill by apache.

the class LocalPersistentStore method getRange.

@Override
public Iterator<Map.Entry<String, V>> getRange(int skip, int take) {
    try (AutoCloseableLock lock = readLock.open()) {
        try {
            List<FileStatus> f = fs.list(false, basePath);
            if (f == null || f.isEmpty()) {
                return Collections.emptyIterator();
            }
            List<String> files = Lists.newArrayList();
            for (FileStatus stat : f) {
                String s = stat.getPath().getName();
                if (s.endsWith(DRILL_SYS_FILE_SUFFIX)) {
                    files.add(s.substring(0, s.length() - DRILL_SYS_FILE_SUFFIX.length()));
                }
            }
            Collections.sort(files);
            return Iterables.transform(Iterables.limit(Iterables.skip(files, skip), take), new Function<String, Entry<String, V>>() {

                @Nullable
                @Override
                public Entry<String, V> apply(String key) {
                    return new ImmutableEntry<>(key, get(key));
                }
            }).iterator();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
}

Example 2 with ImmutableEntry

use of org.apache.drill.common.collections.ImmutableEntry in project drill by axbaretto.

the class LocalPersistentStore method getRange.

@Override
public Iterator<Map.Entry<String, V>> getRange(int skip, int take) {
    try {
        // list only files with sys file suffix
        PathFilter sysFileSuffixFilter = new PathFilter() {

            @Override
            public boolean accept(Path path) {
                return path.getName().endsWith(DRILL_SYS_FILE_SUFFIX);
            }
        };
        List<FileStatus> fileStatuses = DrillFileSystemUtil.listFiles(fs, basePath, false, sysFileSuffixFilter);
        if (fileStatuses.isEmpty()) {
            return Collections.emptyIterator();
        }
        List<String> files = Lists.newArrayList();
        for (FileStatus stat : fileStatuses) {
            String s = stat.getPath().getName();
            files.add(s.substring(0, s.length() - DRILL_SYS_FILE_SUFFIX.length()));
        }
        Collections.sort(files);
        return Iterables.transform(Iterables.limit(Iterables.skip(files, skip), take), new Function<String, Entry<String, V>>() {

            @Nullable
            @Override
            public Entry<String, V> apply(String key) {
                return new ImmutableEntry<>(key, get(key));
            }
        }).iterator();
    } catch (IOException e) {
        throw new RuntimeException(e);
    }
}

Example 3 with ImmutableEntry

use of org.apache.drill.common.collections.ImmutableEntry in project drill by axbaretto.

the class TestZookeeperClient method testEntriesReturnsRelativePaths.

@Test
public void testEntriesReturnsRelativePaths() throws Exception {
    final ChildData child = Mockito.mock(ChildData.class);
    Mockito.when(child.getPath()).thenReturn(abspath);
    Mockito.when(child.getData()).thenReturn(data);
    final List<ChildData> children = Lists.newArrayList(child);
    Mockito.when(client.getCache().getCurrentData()).thenReturn(children);
    final Iterator<Map.Entry<String, byte[]>> entries = client.entries();
    // returned entry must contain the given relative path
    final Map.Entry<String, byte[]> expected = new ImmutableEntry<>(path, data);
    assertEquals("entries do not match", expected, entries.next());
}

Example 4 with ImmutableEntry

use of org.apache.drill.common.collections.ImmutableEntry in project drill by apache.

the class TestZookeeperClient method testEntriesReturnsRelativePaths.

@Test
public void testEntriesReturnsRelativePaths() {
    final ChildData child = Mockito.mock(ChildData.class);
    Mockito.when(child.getPath()).thenReturn(abspath);
    Mockito.when(child.getData()).thenReturn(data);
    final List<ChildData> children = Lists.newArrayList(child);
    Mockito.when(client.getCache().getCurrentData()).thenReturn(children);
    final Iterator<Map.Entry<String, byte[]>> entries = client.entries();
    // returned entry must contain the given relative path
    final Map.Entry<String, byte[]> expected = new ImmutableEntry<>(path, data);
    assertEquals("entries do not match", expected, entries.next());
}