-
Notifications
You must be signed in to change notification settings - Fork 88
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test: Support query params in test proxy
Change-Id: I53122fdf0301fb41d4f24881dc8de65d174dee35
- Loading branch information
1 parent
49eec0e
commit ec94ea6
Showing
4 changed files
with
208 additions
and
18 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
167 changes: 167 additions & 0 deletions
167
test-proxy/src/main/java/com/google/cloud/bigtable/testproxy/StatementDeserializer.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,167 @@ | ||
| /* | ||
| * Copyright 2024 Google LLC | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); | ||
| * you may not use this file except in compliance with the License. | ||
| * You may obtain a copy of the License at | ||
| * | ||
| * https://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
| package com.google.cloud.bigtable.testproxy; | ||
|
|
||
| import com.google.bigtable.v2.Value; | ||
| import com.google.bigtable.v2.Value.KindCase; | ||
| import com.google.cloud.Date; | ||
| import com.google.cloud.bigtable.data.v2.models.sql.SqlType; | ||
| import com.google.cloud.bigtable.data.v2.models.sql.Statement; | ||
| import com.google.protobuf.Timestamp; | ||
| import java.util.ArrayList; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import org.threeten.bp.Instant; | ||
|
|
||
| public class StatementDeserializer { | ||
|
|
||
| static Statement toStatement(ExecuteQueryRequest request) { | ||
| Statement.Builder statementBuilder = Statement.newBuilder(request.getRequest().getQuery()); | ||
| for (Map.Entry<String, Value> paramEntry : request.getRequest().getParamsMap().entrySet()) { | ||
| String name = paramEntry.getKey(); | ||
| Value value = paramEntry.getValue(); | ||
| switch (value.getType().getKindCase()) { | ||
| case BYTES_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setBytesParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.BYTES_VALUE)) { | ||
| statementBuilder.setBytesParam(name, value.getBytesValue()); | ||
| } else { | ||
| throw new IllegalArgumentException("Unexpected bytes value: " + value); | ||
| } | ||
| break; | ||
| case STRING_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setStringParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.STRING_VALUE)) { | ||
| statementBuilder.setStringParam(name, value.getStringValue()); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed string value: " + value); | ||
| } | ||
| break; | ||
| case INT64_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setLongParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.INT_VALUE)) { | ||
| statementBuilder.setLongParam(name, value.getIntValue()); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed int64 value: " + value); | ||
| } | ||
| break; | ||
| case FLOAT32_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setFloatParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.FLOAT_VALUE)) { | ||
| statementBuilder.setFloatParam(name, (float) value.getFloatValue()); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed float32 value: " + value); | ||
| } | ||
| break; | ||
| case FLOAT64_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setDoubleParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.FLOAT_VALUE)) { | ||
| statementBuilder.setDoubleParam(name, value.getFloatValue()); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed float64 value: " + value); | ||
| } | ||
| break; | ||
| case BOOL_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setBooleanParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.BOOL_VALUE)) { | ||
| statementBuilder.setBooleanParam(name, value.getBoolValue()); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed boolean value: " + value); | ||
| } | ||
| break; | ||
| case TIMESTAMP_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setTimestampParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.TIMESTAMP_VALUE)) { | ||
| Timestamp ts = value.getTimestampValue(); | ||
| statementBuilder.setTimestampParam(name, toInstant(ts)); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed timestamp value: " + value); | ||
| } | ||
| break; | ||
| case DATE_TYPE: | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setDateParam(name, null); | ||
| } else if (value.getKindCase().equals(KindCase.DATE_VALUE)) { | ||
| com.google.type.Date protoDate = value.getDateValue(); | ||
| statementBuilder.setDateParam(name, fromProto(protoDate)); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed boolean value: " + value); | ||
| } | ||
| break; | ||
| case ARRAY_TYPE: | ||
| SqlType.Array sqlType = (SqlType.Array) SqlType.fromProto(value.getType()); | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| statementBuilder.setListParam(name, null, sqlType); | ||
| } else if (value.getKindCase().equals(KindCase.ARRAY_VALUE)) { | ||
| List<Object> array = new ArrayList<>(); | ||
| for (Value elem : value.getArrayValue().getValuesList()) { | ||
| array.add(decodeArrayElement(elem, sqlType.getElementType())); | ||
| } | ||
| statementBuilder.setListParam(name, array, sqlType); | ||
| } else { | ||
| throw new IllegalArgumentException("Malformed array value: " + value); | ||
| } | ||
| break; | ||
| default: | ||
| throw new IllegalArgumentException("Unexpected query param type in param: " + value); | ||
| } | ||
| } | ||
| return statementBuilder.build(); | ||
| } | ||
|
|
||
| static Object decodeArrayElement(Value value, SqlType<?> elemType) { | ||
| if (value.getKindCase().equals(KindCase.KIND_NOT_SET)) { | ||
| return null; | ||
| } | ||
| switch (elemType.getCode()) { | ||
| case BYTES: | ||
| return value.getBytesValue(); | ||
| case STRING: | ||
| return value.getStringValue(); | ||
| case INT64: | ||
| return value.getIntValue(); | ||
| case FLOAT64: | ||
| return value.getFloatValue(); | ||
| case FLOAT32: | ||
| // cast to float so we produce List<Float>, etc | ||
| return (float) value.getFloatValue(); | ||
| case BOOL: | ||
| return value.getBoolValue(); | ||
| case TIMESTAMP: | ||
| return toInstant(value.getTimestampValue()); | ||
| case DATE: | ||
| return fromProto(value.getDateValue()); | ||
| default: | ||
| // We should have already thrown an exception in the SqlRowMerger | ||
| throw new IllegalStateException("Unsupported array query param element type: " + elemType); | ||
| } | ||
| } | ||
|
|
||
| private static Instant toInstant(Timestamp timestamp) { | ||
| return Instant.ofEpochSecond(timestamp.getSeconds(), timestamp.getNanos()); | ||
| } | ||
|
|
||
| private static Date fromProto(com.google.type.Date proto) { | ||
| return Date.fromYearMonthDay(proto.getYear(), proto.getMonth(), proto.getDay()); | ||
| } | ||
| } |
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