1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.session;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.provision.AllocatedHosts;
import com.yahoo.config.provision.TenantName;
import com.yahoo.lang.SettableOptional;
import java.util.logging.Level;
import com.yahoo.transaction.Transaction;
import com.yahoo.vespa.config.server.GlobalComponentRegistry;
import com.yahoo.vespa.config.server.ReloadHandler;
import com.yahoo.vespa.config.server.application.ApplicationSet;
import com.yahoo.vespa.config.server.modelfactory.ActivatedModelsBuilder;
import com.yahoo.vespa.config.server.tenant.TenantRepository;
import com.yahoo.vespa.curator.Curator;
import org.apache.zookeeper.KeeperException;
import java.time.Clock;
import java.time.Instant;
import java.util.Optional;
import java.util.logging.Logger;
/**
* A RemoteSession represents a session created on another config server. This session can
* be regarded as read only, and this interface only allows reading information about a session.
*
* @author Ulf Lilleengen
*/
public class RemoteSession extends Session {
private static final Logger log = Logger.getLogger(RemoteSession.class.getName());
private ApplicationSet applicationSet = null;
private final ActivatedModelsBuilder applicationLoader;
private final Clock clock;
/**
* Creates a session. This involves loading the application, validating it and distributing it.
*
* @param tenant The name of the tenant creating session
* @param sessionId The session id for this session.
* @param componentRegistry a registry of global components
* @param zooKeeperClient a SessionZooKeeperClient instance
*/
public RemoteSession(TenantName tenant,
long sessionId,
GlobalComponentRegistry componentRegistry,
SessionZooKeeperClient zooKeeperClient) {
super(tenant, sessionId, zooKeeperClient);
this.applicationLoader = new ActivatedModelsBuilder(tenant, sessionId, zooKeeperClient, componentRegistry);
this.clock = componentRegistry.getClock();
}
void loadPrepared() {
Curator.CompletionWaiter waiter = zooKeeperClient.getPrepareWaiter();
ensureApplicationLoaded();
notifyCompletion(waiter);
}
private ApplicationSet loadApplication() {
ApplicationPackage applicationPackage = zooKeeperClient.loadApplicationPackage();
// Read hosts allocated on the config server instance which created this
Optional<AllocatedHosts> allocatedHosts = applicationPackage.getAllocatedHosts();
return ApplicationSet.fromList(applicationLoader.buildModels(zooKeeperClient.readApplicationId(),
zooKeeperClient.readDockerImageRepository(),
zooKeeperClient.readVespaVersion(),
applicationPackage,
new SettableOptional<>(allocatedHosts),
clock.instant()));
}
public synchronized ApplicationSet ensureApplicationLoaded() {
return applicationSet == null ? applicationSet = loadApplication() : applicationSet;
}
public Session.Status getStatus() {
return zooKeeperClient.readStatus();
}
public synchronized void deactivate() {
applicationSet = null;
}
public Transaction createDeleteTransaction() {
return zooKeeperClient.createWriteStatusTransaction(Status.DELETE);
}
void makeActive(ReloadHandler reloadHandler) {
Curator.CompletionWaiter waiter = zooKeeperClient.getActiveWaiter();
log.log(LogLevel.DEBUG, () -> logPre() + "Getting session from repo: " + getSessionId());
ApplicationSet app = ensureApplicationLoaded();
log.log(LogLevel.DEBUG, () -> logPre() + "Reloading config for " + getSessionId());
reloadHandler.reloadConfig(app);
log.log(LogLevel.DEBUG, () -> logPre() + "Notifying " + waiter);
notifyCompletion(waiter);
log.log(LogLevel.INFO, logPre() + "Session activated: " + getSessionId());
}
@Override
public String logPre() {
if (applicationSet != null) {
return TenantRepository.logPre(applicationSet.getForVersionOrLatest(Optional.empty(), Instant.now()).getId());
}
return TenantRepository.logPre(getTenant());
}
void confirmUpload() {
Curator.CompletionWaiter waiter = zooKeeperClient.getUploadWaiter();
log.log(LogLevel.DEBUG, "Notifying upload waiter for session " + getSessionId());
notifyCompletion(waiter);
log.log(LogLevel.DEBUG, "Done notifying upload for session " + getSessionId());
}
private void notifyCompletion(Curator.CompletionWaiter completionWaiter) {
try {
completionWaiter.notifyCompletion();
} catch (RuntimeException e) {
// Throw only if we get something else than NoNodeException -- NoNodeException might happen when
// the session is no longer in use (e.g. the app using this session has been deleted) and this method
// has not been called yet for the previous session operation
// on a minority of the config servers (see awaitInternal() method in this class)
if (e.getCause().getClass() != KeeperException.NoNodeException.class) {
throw e;
} else {
log.log(LogLevel.INFO, "Not able to notify completion for session: " + getSessionId() + ", node has been deleted");
}
}
}
public void delete() {
Transaction transaction = zooKeeperClient.deleteTransaction();
transaction.commit();
transaction.close();
}
}
|
