Examples with ZooKeeperClient com.yahoo.vespa.config.server.deploy.ZooKeeperClient used on opensource projects

Example 1 with ZooKeeperClient

use of com.yahoo.vespa.config.server.deploy.ZooKeeperClient in project vespa by vespa-engine.

the class LocalSessionTest method createSession.

private LocalSession createSession(TenantName tenant, long sessionId, SessionTest.MockSessionPreparer preparer, Optional<AllocatedHosts> allocatedHosts) throws Exception {
    SessionZooKeeperClient zkc = new MockSessionZKClient(curator, tenant, sessionId, allocatedHosts);
    zkc.createWriteStatusTransaction(Session.Status.NEW).commit();
    ZooKeeperClient zkClient = new ZooKeeperClient(configCurator, new BaseDeployLogger(), false, Tenants.getSessionsPath(tenant).append(String.valueOf(sessionId)));
    if (allocatedHosts.isPresent()) {
        zkClient.write(allocatedHosts.get());
    }
    zkClient.write(Collections.singletonMap(Version.fromIntValues(0, 0, 0), new MockFileRegistry()));
    File sessionDir = new File(tenantFileSystemDirs.sessionsPath(), String.valueOf(sessionId));
    sessionDir.createNewFile();
    return new LocalSession(tenant, sessionId, preparer, new SessionContext(FilesApplicationPackage.fromFile(testApp), zkc, sessionDir, new MemoryTenantApplications(), new HostRegistry<>(), superModelGenerationCounter));
}

Example 2 with ZooKeeperClient

use of com.yahoo.vespa.config.server.deploy.ZooKeeperClient in project vespa by vespa-engine.

the class SessionActiveHandlerTest method createRemoteSession.

private RemoteSession createRemoteSession(long sessionId, Session.Status status, SessionZooKeeperClient zkClient, Clock clock) throws IOException {
    zkClient.writeStatus(status);
    ZooKeeperClient zkC = new ZooKeeperClient(configCurator, new BaseDeployLogger(), false, Tenants.getSessionsPath(tenant).append(String.valueOf(sessionId)));
    zkC.write(Collections.singletonMap(modelFactory.getVersion(), new MockFileRegistry()));
    zkC.write(AllocatedHosts.withHosts(Collections.emptySet()));
    RemoteSession session = new RemoteSession(TenantName.from("default"), sessionId, componentRegistry, zkClient, clock);
    remoteSessionRepo.addSession(session);
    return session;
}