Skip to content

Commit a0d2898

Browse files
authored
Merge pull request #58 from ldbc/fix_merge_foreign
Fixed merge foreign to properly merge tag's class foreign key
2 parents fcc3eac + db1d52d commit a0d2898

File tree

1 file changed

+5
-11
lines changed

1 file changed

+5
-11
lines changed

src/main/java/ldbc/snb/datagen/serializer/snb/interactive/CSVMergeForeignInvariantSerializer.java

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,6 @@ private enum FileNames {
6060
TAG("tag"),
6161
TAG_HAS_TYPE_TAGCLASS("tag_hasType_tagclass"),
6262
TAGCLASS("tagclass"),
63-
TAGCLASS_IS_SUBCLASS_OF_TAGCLASS("tagclass_isSubclassOf_tagclass"),
6463
PLACE("place"),
6564
ORGANIZATION("organisation");
6665

@@ -101,13 +100,9 @@ public void initialize(Configuration conf, int reducerId) throws IOException {
101100
arguments.add("id");
102101
arguments.add("name");
103102
arguments.add("url");
103+
arguments.add("isSubclassOf");
104104
writers[FileNames.TAGCLASS.ordinal()].writeHeader(arguments);
105105

106-
arguments.clear();
107-
arguments.add("TagClass.id");
108-
arguments.add("TagClass.id");
109-
writers[FileNames.TAGCLASS_IS_SUBCLASS_OF_TAGCLASS.ordinal()].writeHeader(arguments);
110-
111106
arguments.clear();
112107
arguments.add("id");
113108
arguments.add("name");
@@ -171,14 +166,13 @@ protected void serialize(final TagClass tagClass) {
171166
} else {
172167
arguments.add(DBPOWL.getUrl(tagClass.name));
173168
}
174-
writers[FileNames.TAGCLASS.ordinal()].writeEntry(arguments);
175-
176169
if (tagClass.parent != -1) {
177-
arguments.clear();
178-
arguments.add(Integer.toString(tagClass.id));
179170
arguments.add(Integer.toString(tagClass.parent));
180-
writers[FileNames.TAGCLASS_IS_SUBCLASS_OF_TAGCLASS.ordinal()].writeEntry(arguments);
171+
} else {
172+
arguments.add("");
181173
}
174+
writers[FileNames.TAGCLASS.ordinal()].writeEntry(arguments);
175+
182176
}
183177

184178
protected void serialize(final Tag tag) {

0 commit comments

Comments
 (0)