Examples with ClusterList com.yahoo.vespaclient.ClusterList used on opensource projects

Example 31 with ClusterList

use of com.yahoo.vespaclient.ClusterList in project vespa by vespa-engine.

the class VdsVisitTestCase method testRequireClusterOptionIfMultipleClusters.

@Test
public void testRequireClusterOptionIfMultipleClusters() {
    List<ClusterDef> clusterDefs = new ArrayList<>();
    clusterDefs.add(new ClusterDef("storage", "content/cluster.foo/storage"));
    clusterDefs.add(new ClusterDef("storage2", "content/cluster.bar/storage"));
    ClusterList clusterList = new ClusterList(clusterDefs);
    try {
        VdsVisit.resolveClusterRoute(clusterList, null);
    } catch (IllegalArgumentException e) {
        assertTrue(e.getMessage().contains("Please use the -c option to select one of them"));
    }
}

Example 32 with ClusterList

use of com.yahoo.vespaclient.ClusterList in project vespa by vespa-engine.

the class VespaFeeder method parseFiles.

void parseFiles(InputStream stdin, PrintStream output) throws Exception {
    FeedContext context = new FeedContext(args.getPropertyProcessor(), args.getSessionFactory(), manager, new ClusterList(), new NullFeedMetric());
    final BufferedInputStream input = new BufferedInputStream(stdin);
    VespaFeedHandler handler = VespaFeedHandler.createFromContext(context, threadPool);
    if (args.getFiles().isEmpty()) {
        InputStreamRequest req = new InputStreamRequest(input);
        setProperties(req, input);
        FeedResponse response = (FeedResponse) handler.handle(req.toRequest(), createProgressCallback(output));
        if (!response.isSuccess()) {
            throw renderErrors(response.getErrorList());
        }
    } else {
        if (args.isVerbose()) {
            for (String fileName : args.getFiles()) {
                long thisSize = new File(fileName).length();
                output.println("Size of file '" + fileName + "' is " + thisSize + " B.");
            }
        }
        for (String fileName : args.getFiles()) {
            File f = new File(fileName);
            FileRequest req = new FileRequest(f);
            final BufferedInputStream inputSnooper = new BufferedInputStream(new FileInputStream(fileName));
            setProperties(req, inputSnooper);
            inputSnooper.close();
            FeedResponse response = (FeedResponse) handler.handle(req.toRequest(), createProgressCallback(output));
            if (!response.isSuccess()) {
                throw renderErrors(response.getErrorList());
            }
        }
    }
}

Example 33 with ClusterList

use of com.yahoo.vespaclient.ClusterList in project vespa by vespa-engine.

the class DocumentRetrieverTest method testClusterLookup.

@Test
public void testClusterLookup() throws DocumentRetrieverException {
    final String cluster = "storage", configId = "content/cluster.foo/storage", expectedRoute = "[Storage:cluster=storage;clusterconfigid=content/cluster.foo/storage]";
    ClientParameters params = createParameters().setCluster(cluster).build();
    ClusterList clusterList = new ClusterList(Collections.singletonList(new ClusterDef(cluster, configId)));
    DocumentRetriever documentRetriever = createDocumentRetriever(params, clusterList);
    documentRetriever.retrieveDocuments();
    verify(mockedFactory).createDocumentAccess(argThat(new ArgumentMatcher<MessageBusParams>() {

        @Override
        public boolean matches(Object o) {
            return ((MessageBusParams) o).getRoute().equals(expectedRoute);
        }
    }));
}

Example 34 with ClusterList

use of com.yahoo.vespaclient.ClusterList in project vespa by vespa-engine.

the class DocumentRetrieverTest method testInvalidClusterName.

@Test
public void testInvalidClusterName() throws DocumentRetrieverException {
    exception.expect(DocumentRetrieverException.class);
    exception.expectMessage("The Vespa cluster contains the content clusters storage, not invalidclustername. Please select a valid vespa cluster.");
    ClientParameters params = createParameters().setCluster("invalidclustername").build();
    ClusterList clusterList = new ClusterList(Collections.singletonList(new ClusterDef("storage", "content/cluster.foo/storage")));
    DocumentRetriever documentRetriever = createDocumentRetriever(params, clusterList);
    documentRetriever.retrieveDocuments();
}