summaryrefslogtreecommitdiffstats
path: root/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/persistence/VersionStatusSerializer.java
blob: e0bc592c82c71c9b562ccba167c0979a0c1a391e (plain) (blame)
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
package com.yahoo.vespa.hosted.controller.persistence;

import com.yahoo.component.Version;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.slime.ArrayTraverser;
import com.yahoo.slime.Cursor;
import com.yahoo.slime.Inspector;
import com.yahoo.slime.Slime;
import com.yahoo.vespa.hosted.controller.versions.DeploymentStatistics;
import com.yahoo.vespa.hosted.controller.versions.VersionStatus;
import com.yahoo.vespa.hosted.controller.versions.VespaVersion;

import java.time.Instant;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

/**
 * Serializes VersionStatus to and from slime
 *
 * @author mpolden
 */
public class VersionStatusSerializer {

    // VersionStatus fields
    private static final String versionsField = "versions";

    // VespaVersion fields
    private static final String releaseCommitField = "releaseCommit";
    private static final String releasedAtField = "releasedAt";
    private static final String isCurrentSystemVersionField = "isCurrentSystemVersion";
    private static final String deploymentStatisticsField = "deploymentStatistics";
    private static final String confidenceField = "confidence";
    private static final String configServersField = "configServerHostnames";

    // DeploymentStatistics fields
    private static final String versionField = "version";
    private static final String failingField = "failing";
    private static final String productionField = "production";

    public Slime toSlime(VersionStatus status) {
        Slime slime = new Slime();
        Cursor root = slime.setObject();
        versionsToSlime(status.versions(), root.setArray(versionsField));
        return slime;
    }

    public VersionStatus fromSlime(Slime slime) {
        Inspector root = slime.get();
        return new VersionStatus(vespaVersionsFromSlime(root.field(versionsField)));
    }

    private void versionsToSlime(List<VespaVersion> versions, Cursor array) {
        versions.forEach(version -> vespaVersionToSlime(version, array.addObject()));
    }

    private void vespaVersionToSlime(VespaVersion version, Cursor object) {
        object.setString(releaseCommitField, version.releaseCommit());
        object.setLong(releasedAtField, version.releasedAt().toEpochMilli());
        object.setBool(isCurrentSystemVersionField, version.isCurrentSystemVersion());
        deploymentStatisticsToSlime(version.statistics(), object.setObject(deploymentStatisticsField));
        object.setString(confidenceField, version.confidence().name());
        configServersToSlime(version.configServerHostnames(), object.setArray(configServersField));
    }

    private void configServersToSlime(Set<String> configServerHostnames, Cursor array) {
        configServerHostnames.forEach(array::addString);
    }

    private void deploymentStatisticsToSlime(DeploymentStatistics statistics, Cursor object) {
        object.setString(versionField, statistics.version().toString());
        applicationsToSlime(statistics.failing(), object.setArray(failingField));
        applicationsToSlime(statistics.production(), object.setArray(productionField));
    }

    private void applicationsToSlime(List<ApplicationId> applications, Cursor array) {
        applications.forEach(application -> array.addString(application.serializedForm()));
    }

    private List<VespaVersion> vespaVersionsFromSlime(Inspector array) {
        List<VespaVersion> versions = new ArrayList<>();
        array.traverse((ArrayTraverser) (i, object) -> versions.add(vespaVersionFromSlime(object)));
        return Collections.unmodifiableList(versions);
    }

    private VespaVersion vespaVersionFromSlime(Inspector object) {
        return new VespaVersion(deploymentStatisticsFromSlime(object.field(deploymentStatisticsField)),
                                object.field(releaseCommitField).asString(),
                                Instant.ofEpochMilli(object.field(releasedAtField).asLong()),
                                object.field(isCurrentSystemVersionField).asBool(),
                                configServersFromSlime(object.field(configServersField)),
                                VespaVersion.Confidence.valueOf(object.field(confidenceField).asString())
        );
    }

    private Set<String> configServersFromSlime(Inspector array) {
        Set<String> configServerHostnames = new LinkedHashSet<>();
        array.traverse((ArrayTraverser) (i, entry) -> configServerHostnames.add(entry.asString()));
        return Collections.unmodifiableSet(configServerHostnames);
    }

    private DeploymentStatistics deploymentStatisticsFromSlime(Inspector object) {
        return new DeploymentStatistics(Version.fromString(object.field(versionField).asString()),
                                        applicationsFromSlime(object.field(failingField)),
                                        applicationsFromSlime(object.field(productionField)));
    }

    private List<ApplicationId> applicationsFromSlime(Inspector array) {
        List<ApplicationId> applications = new ArrayList<>();
        array.traverse((ArrayTraverser) (i, entry) -> applications.add(
                ApplicationId.fromSerializedForm(entry.asString()))
        );
        return Collections.unmodifiableList(applications);
    }

}