Search in sources :

Example 1 with GlobalComponentRegistry

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

the class ApplicationHandlerTest method addApplication.

private static Tenants addApplication(ApplicationId applicationId, long sessionId) throws Exception {
    // This method is a good illustration of the spaghetti wiring resulting from no design
    // TODO: When this setup looks sane we have refactored sufficiently that there is a design
    TenantName tenantName = applicationId.tenant();
    Path tenantPath = Tenants.getTenantPath(tenantName);
    Path sessionPath = tenantPath.append(Tenant.SESSIONS).append(String.valueOf(sessionId));
    MockCurator curator = new MockCurator();
    GlobalComponentRegistry componentRegistry = new TestComponentRegistry.Builder().curator(curator).modelFactoryRegistry(new ModelFactoryRegistry(Collections.singletonList(new VespaModelFactory(new NullConfigModelRegistry())))).build();
    // Creates the application path element in zk
    Tenants tenants = new Tenants(componentRegistry);
    tenants.addTenant(tenantName);
    Tenant tenant = tenants.getTenant(tenantName);
    tenant.getApplicationRepo().createPutApplicationTransaction(applicationId, sessionId).commit();
    ApplicationPackage app = FilesApplicationPackage.fromFile(testApp);
    SessionZooKeeperClient sessionClient = new SessionZooKeeperClient(curator, sessionPath);
    SessionContext context = new SessionContext(app, sessionClient, new File("/serverDb"), tenant.getApplicationRepo(), null, new SuperModelGenerationCounter(curator));
    tenant.getLocalSessionRepo().addSession(new LocalSession(tenantName, sessionId, null, context));
    // TODO: Instead, use ApplicationRepository to deploy the application
    sessionClient.writeApplicationId(applicationId);
    tenant.getRemoteSessionRepo().addSession(new RemoteSession(tenantName, sessionId, componentRegistry, sessionClient, Clock.systemUTC()));
    return tenants;
}
Also used : Path(com.yahoo.path.Path) TenantName(com.yahoo.config.provision.TenantName) ModelFactoryRegistry(com.yahoo.vespa.config.server.modelfactory.ModelFactoryRegistry) LocalSession(com.yahoo.vespa.config.server.session.LocalSession) Tenants(com.yahoo.vespa.config.server.tenant.Tenants) GlobalComponentRegistry(com.yahoo.vespa.config.server.GlobalComponentRegistry) FilesApplicationPackage(com.yahoo.config.model.application.provider.FilesApplicationPackage) ApplicationPackage(com.yahoo.config.application.api.ApplicationPackage) VespaModelFactory(com.yahoo.vespa.model.VespaModelFactory) NullConfigModelRegistry(com.yahoo.config.model.NullConfigModelRegistry) Tenant(com.yahoo.vespa.config.server.tenant.Tenant) TestComponentRegistry(com.yahoo.vespa.config.server.TestComponentRegistry) RemoteSession(com.yahoo.vespa.config.server.session.RemoteSession) SessionContext(com.yahoo.vespa.config.server.session.SessionContext) SuperModelGenerationCounter(com.yahoo.vespa.config.server.SuperModelGenerationCounter) File(java.io.File) MockCurator(com.yahoo.vespa.curator.mock.MockCurator) SessionZooKeeperClient(com.yahoo.vespa.config.server.session.SessionZooKeeperClient)

Example 2 with GlobalComponentRegistry

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

the class LocalSessionRepoTest method setupSessions.

private void setupSessions(TenantName tenantName, boolean createInitialSessions) throws Exception {
    GlobalComponentRegistry globalComponentRegistry = new TestComponentRegistry.Builder().curator(curator).build();
    TenantFileSystemDirs tenantFileSystemDirs = TenantFileSystemDirs.createTestDirs(tenantName);
    if (createInitialSessions) {
        IOUtils.copyDirectory(testApp, new File(tenantFileSystemDirs.sessionsPath(), "1"));
        IOUtils.copyDirectory(testApp, new File(tenantFileSystemDirs.sessionsPath(), "2"));
        IOUtils.copyDirectory(testApp, new File(tenantFileSystemDirs.sessionsPath(), "3"));
    }
    clock = new ManualClock(Instant.ofEpochSecond(1));
    LocalSessionLoader loader = new SessionFactoryImpl(globalComponentRegistry, new SessionCounter(globalComponentRegistry.getConfigCurator(), tenantName), new MemoryTenantApplications(), tenantFileSystemDirs, new HostRegistry<>(), tenantName);
    repo = new LocalSessionRepo(tenantFileSystemDirs, loader, clock, 5);
}
Also used : ManualClock(com.yahoo.test.ManualClock) MemoryTenantApplications(com.yahoo.vespa.config.server.application.MemoryTenantApplications) SessionCounter(com.yahoo.vespa.config.server.zookeeper.SessionCounter) TenantFileSystemDirs(com.yahoo.vespa.config.server.deploy.TenantFileSystemDirs) GlobalComponentRegistry(com.yahoo.vespa.config.server.GlobalComponentRegistry) File(java.io.File)

Aggregations

GlobalComponentRegistry (com.yahoo.vespa.config.server.GlobalComponentRegistry)2 File (java.io.File)2 ApplicationPackage (com.yahoo.config.application.api.ApplicationPackage)1 NullConfigModelRegistry (com.yahoo.config.model.NullConfigModelRegistry)1 FilesApplicationPackage (com.yahoo.config.model.application.provider.FilesApplicationPackage)1 TenantName (com.yahoo.config.provision.TenantName)1 Path (com.yahoo.path.Path)1 ManualClock (com.yahoo.test.ManualClock)1 SuperModelGenerationCounter (com.yahoo.vespa.config.server.SuperModelGenerationCounter)1 TestComponentRegistry (com.yahoo.vespa.config.server.TestComponentRegistry)1 MemoryTenantApplications (com.yahoo.vespa.config.server.application.MemoryTenantApplications)1 TenantFileSystemDirs (com.yahoo.vespa.config.server.deploy.TenantFileSystemDirs)1 ModelFactoryRegistry (com.yahoo.vespa.config.server.modelfactory.ModelFactoryRegistry)1 LocalSession (com.yahoo.vespa.config.server.session.LocalSession)1 RemoteSession (com.yahoo.vespa.config.server.session.RemoteSession)1 SessionContext (com.yahoo.vespa.config.server.session.SessionContext)1 SessionZooKeeperClient (com.yahoo.vespa.config.server.session.SessionZooKeeperClient)1 Tenant (com.yahoo.vespa.config.server.tenant.Tenant)1 Tenants (com.yahoo.vespa.config.server.tenant.Tenants)1 SessionCounter (com.yahoo.vespa.config.server.zookeeper.SessionCounter)1