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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.schema.processing;
import com.yahoo.config.application.api.DeployLogger;
import com.yahoo.schema.RankProfile;
import com.yahoo.schema.RankProfileRegistry;
import com.yahoo.schema.Schema;
import com.yahoo.schema.document.Attribute;
import com.yahoo.vespa.model.container.search.QueryProfiles;
/**
* @author baldersheim
*/
public class DiversitySettingsValidator extends Processor {
public DiversitySettingsValidator(Schema schema, DeployLogger deployLogger, RankProfileRegistry rankProfileRegistry, QueryProfiles queryProfiles) {
super(schema, deployLogger, rankProfileRegistry, queryProfiles);
}
@Override
public void process(boolean validate, boolean documentsOnly) {
if ( ! validate) return;
if (documentsOnly) return;
for (RankProfile rankProfile : rankProfileRegistry.rankProfilesOf(schema)) {
if (rankProfile.getMatchPhaseSettings() != null && rankProfile.getMatchPhaseSettings().getDiversity() != null) {
validate(rankProfile, rankProfile.getMatchPhaseSettings().getDiversity());
}
}
}
private void validate(RankProfile rankProfile, RankProfile.DiversitySettings settings) {
String attributeName = settings.getAttribute();
new AttributeValidator(schema.getName(), rankProfile.name(),
schema.getAttribute(attributeName), attributeName).validate();
}
private static class AttributeValidator extends MatchPhaseSettingsValidator.AttributeValidator {
public AttributeValidator(String searchName, String rankProfileName, Attribute attribute, String attributeName) {
super(searchName, rankProfileName, attribute, attributeName);
}
protected void validateThatAttributeIsSingleAndNotPredicate() {
if ( ! attribute.getCollectionType().equals(Attribute.CollectionType.SINGLE) ||
attribute.getType().equals(Attribute.Type.PREDICATE))
{
failValidation("must be single value numeric, or enumerated attribute, but it is '"
+ attribute.getDataType().getName() + "'");
}
}
@Override
public void validate() {
validateThatAttributeExists();
validateThatAttributeIsSingleAndNotPredicate();
}
@Override
public String getValidationType() {
return "diversity";
}
}
}
|
