diff options
author | Lester Solbakken <lesters@yahoo-inc.com> | 2017-05-09 13:59:16 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@yahoo-inc.com> | 2017-05-09 13:59:16 +0200 |
commit | fd7d16d2ea68a38be38783decdfbf1097db3741c (patch) | |
tree | bef755c29e24f557b7f4569bfe73da1eb7a2ea8a /vespa-documentgen-plugin | |
parent | 0e638682e7e319df2f0650033b484f85bfd3a7e0 (diff) |
Add tensor support to vespa-documentgen-plugin
Diffstat (limited to 'vespa-documentgen-plugin')
-rw-r--r-- | vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java b/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java index 67627ca9410..fff93ec4525 100644 --- a/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java +++ b/vespa-documentgen-plugin/src/main/java/com/yahoo/vespa/DocumentGenMojo.java @@ -10,6 +10,7 @@ import com.yahoo.searchdefinition.Search; import com.yahoo.searchdefinition.SearchBuilder; import com.yahoo.searchdefinition.UnprocessingSearchBuilder; import com.yahoo.searchdefinition.parser.ParseException; +import com.yahoo.tensor.TensorType; import org.apache.maven.plugin.AbstractMojo; import org.apache.maven.plugin.MojoExecutionException; import org.apache.maven.plugin.MojoFailureException; @@ -841,6 +842,9 @@ public class DocumentGenMojo extends AbstractMojo { if (dt instanceof ReferenceDataType) { return "com.yahoo.document.DocumentId"; } + if (dt instanceof TensorDataType) { + return "com.yahoo.tensor.Tensor"; + } return "byte[]"; } @@ -871,7 +875,11 @@ public class DocumentGenMojo extends AbstractMojo { return String.format("new com.yahoo.document.ReferenceDataType(%s.type, %d)", className(((ReferenceDataType) dt).getTargetType().getName()), dt.getId()); } - return "DataType.RAW"; + if (dt instanceof TensorDataType) { + return String.format("new com.yahoo.document.TensorDataType(com.yahoo.tensor.TensorType.fromSpec(\"%s\"))", + ((TensorDataType)dt).getTensorType().toString()); + } + return "com.yahoo.document.DataType.RAW"; } @Override |