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
|
package com.yahoo.vespa.config.server.session;
import com.yahoo.component.Version;
import com.yahoo.config.FileReference;
import com.yahoo.config.model.api.Quota;
import com.yahoo.config.model.api.TenantSecretStore;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.AthenzDomain;
import com.yahoo.config.provision.CloudAccount;
import com.yahoo.config.provision.DataplaneToken;
import com.yahoo.config.provision.DockerImage;
import com.yahoo.slime.Cursor;
import com.yahoo.slime.Slime;
import com.yahoo.slime.SlimeUtils;
import com.yahoo.vespa.config.server.tenant.DataplaneTokenSerializer;
import com.yahoo.vespa.config.server.tenant.OperatorCertificateSerializer;
import com.yahoo.vespa.config.server.tenant.TenantSecretStoreSerializer;
import java.io.IOException;
import java.security.cert.X509Certificate;
import java.util.List;
import java.util.Optional;
/**
* Data class for session information, typically parameters supplied in a deployment request that needs
* to be persisted in ZooKeeper. These will be used when creating a new session based on an existing one.
*
* @author hmusum
*/
public record SessionData(ApplicationId applicationId,
FileReference applicationPackageReference,
Version version,
Optional<DockerImage> dockerImageRepository,
Optional<AthenzDomain> athenzDomain,
Optional<Quota> quota,
List<TenantSecretStore> tenantSecretStores,
List<X509Certificate> operatorCertificates,
Optional<CloudAccount> cloudAccount,
List<DataplaneToken> dataplaneTokens) {
// NOTE: Any state added here MUST also be propagated in com.yahoo.vespa.config.server.deploy.Deployment.prepare()
static final String APPLICATION_ID_PATH = "applicationId";
static final String APPLICATION_PACKAGE_REFERENCE_PATH = "applicationPackageReference";
static final String VERSION_PATH = "version";
static final String CREATE_TIME_PATH = "createTime";
static final String DOCKER_IMAGE_REPOSITORY_PATH = "dockerImageRepository";
static final String ATHENZ_DOMAIN = "athenzDomain";
static final String QUOTA_PATH = "quota";
static final String TENANT_SECRET_STORES_PATH = "tenantSecretStores";
static final String OPERATOR_CERTIFICATES_PATH = "operatorCertificates";
static final String CLOUD_ACCOUNT_PATH = "cloudAccount";
static final String DATAPLANE_TOKENS_PATH = "dataplaneTokens";
static final String SESSION_DATA_PATH = "sessionData";
public byte[] toJson() {
try {
Slime slime = new Slime();
toSlime(slime.setObject());
return SlimeUtils.toJsonBytes(slime);
}
catch (IOException e) {
throw new RuntimeException("Serialization of session data to json failed", e);
}
}
private void toSlime(Cursor object) {
object.setString(APPLICATION_ID_PATH, applicationId.serializedForm());
object.setString(APPLICATION_PACKAGE_REFERENCE_PATH, applicationPackageReference.value());
object.setString(VERSION_PATH, version.toString());
object.setLong(CREATE_TIME_PATH, System.currentTimeMillis());
dockerImageRepository.ifPresent(image -> object.setString(DOCKER_IMAGE_REPOSITORY_PATH, image.asString()));
athenzDomain.ifPresent(domain -> object.setString(ATHENZ_DOMAIN, domain.value()));
quota.ifPresent(q -> object.setString(QUOTA_PATH, q.toString()));
Cursor tenantSecretStoresArray = object.setArray(TENANT_SECRET_STORES_PATH);
TenantSecretStoreSerializer.toSlime(tenantSecretStores, tenantSecretStoresArray);
Cursor operatorCertificatesArray = object.setArray(OPERATOR_CERTIFICATES_PATH);
OperatorCertificateSerializer.toSlime(operatorCertificates, operatorCertificatesArray);
cloudAccount.ifPresent(account -> object.setString(CLOUD_ACCOUNT_PATH, account.value()));
Cursor dataplaneTokensArray = object.setArray(DATAPLANE_TOKENS_PATH);
DataplaneTokenSerializer.toSlime(dataplaneTokens, dataplaneTokensArray);
}
}
|