-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
373 additions
and
32 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
155 changes: 155 additions & 0 deletions
155
...java/org/polypheny/simpleclient/scenario/knnbench/queryBuilder/SimpleKnnIdIntFeature.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,155 @@ | ||
| package org.polypheny.simpleclient.scenario.knnbench.queryBuilder; | ||
|
|
||
|
|
||
| import java.util.Arrays; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
| import java.util.Random; | ||
| import kong.unirest.HttpRequest; | ||
| import org.apache.commons.lang3.tuple.ImmutablePair; | ||
| import org.polypheny.simpleclient.query.CottontailQuery; | ||
| import org.polypheny.simpleclient.query.CottontailQuery.QueryType; | ||
| import org.polypheny.simpleclient.query.Query; | ||
| import org.polypheny.simpleclient.query.QueryBuilder; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.Entity; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.From; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.IntVector; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.Knn; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.Knn.Distance; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.Projection; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.Schema; | ||
| import org.vitrivr.cottontail.grpc.CottontailGrpc.Vector; | ||
|
|
||
|
|
||
| public class SimpleKnnIdIntFeature extends QueryBuilder { | ||
|
|
||
| private static final boolean EXPECT_RESULT = true; | ||
|
|
||
| private final long randomSeed; | ||
| private final int dimension; | ||
| private final int limit; | ||
| private final String norm; | ||
|
|
||
| private final Random random; | ||
|
|
||
|
|
||
| public SimpleKnnIdIntFeature( long randomSeed, int dimension, int limit, String norm ) { | ||
| this.randomSeed = randomSeed; | ||
| this.dimension = dimension; | ||
|
|
||
| this.random = new Random( randomSeed ); | ||
| this.limit = limit; | ||
| this.norm = norm; | ||
| } | ||
|
|
||
|
|
||
| private Integer[] getRandomVector() { | ||
| Integer[] integers = new Integer[this.dimension]; | ||
| for ( int i = 0; i < this.dimension; i++ ) { | ||
| integers[i] = random.nextInt( 500 ); | ||
| } | ||
|
|
||
| return integers; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public synchronized Query getNewQuery() { | ||
| return new SimpleKnnIdIntFeatureQuery( | ||
| getRandomVector(), | ||
| limit, | ||
| norm | ||
| ); | ||
| } | ||
|
|
||
|
|
||
| private static class SimpleKnnIdIntFeatureQuery extends Query { | ||
|
|
||
| private static final String SQL_1 = "SELECT id FROM knn_intfeature ORDER BY distance(feature, "; | ||
| private static final String SQL_2 = ", "; | ||
| private static final String SQL_3 = ") ASC LIMIT "; | ||
|
|
||
| private final Integer[] target; | ||
| private final int limit; | ||
| private final String norm; | ||
|
|
||
|
|
||
| public SimpleKnnIdIntFeatureQuery( Integer[] target, int limit, String norm ) { | ||
| super( EXPECT_RESULT ); | ||
| this.target = target; | ||
| this.limit = limit; | ||
| this.norm = norm; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public String getSql() { | ||
| return SQL_1 + "ARRAY" + Arrays.toString( target ) + SQL_2 + "'" + norm + "'" + SQL_3 + limit; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public String getParameterizedSqlQuery() { | ||
| return SQL_1 + "?" + SQL_2 + "'" + norm + "'" + SQL_3 + limit; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public Map<Integer, ImmutablePair<DataTypes, Object>> getParameterValues() { | ||
| Map<Integer, ImmutablePair<DataTypes, Object>> map = new HashMap<>(); | ||
| map.put( 1, new ImmutablePair<>( DataTypes.ARRAY_INT, target ) ); | ||
| return map; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public HttpRequest<?> getRest() { | ||
| return null; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public CottontailQuery getCottontail() { | ||
| Map<String, String> projection = new HashMap<>(); | ||
| projection.put( "id", "id" ); | ||
| CottontailGrpc.Query query = CottontailGrpc.Query.newBuilder() | ||
| .setFrom( From.newBuilder().setEntity( Entity.newBuilder().setSchema( Schema.newBuilder().setName( "public" ).build() ).setName( "knn_intfeature" ).build() ) ) | ||
| .setLimit( limit ) | ||
| .setKnn( Knn.newBuilder() | ||
| .setAttribute( "feature" ) | ||
| .setK( limit ) | ||
| .addQuery( Vector.newBuilder().setIntVector( IntVector.newBuilder().addAllVector( Arrays.asList( target ) ).build() ).build() ) | ||
| .setDistance( getDistance( norm ) ) | ||
| .build() ) | ||
| .setProjection( Projection.newBuilder().putAllAttributes( projection ).build() ) | ||
| .build(); | ||
| return new CottontailQuery( | ||
| QueryType.QUERY, | ||
| query | ||
| ); | ||
| } | ||
|
|
||
|
|
||
| private static Distance getDistance( String norm ) { | ||
| if ( "L2".equalsIgnoreCase( norm ) ) { | ||
| return Distance.L2; | ||
| } | ||
| if ( "L1".equalsIgnoreCase( norm ) ) { | ||
| return Distance.L1; | ||
| } | ||
| if ( "L2SQUARED".equalsIgnoreCase( norm ) ) { | ||
| return Distance.L2SQUARED; | ||
| } | ||
| if ( "CHISQUARED".equalsIgnoreCase( norm ) ) { | ||
| return Distance.CHISQUARED; | ||
| } | ||
| if ( "COSINE".equalsIgnoreCase( norm ) ) { | ||
| return Distance.COSINE; | ||
| } | ||
|
|
||
| throw new RuntimeException( "Unsupported norm: " + norm ); | ||
| } | ||
| } | ||
|
|
||
| } |
Oops, something went wrong.