diff --git a/src/main/java/org/wikitolearn/wikirating/model/Page.java b/src/main/java/org/wikitolearn/wikirating/model/Page.java index 8d99d34..c312d21 100644 --- a/src/main/java/org/wikitolearn/wikirating/model/Page.java +++ b/src/main/java/org/wikitolearn/wikirating/model/Page.java @@ -1,219 +1,221 @@ /** * */ package org.wikitolearn.wikirating.model; +import java.util.ArrayList; +import java.util.HashSet; import java.util.Set; import org.neo4j.ogm.annotation.GraphId; import org.neo4j.ogm.annotation.Index; import org.neo4j.ogm.annotation.Labels; import org.neo4j.ogm.annotation.NodeEntity; import org.neo4j.ogm.annotation.Relationship; /** * @author aletundo * */ @NodeEntity( label = "Page") public class Page { @GraphId private Long graphId; private int pageid; @Index private String title; @Index private String lang; @Index(unique = true, primary = true) private String langPageId; private double pageRank; @Relationship(type = "LAST_REVISION", direction = Relationship.OUTGOING) private Revision lastRevision; @Relationship(type = "FIRST_REVISION", direction = Relationship.OUTGOING) private Revision fistRevision; @Relationship(type = "LEVEL_TWO", direction = Relationship.UNDIRECTED) private Set levelsTwo; @Relationship(type = "LEVEL_THREE", direction = Relationship.UNDIRECTED) private Set levelsThree; @Labels - private Set labels; + private Set labels = new HashSet<>(); /** * */ public Page() {} /** * @param pageid * @param title * @param lang */ public Page(int pageid, String title, String lang, String langPageId) { this.pageid = pageid; this.title = title; this.lang = lang; this.langPageId = langPageId; } /** * @return the graphId */ public Long getGraphId() { return graphId; } /** * @param graphId the graphId to set */ public void setGraphId(Long graphId) { this.graphId = graphId; } /** * @return the pageid */ public int getPageid() { return pageid; } /** * @param pageid the pageid to set */ public void setPageid(int pageid) { this.pageid = pageid; } /** * @return the title */ public String getTitle() { return title; } /** * @param title the title to set */ public void setTitle(String title) { this.title = title; } /** * @return the lang */ public String getLang() { return lang; } /** * @param lang the lang to set */ public void setLang(String lang) { this.lang = lang; } /** * @return the langPageId */ public String getLangPageId() { return langPageId; } /** * @param langPageId the langPageId to set */ public void setLangPageId(String langPageId) { this.langPageId = langPageId; } /** * @return the pageRank */ public double getPageRank() { return pageRank; } /** * @param pageRank the pageRank to set */ public void setPageRank(double pageRank) { this.pageRank = pageRank; } /** * @return the lastRevision */ public Revision getLastRevision() { return lastRevision; } /** * @param lastRevision the lastRevision to set */ public void setLastRevision(Revision lastRevision) { this.lastRevision = lastRevision; } /** * @return the fistRevision */ public Revision getFistRevision() { return fistRevision; } /** * @param fistRevision the fistRevision to set */ public void setFistRevision(Revision fistRevision) { this.fistRevision = fistRevision; } /** * @return the levelsTwo */ public Set getLevelsTwo() { return levelsTwo; } /** * @param levelsTwo the levelsTwo to set */ public void setLevelsTwo(Set levelsTwo) { this.levelsTwo = levelsTwo; } /** * @return the levelsThree */ public Set getLevelsThree() { return levelsThree; } /** * @param levelsThree the levelsThree to set */ public void setLevelsThree(Set levelsThree) { this.levelsThree = levelsThree; } /** * @return the labels */ public Set getLabels() { return labels; } /** * @param labels the labels to set */ public void setLabels(Set labels) { this.labels = labels; } /** * @param label the label to set */ public void addLabel(String label) { this.labels.add(label); } } \ No newline at end of file