2021-04-18 15:41:13 -04:00
// Copyright (c) 2015-2021 MinIO, Inc.
//
// This file is part of MinIO Object Storage stack
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
2018-08-15 06:30:19 -04:00
package s3select
import (
2018-11-14 18:55:10 -05:00
"bytes"
2020-09-08 18:56:11 -04:00
"encoding/xml"
2019-09-13 17:18:35 -04:00
"fmt"
2019-01-08 19:53:04 -05:00
"io"
"io/ioutil"
"net/http"
"os"
"reflect"
2019-10-06 10:52:45 -04:00
"strings"
2018-08-15 06:30:19 -04:00
"testing"
2019-10-06 10:52:45 -04:00
2021-03-10 12:41:29 -05:00
"github.com/klauspost/cpuid/v2"
2020-07-14 12:38:05 -04:00
"github.com/minio/minio-go/v7"
2020-02-13 17:03:52 -05:00
"github.com/minio/simdjson-go"
2018-08-15 06:30:19 -04:00
)
2019-01-08 19:53:04 -05:00
type testResponseWriter struct {
statusCode int
response [ ] byte
2018-08-15 06:30:19 -04:00
}
2019-01-08 19:53:04 -05:00
func ( w * testResponseWriter ) Header ( ) http . Header {
return nil
2018-08-15 06:30:19 -04:00
}
2019-01-08 19:53:04 -05:00
func ( w * testResponseWriter ) Write ( p [ ] byte ) ( int , error ) {
w . response = append ( w . response , p ... )
return len ( p ) , nil
2018-08-15 06:30:19 -04:00
}
2019-01-08 19:53:04 -05:00
func ( w * testResponseWriter ) WriteHeader ( statusCode int ) {
w . statusCode = statusCode
2018-08-15 06:30:19 -04:00
}
2019-01-08 19:53:04 -05:00
func ( w * testResponseWriter ) Flush ( ) {
}
2018-10-22 15:12:22 -04:00
2019-10-06 10:52:45 -04:00
func TestJSONQueries ( t * testing . T ) {
input := ` { "id" : 0 , "title" : "Test Record" , "desc" : "Some text" , "synonyms" : [ "foo" , "bar" , "whatever" ] }
{ "id" : 1 , "title" : "Second Record" , "desc" : "another text" , "synonyms" : [ "some" , "synonym" , "value" ] }
{ "id" : 2 , "title" : "Second Record" , "desc" : "another text" , "numbers" : [ 2 , 3.0 , 4 ] }
{ "id" : 3 , "title" : "Second Record" , "desc" : "another text" , "nested" : [ [ 2 , 3.0 , 4 ] , [ 7 , 8.5 , 9 ] ] } `
2020-02-13 17:03:52 -05:00
2019-10-06 10:52:45 -04:00
var testTable = [ ] struct {
name string
query string
2019-11-09 12:10:35 -05:00
requestXML [ ] byte // override request XML
2019-10-06 10:52:45 -04:00
wantResult string
2019-12-06 10:24:41 -05:00
withJSON string // Override JSON input
2019-10-06 10:52:45 -04:00
} {
{
name : "select-in-array-full" ,
query : ` SELECT * from s3object s WHERE 'bar' IN s.synonyms[*] ` ,
wantResult : ` { "id":0,"title":"Test Record","desc":"Some text","synonyms":["foo","bar","whatever"]} ` ,
} ,
{
name : "simple-in-array" ,
query : ` SELECT * from s3object s WHERE s.id IN (1,3) ` ,
wantResult : ` { "id" : 1 , "title" : "Second Record" , "desc" : "another text" , "synonyms" : [ "some" , "synonym" , "value" ] }
{ "id" : 3 , "title" : "Second Record" , "desc" : "another text" , "nested" : [ [ 2 , 3 , 4 ] , [ 7 , 8.5 , 9 ] ] } ` ,
} ,
{
name : "select-in-array-single" ,
query : ` SELECT synonyms from s3object s WHERE 'bar' IN s.synonyms[*] ` ,
wantResult : ` { "synonyms":["foo","bar","whatever"]} ` ,
} ,
{
name : "donatello-1" ,
query : ` SELECT * from s3object s WHERE 'bar' in s.synonyms ` ,
wantResult : ` { "id":0,"title":"Test Record","desc":"Some text","synonyms":["foo","bar","whatever"]} ` ,
} ,
{
name : "donatello-2" ,
query : ` SELECT * from s3object s WHERE 'bar' in s.synonyms[*] ` ,
wantResult : ` { "id":0,"title":"Test Record","desc":"Some text","synonyms":["foo","bar","whatever"]} ` ,
} ,
2020-09-08 18:56:11 -04:00
{
name : "bignum-1" ,
query : ` SELECT id from s3object s WHERE s.id <= 9223372036854775807 ` ,
wantResult : ` { "id" : 0 }
{ "id" : 1 }
{ "id" : 2 }
{ "id" : 3 } ` } ,
{
name : "bignum-2" ,
query : ` SELECT id from s3object s WHERE s.id >= -9223372036854775808 ` ,
wantResult : ` { "id" : 0 }
{ "id" : 1 }
{ "id" : 2 }
{ "id" : 3 } ` } ,
2019-10-06 10:52:45 -04:00
{
name : "donatello-3" ,
query : ` SELECT * from s3object s WHERE 'value' IN s.synonyms[*] ` ,
wantResult : ` { "id":1,"title":"Second Record","desc":"another text","synonyms":["some","synonym","value"]} ` ,
} ,
{
name : "select-in-number" ,
query : ` SELECT * from s3object s WHERE 4 in s.numbers[*] ` ,
wantResult : ` { "id":2,"title":"Second Record","desc":"another text","numbers":[2,3,4]} ` ,
} ,
{
name : "select-in-number-float" ,
query : ` SELECT * from s3object s WHERE 3 in s.numbers[*] ` ,
wantResult : ` { "id":2,"title":"Second Record","desc":"another text","numbers":[2,3,4]} ` ,
} ,
{
name : "select-in-number-float-in-sql" ,
query : ` SELECT * from s3object s WHERE 3.0 in s.numbers[*] ` ,
wantResult : ` { "id":2,"title":"Second Record","desc":"another text","numbers":[2,3,4]} ` ,
} ,
{
name : "select-in-list-match" ,
query : ` SELECT * from s3object s WHERE (2,3,4) IN s.nested[*] ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "select-in-nested-float" ,
query : ` SELECT s.nested from s3object s WHERE 8.5 IN s.nested[*][*] ` ,
wantResult : ` { "nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "select-in-combine-and" ,
query : ` SELECT s.nested from s3object s WHERE (8.5 IN s.nested[*][*]) AND (s.id > 0) ` ,
wantResult : ` { "nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "select-in-combine-and-no" ,
query : ` SELECT s.nested from s3object s WHERE (8.5 IN s.nested[*][*]) AND (s.id = 0) ` ,
wantResult : ` ` ,
} ,
{
name : "select-in-nested-float-no-flat" ,
query : ` SELECT s.nested from s3object s WHERE 8.5 IN s.nested[*] ` ,
wantResult : ` ` ,
} ,
{
name : "select-empty-field-result" ,
query : ` SELECT * from s3object s WHERE s.nested[0][0] = 2 ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "select-arrays-specific" ,
query : ` SELECT * from s3object s WHERE s.nested[1][0] = 7 ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "wrong-index-no-result" ,
query : ` SELECT * from s3object s WHERE s.nested[0][0] = 7 ` ,
wantResult : ` ` ,
} ,
{
name : "not-equal-result" ,
query : ` SELECT * from s3object s WHERE s.nested[1][0] != 7 ` ,
wantResult : ` { "id" : 0 , "title" : "Test Record" , "desc" : "Some text" , "synonyms" : [ "foo" , "bar" , "whatever" ] }
{ "id" : 1 , "title" : "Second Record" , "desc" : "another text" , "synonyms" : [ "some" , "synonym" , "value" ] }
{ "id" : 2 , "title" : "Second Record" , "desc" : "another text" , "numbers" : [ 2 , 3 , 4 ] } ` ,
} ,
{
name : "indexed-list-match" ,
query : ` SELECT * from s3object s WHERE (7,8.5,9) IN s.nested[1] ` ,
wantResult : ` ` ,
} ,
{
name : "indexed-list-match-equals" ,
query : ` SELECT * from s3object s WHERE (7,8.5,9) = s.nested[1] ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
2019-10-30 04:16:55 -04:00
{
name : "indexed-list-match-equals-s-star" ,
query : ` SELECT s.* from s3object s WHERE (7,8.5,9) = s.nested[1] ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "indexed-list-match-equals-s-index" ,
query : ` SELECT s.nested[1], s.nested[0] from s3object s WHERE (7,8.5,9) = s.nested[1] ` ,
wantResult : ` { "_1":[7,8.5,9],"_2":[2,3,4]} ` ,
} ,
2019-10-06 10:52:45 -04:00
{
name : "indexed-list-match-not-equals" ,
query : ` SELECT * from s3object s WHERE (7,8.5,9) != s.nested[1] ` ,
wantResult : ` { "id" : 0 , "title" : "Test Record" , "desc" : "Some text" , "synonyms" : [ "foo" , "bar" , "whatever" ] }
{ "id" : 1 , "title" : "Second Record" , "desc" : "another text" , "synonyms" : [ "some" , "synonym" , "value" ] }
{ "id" : 2 , "title" : "Second Record" , "desc" : "another text" , "numbers" : [ 2 , 3 , 4 ] } ` ,
2019-10-30 02:04:40 -04:00
} ,
{
name : "indexed-list-square-bracket" ,
query : ` SELECT * from s3object s WHERE [7,8.5,9] = s.nested[1] ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "indexed-list-square-bracket" ,
query : ` SELECT * from s3object s WHERE [7,8.5,9] IN s.nested ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "indexed-list-square-bracket" ,
query : ` SELECT * from s3object s WHERE id IN [3,2] ` ,
wantResult : ` { "id" : 2 , "title" : "Second Record" , "desc" : "another text" , "numbers" : [ 2 , 3 , 4 ] }
{ "id" : 3 , "title" : "Second Record" , "desc" : "another text" , "nested" : [ [ 2 , 3 , 4 ] , [ 7 , 8.5 , 9 ] ] } ` ,
2019-10-06 10:52:45 -04:00
} ,
{
name : "index-wildcard-in" ,
query : ` SELECT * from s3object s WHERE (8.5) IN s.nested[1][*] ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
{
name : "index-wildcard-in" ,
query : ` SELECT * from s3object s WHERE (8.0+0.5) IN s.nested[1][*] ` ,
wantResult : ` { "id":3,"title":"Second Record","desc":"another text","nested":[[2,3,4],[7,8.5,9]]} ` ,
} ,
2019-12-06 10:24:41 -05:00
{
name : "compare-mixed" ,
query : ` SELECT id from s3object s WHERE value = true ` ,
wantResult : ` { "id":1} ` ,
withJSON : ` { "id" : 0 , "value" : false }
{ "id" : 1 , "value" : true }
{ "id" : 2 , "value" : 42 }
{ "id" : 3 , "value" : "true" }
` ,
} ,
{
name : "compare-mixed-not" ,
query : ` SELECT COUNT(id) as n from s3object s WHERE value != true ` ,
wantResult : ` { "n":3} ` ,
withJSON : ` { "id" : 0 , "value" : false }
{ "id" : 1 , "value" : true }
{ "id" : 2 , "value" : 42 }
{ "id" : 3 , "value" : "true" }
` ,
} ,
2020-02-13 17:03:52 -05:00
{
name : "index-wildcard-in" ,
query : ` SELECT * from s3object s WHERE title = 'Test Record' ` ,
wantResult : ` { "id":0,"title":"Test Record","desc":"Some text","synonyms":["foo","bar","whatever"]} ` ,
} ,
2019-10-06 10:52:45 -04:00
{
name : "select-output-field-as-csv" ,
requestXML : [ ] byte ( ` < ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT s . synonyms from s3object s WHERE ' whatever ' IN s . synonyms < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
< Type > DOCUMENT < / Type >
< / JSON >
< / InputSerialization >
< OutputSerialization >
< CSV >
2020-03-14 01:09:34 -04:00
< QuoteCharacter > " < / QuoteCharacter >
2019-10-06 10:52:45 -04:00
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > ` ) ,
wantResult : ` "[""foo"",""bar"",""whatever""]" ` ,
} ,
2020-03-11 01:34:58 -04:00
{
name : "document" ,
query : "" ,
requestXML : [ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > select * from s3object [ * ] . elements [ * ] s where s . element_type = ' __elem__merfu ' < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
< Type > DOCUMENT < / Type >
< / JSON >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > ` ) ,
withJSON : `
{
"name" : "small_pdf1.pdf" ,
"lume_id" : "9507193e-572d-4f95-bcf1-e9226d96be65" ,
"elements" : [
{
"element_type" : "__elem__image" ,
"element_id" : "859d09c4-7cf1-4a37-9674-3a7de8b56abc" ,
"attributes" : {
"__attr__image_dpi" : 300 ,
"__attr__image_size" : [
2550 ,
3299
] ,
"__attr__image_index" : 1 ,
"__attr__image_format" : "JPEG" ,
"__attr__file_extension" : "jpg" ,
"__attr__data" : null
}
} ,
{
"element_type" : "__elem__merfu" ,
"element_id" : "d868aefe-ef9a-4be2-b9b2-c9fd89cc43eb" ,
"attributes" : {
"__attr__image_dpi" : 300 ,
"__attr__image_size" : [
2550 ,
3299
] ,
"__attr__image_index" : 2 ,
"__attr__image_format" : "JPEG" ,
"__attr__file_extension" : "jpg" ,
"__attr__data" : null
}
}
] ,
"data" : "asdascasdc1234e123erdasdas"
} ` ,
wantResult : ` { "element_type":"__elem__merfu","element_id":"d868aefe-ef9a-4be2-b9b2-c9fd89cc43eb","attributes": { "__attr__image_dpi":300,"__attr__image_size":[2550,3299],"__attr__image_index":2,"__attr__image_format":"JPEG","__attr__file_extension":"jpg","__attr__data":null}} ` ,
} ,
2021-03-15 17:15:52 -04:00
{
name : "date_diff_month" ,
query : ` SELECT date_diff(MONTH, '2019-10-20T', '2020-01-20T') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":3} ` ,
} ,
{
name : "date_diff_month_neg" ,
query : ` SELECT date_diff(MONTH, '2020-01-20T', '2019-10-20T') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":-3} ` ,
} ,
// Examples from https://docs.aws.amazon.com/AmazonS3/latest/userguide/s3-glacier-select-sql-reference-date.html#s3-glacier-select-sql-reference-date-diff
{
name : "date_diff_year" ,
query : ` SELECT date_diff(year, '2010-01-01T', '2011-01-01T') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":1} ` ,
} ,
{
name : "date_diff_year" ,
query : ` SELECT date_diff(month, '2010-01-01T', '2010-05T') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":4} ` ,
} ,
{
name : "date_diff_month_oney" ,
query : ` SELECT date_diff(month, '2010T', '2011T') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":12} ` ,
} ,
{
name : "date_diff_month_neg" ,
query : ` SELECT date_diff(month, '2011T', '2010T') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":-12} ` ,
} ,
{
name : "date_diff_days" ,
query : ` SELECT date_diff(day, '2010-01-01T23:00:00Z', '2010-01-02T01:00:00Z') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":0} ` ,
} ,
{
name : "date_diff_days_one" ,
query : ` SELECT date_diff(day, '2010-01-01T23:00:00Z', '2010-01-02T23:00:00Z') FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":1} ` ,
} ,
2021-03-19 12:14:38 -04:00
{
name : "cast_from_int_to_float" ,
query : ` SELECT cast(1 as float) FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":1} ` ,
} ,
{
name : "cast_from_float_to_float" ,
query : ` SELECT cast(1.0 as float) FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":1} ` ,
} ,
{
name : "arithmetic_integer_operand" ,
query : ` SELECT 1 / 2 FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":0} ` ,
} ,
{
name : "arithmetic_float_operand" ,
query : ` SELECT 1.0 / 2.0 * .3 FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":0.15} ` ,
} ,
{
name : "arithmetic_integer_float_operand" ,
query : ` SELECT 3.0 / 2, 5 / 2.0 FROM S3Object LIMIT 1 ` ,
wantResult : ` { "_1":1.5,"_2":2.5} ` ,
} ,
2021-11-02 22:14:46 -04:00
{
name : "limit-1" ,
query : ` SELECT * FROM S3Object[*].elements[*] LIMIT 1 ` ,
requestXML : [ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > select * from s3object [ * ] . elements [ * ] s where s . element_type = ' __elem__merfu ' < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
< Type > DOCUMENT < / Type >
< / JSON >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > ` ) ,
wantResult : ` { "element_type":"__elem__merfu","element_id":"d868aefe-ef9a-4be2-b9b2-c9fd89cc43eb","attributes": { "__attr__image_dpi":300,"__attr__image_size":[2550,3299],"__attr__image_index":2,"__attr__image_format":"JPEG","__attr__file_extension":"jpg","__attr__data":null}} ` ,
withJSON : `
{
"name" : "small_pdf1.pdf" ,
"lume_id" : "9507193e-572d-4f95-bcf1-e9226d96be65" ,
"elements" : [
{
"element_type" : "__elem__image" ,
"element_id" : "859d09c4-7cf1-4a37-9674-3a7de8b56abc" ,
"attributes" : {
"__attr__image_dpi" : 300 ,
"__attr__image_size" : [
2550 ,
3299
] ,
"__attr__image_index" : 1 ,
"__attr__image_format" : "JPEG" ,
"__attr__file_extension" : "jpg" ,
"__attr__data" : null
}
} ,
{
"element_type" : "__elem__merfu" ,
"element_id" : "d868aefe-ef9a-4be2-b9b2-c9fd89cc43eb" ,
"attributes" : {
"__attr__image_dpi" : 300 ,
"__attr__image_size" : [
2550 ,
3299
] ,
"__attr__image_index" : 2 ,
"__attr__image_format" : "JPEG" ,
"__attr__file_extension" : "jpg" ,
"__attr__data" : null
}
}
] ,
"data" : "asdascasdc1234e123erdasdas"
} ` ,
} ,
{
name : "limit-2" ,
query : ` select * from s3object[*].person[*] limit 1 ` ,
wantResult : ` { "Id":1,"Name":"Anshu","Address":"Templestowe","Car":"Jeep"} ` ,
withJSON : ` { "person": [ { "Id": 1, "Name": "Anshu", "Address": "Templestowe", "Car": "Jeep" }, { "Id": 2, "Name": "Ben Mostafa", "Address": "Las Vegas", "Car": "Mustang" }, { "Id": 3, "Name": "Rohan Wood", "Address": "Wooddon", "Car": "VW" } ] } ` ,
} ,
2019-10-06 10:52:45 -04:00
}
defRequest := ` < ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > % s < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
2020-02-13 17:03:52 -05:00
< Type > LINES < / Type >
2019-10-06 10:52:45 -04:00
< / JSON >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > `
2019-11-09 12:10:35 -05:00
for _ , testCase := range testTable {
t . Run ( testCase . name , func ( t * testing . T ) {
2020-02-13 17:03:52 -05:00
// Hack cpuid to the CPU doesn't appear to support AVX2.
// Restore whatever happens.
2021-03-10 12:41:29 -05:00
if cpuid . CPU . Supports ( cpuid . AVX2 ) {
cpuid . CPU . Disable ( cpuid . AVX2 )
defer cpuid . CPU . Enable ( cpuid . AVX2 )
}
if simdjson . SupportedCPU ( ) {
t . Fatal ( "setup error: expected cpu to be unsupported" )
}
2020-02-13 17:03:52 -05:00
testReq := testCase . requestXML
if len ( testReq ) == 0 {
2020-09-08 18:56:11 -04:00
var escaped bytes . Buffer
xml . EscapeText ( & escaped , [ ] byte ( testCase . query ) )
testReq = [ ] byte ( fmt . Sprintf ( defRequest , escaped . String ( ) ) )
2020-02-13 17:03:52 -05:00
}
s3Select , err := NewS3Select ( bytes . NewReader ( testReq ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
in := input
if len ( testCase . withJSON ) > 0 {
in = testCase . withJSON
}
return ioutil . NopCloser ( bytes . NewBufferString ( in ) ) , nil
} ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if ! reflect . DeepEqual ( gotS , testCase . wantResult ) {
t . Errorf ( "received response does not match with expected reply. Query: %s\ngot: %s\nwant:%s" , testCase . query , gotS , testCase . wantResult )
}
} )
t . Run ( "simd-" + testCase . name , func ( t * testing . T ) {
if ! simdjson . SupportedCPU ( ) {
t . Skip ( "No CPU support" )
}
2019-11-09 12:10:35 -05:00
testReq := testCase . requestXML
if len ( testReq ) == 0 {
2020-09-08 18:56:11 -04:00
var escaped bytes . Buffer
xml . EscapeText ( & escaped , [ ] byte ( testCase . query ) )
testReq = [ ] byte ( fmt . Sprintf ( defRequest , escaped . String ( ) ) )
2019-11-09 12:10:35 -05:00
}
s3Select , err := NewS3Select ( bytes . NewReader ( testReq ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
2019-12-06 10:24:41 -05:00
in := input
if len ( testCase . withJSON ) > 0 {
in = testCase . withJSON
}
return ioutil . NopCloser ( bytes . NewBufferString ( in ) ) , nil
2019-11-09 12:10:35 -05:00
} ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if ! reflect . DeepEqual ( gotS , testCase . wantResult ) {
t . Errorf ( "received response does not match with expected reply. Query: %s\ngot: %s\nwant:%s" , testCase . query , gotS , testCase . wantResult )
}
} )
}
}
func TestCSVQueries ( t * testing . T ) {
2020-02-13 17:03:52 -05:00
input := ` index , ID , CaseNumber , Date , Day , Month , Year , Block , IUCR , PrimaryType , Description , LocationDescription , Arrest , Domestic , Beat , District , Ward , CommunityArea , FBI Code , XCoordinate , YCoordinate , UpdatedOn , Latitude , Longitude , Location
2700763 , 7732229 , , 2010 - 05 - 26 00 : 00 : 00 , 26 , May , 2010 , 113 XX S HALSTED ST , 1150 , , CREDIT CARD FRAUD , , False , False , 2233 , 22.0 , 34.0 , , 11 , , , , 41.688043288 , - 87.6422444 , "(41.688043288, -87.6422444)" `
var testTable = [ ] struct {
name string
query string
requestXML [ ] byte
wantResult string
} {
{
name : "select-in-text-simple" ,
query : ` SELECT index FROM s3Object s WHERE "Month"='May' ` ,
wantResult : ` 2700763 ` ,
} ,
}
defRequest := ` < ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > % s < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FieldDelimiter > , < / FieldDelimiter >
< FileHeaderInfo > USE < / FileHeaderInfo >
< QuoteCharacter > " < / QuoteCharacter >
< QuoteEscapeCharacter > " < / QuoteEscapeCharacter >
< RecordDelimiter > \ n < / RecordDelimiter >
< / CSV >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > `
for _ , testCase := range testTable {
t . Run ( testCase . name , func ( t * testing . T ) {
testReq := testCase . requestXML
if len ( testReq ) == 0 {
testReq = [ ] byte ( fmt . Sprintf ( defRequest , testCase . query ) )
}
s3Select , err := NewS3Select ( bytes . NewReader ( testReq ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
return ioutil . NopCloser ( bytes . NewBufferString ( input ) ) , nil
} ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if ! reflect . DeepEqual ( gotS , testCase . wantResult ) {
t . Errorf ( "received response does not match with expected reply. Query: %s\ngot: %s\nwant:%s" , testCase . query , gotS , testCase . wantResult )
}
} )
}
}
func TestCSVQueries2 ( t * testing . T ) {
2021-12-14 12:54:50 -05:00
testInput := [ ] byte ( ` id , time , num , num2 , text
2019-11-09 12:10:35 -05:00
1 , 2010 - 01 - 01 T , 7867786 , 4565.908123 , "a text, with comma"
2 , 2017 - 01 - 02 T03 : 04 Z , - 5 , 0.765111 ,
2021-12-14 12:54:50 -05:00
` )
2019-11-09 12:10:35 -05:00
var testTable = [ ] struct {
name string
query string
2021-12-14 12:54:50 -05:00
input [ ] byte
2019-11-09 12:10:35 -05:00
requestXML [ ] byte // override request XML
wantResult string
} {
{
name : "select-all" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT * from s3object AS s WHERE id = '1' ` ,
wantResult : ` { "id":"1","time":"2010-01-01T","num":"7867786","num2":"4565.908123","text":"a text, with comma"} ` ,
} ,
{
name : "select-all-2" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT * from s3object s WHERE id = 2 ` ,
wantResult : ` { "id":"2","time":"2017-01-02T03:04Z","num":"-5","num2":" 0.765111","text":""} ` ,
} ,
{
name : "select-text-convert" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT CAST(text AS STRING) AS text from s3object s WHERE id = 1 ` ,
wantResult : ` { "text":"a text, with comma"} ` ,
} ,
{
name : "select-text-direct" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT text from s3object s WHERE id = 1 ` ,
wantResult : ` { "text":"a text, with comma"} ` ,
} ,
{
name : "select-time-direct" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT time from s3object s WHERE id = 2 ` ,
wantResult : ` { "time":"2017-01-02T03:04Z"} ` ,
} ,
{
name : "select-int-direct" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT num from s3object s WHERE id = 2 ` ,
wantResult : ` { "num":"-5"} ` ,
} ,
{
name : "select-float-direct" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT num2 from s3object s WHERE id = 2 ` ,
wantResult : ` { "num2":" 0.765111"} ` ,
} ,
2020-04-24 16:02:59 -04:00
{
name : "select-in-array" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2020-04-24 16:02:59 -04:00
query : ` select id from S3Object s WHERE id in [1,3] ` ,
wantResult : ` { "id":"1"} ` ,
} ,
{
name : "select-in-array-matchnone" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2020-04-24 16:02:59 -04:00
query : ` select id from S3Object s WHERE s.id in [4,3] ` ,
wantResult : ` ` ,
} ,
2019-11-09 12:10:35 -05:00
{
name : "select-float-by-val" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2019-11-09 12:10:35 -05:00
query : ` SELECT num2 from s3object s WHERE num2 = 0.765111 ` ,
wantResult : ` { "num2":" 0.765111"} ` ,
} ,
2021-09-13 12:13:25 -04:00
{
name : "select-non_exiting_values" ,
2021-12-14 12:54:50 -05:00
input : testInput ,
2021-09-13 12:13:25 -04:00
query : ` SELECT _1 as first, s._100 from s3object s LIMIT 1 ` ,
wantResult : ` { "first":"1","_100":null} ` ,
} ,
2021-12-14 12:54:50 -05:00
{
name : "select-is_null_noresults" ,
input : testInput ,
query : ` select _2 from S3object where _2 IS NULL ` ,
wantResult : ` ` ,
} ,
{
name : "select-is_null_results" ,
input : testInput ,
query : ` select _2 from S3object WHERE _100 IS NULL ` ,
wantResult : ` { "_2" : "2010-01-01T" }
{ "_2" : "2017-01-02T03:04Z" } ` ,
} ,
{
name : "select-is_not_null_results" ,
input : testInput ,
query : ` select _2 from S3object where _2 IS NOT NULL ` ,
wantResult : ` { "_2" : "2010-01-01T" }
{ "_2" : "2017-01-02T03:04Z" } ` ,
} ,
{
name : "select-is_not_null_noresults" ,
input : testInput ,
query : ` select _2 from S3object WHERE _100 IS NOT NULL ` ,
wantResult : ` ` ,
} ,
{
name : "select-is_not_string" ,
input : [ ] byte ( ` c1 , c2 , c3
1 , 2 , 3
1 , , 3 ` ) ,
query : ` select * from S3object where _2 IS NOT '' ` ,
wantResult : ` { "c1":"1","c2":"2","c3":"3"} ` ,
} ,
2019-11-09 12:10:35 -05:00
}
defRequest := ` < ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > % s < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FileHeaderInfo > USE < / FileHeaderInfo >
2020-03-14 01:09:34 -04:00
< QuoteCharacter > " < / QuoteCharacter >
2019-11-09 12:10:35 -05:00
< / CSV >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > `
2019-10-06 10:52:45 -04:00
for _ , testCase := range testTable {
t . Run ( testCase . name , func ( t * testing . T ) {
testReq := testCase . requestXML
if len ( testReq ) == 0 {
testReq = [ ] byte ( fmt . Sprintf ( defRequest , testCase . query ) )
}
s3Select , err := NewS3Select ( bytes . NewReader ( testReq ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
2021-12-14 12:54:50 -05:00
return ioutil . NopCloser ( bytes . NewBuffer ( testCase . input ) ) , nil
2019-10-06 10:52:45 -04:00
} ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if ! reflect . DeepEqual ( gotS , testCase . wantResult ) {
t . Errorf ( "received response does not match with expected reply. Query: %s\ngot: %s\nwant:%s" , testCase . query , gotS , testCase . wantResult )
}
} )
}
}
2021-04-06 11:49:04 -04:00
func TestCSVQueries3 ( t * testing . T ) {
input := ` na . me , qty , CAST
apple , 1 , true
mango , 3 , false
`
var testTable = [ ] struct {
name string
query string
requestXML [ ] byte // override request XML
wantResult string
} {
{
name : "Select a column containing dot" ,
query : ` select "na.me" from S3Object s ` ,
wantResult : ` apple
mango ` ,
} ,
{
name : "Select column containing dot with table name prefix" ,
query : ` select count(S3Object."na.me") from S3Object ` ,
wantResult : ` 2 ` ,
} ,
{
name : "Select column containing dot with table alias prefix" ,
query : ` select s."na.me" from S3Object as s ` ,
wantResult : ` apple
mango ` ,
} ,
{
name : "Select column simplest" ,
query : ` select qty from S3Object ` ,
wantResult : ` 1
3 ` ,
} ,
{
name : "Select column with table name prefix" ,
query : ` select S3Object.qty from S3Object ` ,
wantResult : ` 1
3 ` ,
} ,
{
name : "Select column without table alias" ,
query : ` select qty from S3Object s ` ,
wantResult : ` 1
3 ` ,
} ,
{
name : "Select column with table alias" ,
query : ` select s.qty from S3Object s ` ,
wantResult : ` 1
3 ` ,
} ,
{
name : "Select reserved word column" ,
query : ` select "CAST" from s3object ` ,
wantResult : ` true
false ` ,
} ,
{
name : "Select reserved word column with table alias" ,
query : ` select S3Object."CAST" from s3object ` ,
wantResult : ` true
false ` ,
} ,
{
name : "Select reserved word column with unused table alias" ,
query : ` select "CAST" from s3object s ` ,
wantResult : ` true
false ` ,
} ,
{
name : "Select reserved word column with table alias" ,
query : ` select s."CAST" from s3object s ` ,
wantResult : ` true
false ` ,
} ,
{
name : "Select reserved word column with table alias" ,
query : ` select NOT CAST(s."CAST" AS Bool) from s3object s ` ,
wantResult : ` false
true ` ,
} ,
}
defRequest := ` < ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > % s < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FileHeaderInfo > USE < / FileHeaderInfo >
< QuoteCharacter > " < / QuoteCharacter >
< / CSV >
< / InputSerialization >
< OutputSerialization >
< CSV / >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest > `
for _ , testCase := range testTable {
t . Run ( testCase . name , func ( t * testing . T ) {
testReq := testCase . requestXML
if len ( testReq ) == 0 {
testReq = [ ] byte ( fmt . Sprintf ( defRequest , testCase . query ) )
}
s3Select , err := NewS3Select ( bytes . NewReader ( testReq ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
return ioutil . NopCloser ( bytes . NewBufferString ( input ) ) , nil
} ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if gotS != testCase . wantResult {
t . Errorf ( "received response does not match with expected reply.\nQuery: %s\n=====\ngot: %s\n=====\nwant: %s\n=====\n" , testCase . query , gotS , testCase . wantResult )
}
} )
}
}
2019-01-28 23:00:18 -05:00
func TestCSVInput ( t * testing . T ) {
2019-07-03 19:34:54 -04:00
var testTable = [ ] struct {
requestXML [ ] byte
expectedResult [ ] byte
} {
{
[ ] byte ( `
2019-01-08 19:53:04 -05:00
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT one , two , three from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FileHeaderInfo > USE < / FileHeaderInfo >
< / CSV >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
2019-07-03 19:34:54 -04:00
` ) , [ ] byte {
0 , 0 , 0 , 137 , 0 , 0 , 0 , 85 , 194 , 213 , 168 , 241 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 24 , 97 , 112 , 112 , 108 , 105 , 99 , 97 , 116 , 105 , 111 , 110 , 47 , 111 , 99 , 116 , 101 , 116 , 45 , 115 , 116 , 114 , 101 , 97 , 109 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 7 , 82 , 101 , 99 , 111 , 114 , 100 , 115 , 45 , 49 , 44 , 102 , 111 , 111 , 44 , 116 , 114 , 117 , 101 , 10 , 44 , 98 , 97 , 114 , 44 , 102 , 97 , 108 , 115 , 101 , 10 , 50 , 46 , 53 , 44 , 98 , 97 , 122 , 44 , 116 , 114 , 117 , 101 , 10 , 75 , 182 , 193 , 80 , 0 , 0 , 0 , 235 , 0 , 0 , 0 , 67 , 213 , 243 , 57 , 141 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 8 , 116 , 101 , 120 , 116 , 47 , 120 , 109 , 108 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 83 , 116 , 97 , 116 , 115 , 60 , 63 , 120 , 109 , 108 , 32 , 118 , 101 , 114 , 115 , 105 , 111 , 110 , 61 , 34 , 49 , 46 , 48 , 34 , 32 , 101 , 110 , 99 , 111 , 100 , 105 , 110 , 103 , 61 , 34 , 85 , 84 , 70 , 45 , 56 , 34 , 63 , 62 , 60 , 83 , 116 , 97 , 116 , 115 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 53 , 48 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 53 , 48 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 51 , 54 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 60 , 47 , 83 , 116 , 97 , 116 , 115 , 62 , 253 , 105 , 8 , 216 , 0 , 0 , 0 , 56 , 0 , 0 , 0 , 40 , 193 , 198 , 132 , 212 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 3 , 69 , 110 , 100 , 207 , 151 , 211 , 146 ,
} ,
} ,
{
[ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT COUNT ( * ) AS total_record_count from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FileHeaderInfo > USE < / FileHeaderInfo >
< / CSV >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ) , [ ] byte {
0 , 0 , 0 , 126 , 0 , 0 , 0 , 85 , 56 , 193 , 36 , 188 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 24 , 97 , 112 , 112 , 108 , 105 , 99 , 97 , 116 , 105 , 111 , 110 , 47 , 111 , 99 , 116 , 101 , 116 , 45 , 115 , 116 , 114 , 101 , 97 , 109 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 7 , 82 , 101 , 99 , 111 , 114 , 100 , 115 , 123 , 34 , 116 , 111 , 116 , 97 , 108 , 95 , 114 , 101 , 99 , 111 , 114 , 100 , 95 , 99 , 111 , 117 , 110 , 116 , 34 , 58 , 51 , 125 , 10 , 196 , 183 , 134 , 242 , 0 , 0 , 0 , 235 , 0 , 0 , 0 , 67 , 213 , 243 , 57 , 141 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 8 , 116 , 101 , 120 , 116 , 47 , 120 , 109 , 108 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 83 , 116 , 97 , 116 , 115 , 60 , 63 , 120 , 109 , 108 , 32 , 118 , 101 , 114 , 115 , 105 , 111 , 110 , 61 , 34 , 49 , 46 , 48 , 34 , 32 , 101 , 110 , 99 , 111 , 100 , 105 , 110 , 103 , 61 , 34 , 85 , 84 , 70 , 45 , 56 , 34 , 63 , 62 , 60 , 83 , 116 , 97 , 116 , 115 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 53 , 48 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 53 , 48 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 50 , 53 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 60 , 47 , 83 , 116 , 97 , 116 , 115 , 62 , 47 , 153 , 24 , 28 , 0 , 0 , 0 , 56 , 0 , 0 , 0 , 40 , 193 , 198 , 132 , 212 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 3 , 69 , 110 , 100 , 207 , 151 , 211 , 146 ,
} ,
} ,
2019-09-20 14:00:17 -04:00
{
[ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
2019-09-27 15:33:14 -04:00
< SelectObjectContentRequest >
< Expression > SELECT * from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FileHeaderInfo > USE < / FileHeaderInfo >
< / CSV >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ) , [ ] byte { 0x0 , 0x0 , 0x0 , 0xdd , 0x0 , 0x0 , 0x0 , 0x55 , 0xf , 0x46 , 0xc1 , 0xfa , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xd , 0x3a , 0x63 , 0x6f , 0x6e , 0x74 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x18 , 0x61 , 0x70 , 0x70 , 0x6c , 0x69 , 0x63 , 0x61 , 0x74 , 0x69 , 0x6f , 0x6e , 0x2f , 0x6f , 0x63 , 0x74 , 0x65 , 0x74 , 0x2d , 0x73 , 0x74 , 0x72 , 0x65 , 0x61 , 0x6d , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x7 , 0x52 , 0x65 , 0x63 , 0x6f , 0x72 , 0x64 , 0x73 , 0x7b , 0x22 , 0x6f , 0x6e , 0x65 , 0x22 , 0x3a , 0x22 , 0x2d , 0x31 , 0x22 , 0x2c , 0x22 , 0x74 , 0x77 , 0x6f , 0x22 , 0x3a , 0x22 , 0x66 , 0x6f , 0x6f , 0x22 , 0x2c , 0x22 , 0x74 , 0x68 , 0x72 , 0x65 , 0x65 , 0x22 , 0x3a , 0x22 , 0x74 , 0x72 , 0x75 , 0x65 , 0x22 , 0x7d , 0xa , 0x7b , 0x22 , 0x6f , 0x6e , 0x65 , 0x22 , 0x3a , 0x22 , 0x22 , 0x2c , 0x22 , 0x74 , 0x77 , 0x6f , 0x22 , 0x3a , 0x22 , 0x62 , 0x61 , 0x72 , 0x22 , 0x2c , 0x22 , 0x74 , 0x68 , 0x72 , 0x65 , 0x65 , 0x22 , 0x3a , 0x22 , 0x66 , 0x61 , 0x6c , 0x73 , 0x65 , 0x22 , 0x7d , 0xa , 0x7b , 0x22 , 0x6f , 0x6e , 0x65 , 0x22 , 0x3a , 0x22 , 0x32 , 0x2e , 0x35 , 0x22 , 0x2c , 0x22 , 0x74 , 0x77 , 0x6f , 0x22 , 0x3a , 0x22 , 0x62 , 0x61 , 0x7a , 0x22 , 0x2c , 0x22 , 0x74 , 0x68 , 0x72 , 0x65 , 0x65 , 0x22 , 0x3a , 0x22 , 0x74 , 0x72 , 0x75 , 0x65 , 0x22 , 0x7d , 0xa , 0x7e , 0xb5 , 0x99 , 0xfb , 0x0 , 0x0 , 0x0 , 0xec , 0x0 , 0x0 , 0x0 , 0x43 , 0x67 , 0xd3 , 0xe5 , 0x9d , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xd , 0x3a , 0x63 , 0x6f , 0x6e , 0x74 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x8 , 0x74 , 0x65 , 0x78 , 0x74 , 0x2f , 0x78 , 0x6d , 0x6c , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3c , 0x3f , 0x78 , 0x6d , 0x6c , 0x20 , 0x76 , 0x65 , 0x72 , 0x73 , 0x69 , 0x6f , 0x6e , 0x3d , 0x22 , 0x31 , 0x2e , 0x30 , 0x22 , 0x20 , 0x65 , 0x6e , 0x63 , 0x6f , 0x64 , 0x69 , 0x6e , 0x67 , 0x3d , 0x22 , 0x55 , 0x54 , 0x46 , 0x2d , 0x38 , 0x22 , 0x3f , 0x3e , 0x3c , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x53 , 0x63 , 0x61 , 0x6e , 0x6e , 0x65 , 0x64 , 0x3e , 0x35 , 0x30 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x53 , 0x63 , 0x61 , 0x6e , 0x6e , 0x65 , 0x64 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x50 , 0x72 , 0x6f , 0x63 , 0x65 , 0x73 , 0x73 , 0x65 , 0x64 , 0x3e , 0x35 , 0x30 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x50 , 0x72 , 0x6f , 0x63 , 0x65 , 0x73 , 0x73 , 0x65 , 0x64 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x52 , 0x65 , 0x74 , 0x75 , 0x72 , 0x6e , 0x65 , 0x64 , 0x3e , 0x31 , 0x32 , 0x30 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x52 , 0x65 , 0x74 , 0x75 , 0x72 , 0x6e , 0x65 , 0x64 , 0x3e , 0x3c , 0x2f , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3e , 0x5a , 0xe5 , 0xd , 0x84 , 0x0 , 0x0 , 0x0 , 0x38 , 0x0 , 0x0 , 0x0 , 0x28 , 0xc1 , 0xc6 , 0x84 , 0xd4 , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x3 , 0x45 , 0x6e , 0x64 , 0xcf , 0x97 , 0xd3 , 0x92 } ,
} ,
{
[ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
2019-09-20 14:00:17 -04:00
< SelectObjectContentRequest >
< Expression > SELECT one from S3Object limit 1 < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< CSV >
< FileHeaderInfo > USE < / FileHeaderInfo >
< / CSV >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ) , [ ] byte {
0x0 , 0x0 , 0x0 , 0x68 , 0x0 , 0x0 , 0x0 , 0x55 , 0xd7 , 0x61 , 0x46 , 0x9e , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xd , 0x3a , 0x63 , 0x6f , 0x6e , 0x74 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x18 , 0x61 , 0x70 , 0x70 , 0x6c , 0x69 , 0x63 , 0x61 , 0x74 , 0x69 , 0x6f , 0x6e , 0x2f , 0x6f , 0x63 , 0x74 , 0x65 , 0x74 , 0x2d , 0x73 , 0x74 , 0x72 , 0x65 , 0x61 , 0x6d , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x7 , 0x52 , 0x65 , 0x63 , 0x6f , 0x72 , 0x64 , 0x73 , 0x2d , 0x31 , 0xa , 0x17 , 0xfb , 0x1 , 0x90 , 0x0 , 0x0 , 0x0 , 0xea , 0x0 , 0x0 , 0x0 , 0x43 , 0xe8 , 0x93 , 0x10 , 0x3d , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xd , 0x3a , 0x63 , 0x6f , 0x6e , 0x74 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x8 , 0x74 , 0x65 , 0x78 , 0x74 , 0x2f , 0x78 , 0x6d , 0x6c , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3c , 0x3f , 0x78 , 0x6d , 0x6c , 0x20 , 0x76 , 0x65 , 0x72 , 0x73 , 0x69 , 0x6f , 0x6e , 0x3d , 0x22 , 0x31 , 0x2e , 0x30 , 0x22 , 0x20 , 0x65 , 0x6e , 0x63 , 0x6f , 0x64 , 0x69 , 0x6e , 0x67 , 0x3d , 0x22 , 0x55 , 0x54 , 0x46 , 0x2d , 0x38 , 0x22 , 0x3f , 0x3e , 0x3c , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x53 , 0x63 , 0x61 , 0x6e , 0x6e , 0x65 , 0x64 , 0x3e , 0x35 , 0x30 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x53 , 0x63 , 0x61 , 0x6e , 0x6e , 0x65 , 0x64 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x50 , 0x72 , 0x6f , 0x63 , 0x65 , 0x73 , 0x73 , 0x65 , 0x64 , 0x3e , 0x35 , 0x30 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x50 , 0x72 , 0x6f , 0x63 , 0x65 , 0x73 , 0x73 , 0x65 , 0x64 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x52 , 0x65 , 0x74 , 0x75 , 0x72 , 0x6e , 0x65 , 0x64 , 0x3e , 0x33 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x52 , 0x65 , 0x74 , 0x75 , 0x72 , 0x6e , 0x65 , 0x64 , 0x3e , 0x3c , 0x2f , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3e , 0x15 , 0x72 , 0x19 , 0x94 , 0x0 , 0x0 , 0x0 , 0x38 , 0x0 , 0x0 , 0x0 , 0x28 , 0xc1 , 0xc6 , 0x84 , 0xd4 , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x3 , 0x45 , 0x6e , 0x64 , 0xcf , 0x97 , 0xd3 , 0x92 ,
} ,
} ,
2019-07-03 19:34:54 -04:00
}
2018-08-15 06:30:19 -04:00
2019-01-08 19:53:04 -05:00
var csvData = [ ] byte ( ` one , two , three
2019-01-16 11:22:04 -05:00
- 1 , foo , true
, bar , false
2.5 , baz , true
2019-01-08 19:53:04 -05:00
` )
2018-08-15 06:30:19 -04:00
2019-09-13 17:18:35 -04:00
for i , testCase := range testTable {
t . Run ( fmt . Sprint ( i ) , func ( t * testing . T ) {
s3Select , err := NewS3Select ( bytes . NewReader ( testCase . requestXML ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
return ioutil . NopCloser ( bytes . NewReader ( csvData ) ) , nil
} ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
if ! reflect . DeepEqual ( w . response , testCase . expectedResult ) {
2020-02-13 17:03:52 -05:00
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
t . Errorf ( "received response does not match with expected reply\ngot: %#v\nwant:%#v\ndecoded:%s" , w . response , testCase . expectedResult , string ( got ) )
2019-09-13 17:18:35 -04:00
}
} )
2018-08-15 06:30:19 -04:00
}
2019-09-13 17:18:35 -04:00
2018-08-15 06:30:19 -04:00
}
2019-01-08 19:53:04 -05:00
func TestJSONInput ( t * testing . T ) {
2019-07-03 19:34:54 -04:00
var testTable = [ ] struct {
requestXML [ ] byte
expectedResult [ ] byte
} {
{
[ ] byte ( `
2019-01-08 19:53:04 -05:00
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT one , two , three from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
< Type > DOCUMENT < / Type >
< / JSON >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
2019-07-03 19:34:54 -04:00
` ) , [ ] byte {
0 , 0 , 0 , 137 , 0 , 0 , 0 , 85 , 194 , 213 , 168 , 241 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 24 , 97 , 112 , 112 , 108 , 105 , 99 , 97 , 116 , 105 , 111 , 110 , 47 , 111 , 99 , 116 , 101 , 116 , 45 , 115 , 116 , 114 , 101 , 97 , 109 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 7 , 82 , 101 , 99 , 111 , 114 , 100 , 115 , 45 , 49 , 44 , 102 , 111 , 111 , 44 , 116 , 114 , 117 , 101 , 10 , 44 , 98 , 97 , 114 , 44 , 102 , 97 , 108 , 115 , 101 , 10 , 50 , 46 , 53 , 44 , 98 , 97 , 122 , 44 , 116 , 114 , 117 , 101 , 10 , 75 , 182 , 193 , 80 , 0 , 0 , 0 , 237 , 0 , 0 , 0 , 67 , 90 , 179 , 204 , 45 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 8 , 116 , 101 , 120 , 116 , 47 , 120 , 109 , 108 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 83 , 116 , 97 , 116 , 115 , 60 , 63 , 120 , 109 , 108 , 32 , 118 , 101 , 114 , 115 , 105 , 111 , 110 , 61 , 34 , 49 , 46 , 48 , 34 , 32 , 101 , 110 , 99 , 111 , 100 , 105 , 110 , 103 , 61 , 34 , 85 , 84 , 70 , 45 , 56 , 34 , 63 , 62 , 60 , 83 , 116 , 97 , 116 , 115 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 49 , 49 , 50 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 49 , 49 , 50 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 51 , 54 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 60 , 47 , 83 , 116 , 97 , 116 , 115 , 62 , 181 , 40 , 50 , 250 , 0 , 0 , 0 , 56 , 0 , 0 , 0 , 40 , 193 , 198 , 132 , 212 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 3 , 69 , 110 , 100 , 207 , 151 , 211 , 146 ,
} ,
} ,
{
[ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT COUNT ( * ) AS total_record_count from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
< Type > DOCUMENT < / Type >
< / JSON >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ) , [ ] byte {
0 , 0 , 0 , 103 , 0 , 0 , 0 , 85 , 85 , 49 , 209 , 79 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 24 , 97 , 112 , 112 , 108 , 105 , 99 , 97 , 116 , 105 , 111 , 110 , 47 , 111 , 99 , 116 , 101 , 116 , 45 , 115 , 116 , 114 , 101 , 97 , 109 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 7 , 82 , 101 , 99 , 111 , 114 , 100 , 115 , 51 , 10 , 175 , 58 , 213 , 152 , 0 , 0 , 0 , 236 , 0 , 0 , 0 , 67 , 103 , 211 , 229 , 157 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 8 , 116 , 101 , 120 , 116 , 47 , 120 , 109 , 108 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 83 , 116 , 97 , 116 , 115 , 60 , 63 , 120 , 109 , 108 , 32 , 118 , 101 , 114 , 115 , 105 , 111 , 110 , 61 , 34 , 49 , 46 , 48 , 34 , 32 , 101 , 110 , 99 , 111 , 100 , 105 , 110 , 103 , 61 , 34 , 85 , 84 , 70 , 45 , 56 , 34 , 63 , 62 , 60 , 83 , 116 , 97 , 116 , 115 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 49 , 49 , 50 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 49 , 49 , 50 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 50 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 60 , 47 , 83 , 116 , 97 , 116 , 115 , 62 , 52 , 192 , 77 , 114 , 0 , 0 , 0 , 56 , 0 , 0 , 0 , 40 , 193 , 198 , 132 , 212 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 3 , 69 , 110 , 100 , 207 , 151 , 211 , 146 ,
} ,
} ,
2019-09-27 15:33:14 -04:00
{
[ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT * from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< JSON >
< Type > DOCUMENT < / Type >
< / JSON >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ) , [ ] byte { 0x0 , 0x0 , 0x0 , 0x89 , 0x0 , 0x0 , 0x0 , 0x55 , 0xc2 , 0xd5 , 0xa8 , 0xf1 , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xd , 0x3a , 0x63 , 0x6f , 0x6e , 0x74 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x18 , 0x61 , 0x70 , 0x70 , 0x6c , 0x69 , 0x63 , 0x61 , 0x74 , 0x69 , 0x6f , 0x6e , 0x2f , 0x6f , 0x63 , 0x74 , 0x65 , 0x74 , 0x2d , 0x73 , 0x74 , 0x72 , 0x65 , 0x61 , 0x6d , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x7 , 0x52 , 0x65 , 0x63 , 0x6f , 0x72 , 0x64 , 0x73 , 0x74 , 0x72 , 0x75 , 0x65 , 0x2c , 0x66 , 0x6f , 0x6f , 0x2c , 0x2d , 0x31 , 0xa , 0x66 , 0x61 , 0x6c , 0x73 , 0x65 , 0x2c , 0x62 , 0x61 , 0x72 , 0x2c , 0xa , 0x74 , 0x72 , 0x75 , 0x65 , 0x2c , 0x62 , 0x61 , 0x7a , 0x2c , 0x32 , 0x2e , 0x35 , 0xa , 0xef , 0x22 , 0x13 , 0xa3 , 0x0 , 0x0 , 0x0 , 0xed , 0x0 , 0x0 , 0x0 , 0x43 , 0x5a , 0xb3 , 0xcc , 0x2d , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xd , 0x3a , 0x63 , 0x6f , 0x6e , 0x74 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x8 , 0x74 , 0x65 , 0x78 , 0x74 , 0x2f , 0x78 , 0x6d , 0x6c , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3c , 0x3f , 0x78 , 0x6d , 0x6c , 0x20 , 0x76 , 0x65 , 0x72 , 0x73 , 0x69 , 0x6f , 0x6e , 0x3d , 0x22 , 0x31 , 0x2e , 0x30 , 0x22 , 0x20 , 0x65 , 0x6e , 0x63 , 0x6f , 0x64 , 0x69 , 0x6e , 0x67 , 0x3d , 0x22 , 0x55 , 0x54 , 0x46 , 0x2d , 0x38 , 0x22 , 0x3f , 0x3e , 0x3c , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x53 , 0x63 , 0x61 , 0x6e , 0x6e , 0x65 , 0x64 , 0x3e , 0x31 , 0x31 , 0x32 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x53 , 0x63 , 0x61 , 0x6e , 0x6e , 0x65 , 0x64 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x50 , 0x72 , 0x6f , 0x63 , 0x65 , 0x73 , 0x73 , 0x65 , 0x64 , 0x3e , 0x31 , 0x31 , 0x32 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x50 , 0x72 , 0x6f , 0x63 , 0x65 , 0x73 , 0x73 , 0x65 , 0x64 , 0x3e , 0x3c , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x52 , 0x65 , 0x74 , 0x75 , 0x72 , 0x6e , 0x65 , 0x64 , 0x3e , 0x33 , 0x36 , 0x3c , 0x2f , 0x42 , 0x79 , 0x74 , 0x65 , 0x73 , 0x52 , 0x65 , 0x74 , 0x75 , 0x72 , 0x6e , 0x65 , 0x64 , 0x3e , 0x3c , 0x2f , 0x53 , 0x74 , 0x61 , 0x74 , 0x73 , 0x3e , 0xb5 , 0x28 , 0x32 , 0xfa , 0x0 , 0x0 , 0x0 , 0x38 , 0x0 , 0x0 , 0x0 , 0x28 , 0xc1 , 0xc6 , 0x84 , 0xd4 , 0xd , 0x3a , 0x6d , 0x65 , 0x73 , 0x73 , 0x61 , 0x67 , 0x65 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x5 , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0xb , 0x3a , 0x65 , 0x76 , 0x65 , 0x6e , 0x74 , 0x2d , 0x74 , 0x79 , 0x70 , 0x65 , 0x7 , 0x0 , 0x3 , 0x45 , 0x6e , 0x64 , 0xcf , 0x97 , 0xd3 , 0x92 } ,
} ,
2019-07-03 19:34:54 -04:00
}
2019-01-08 19:53:04 -05:00
2019-01-16 11:22:04 -05:00
var jsonData = [ ] byte ( ` { "three" : true , "two" : "foo" , "one" : - 1 }
{ "three" : false , "two" : "bar" , "one" : null }
{ "three" : true , "two" : "baz" , "one" : 2.5 }
2019-01-08 19:53:04 -05:00
` )
2019-09-13 17:18:35 -04:00
for i , testCase := range testTable {
t . Run ( fmt . Sprint ( i ) , func ( t * testing . T ) {
s3Select , err := NewS3Select ( bytes . NewReader ( testCase . requestXML ) )
if err != nil {
t . Fatal ( err )
}
2019-01-08 19:53:04 -05:00
2019-09-13 17:18:35 -04:00
if err = s3Select . Open ( func ( offset , length int64 ) ( io . ReadCloser , error ) {
return ioutil . NopCloser ( bytes . NewReader ( jsonData ) ) , nil
} ) ; err != nil {
t . Fatal ( err )
}
2019-01-08 19:53:04 -05:00
2019-09-13 17:18:35 -04:00
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
2019-01-08 19:53:04 -05:00
2019-09-13 17:18:35 -04:00
if ! reflect . DeepEqual ( w . response , testCase . expectedResult ) {
2020-02-13 17:03:52 -05:00
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
t . Errorf ( "received response does not match with expected reply\ngot: %#v\nwant:%#v\ndecoded:%s" , w . response , testCase . expectedResult , string ( got ) )
2019-09-13 17:18:35 -04:00
}
} )
2019-01-08 19:53:04 -05:00
}
}
func TestParquetInput ( t * testing . T ) {
2020-08-18 13:23:28 -04:00
os . Setenv ( "MINIO_API_SELECT_PARQUET" , "on" )
defer os . Setenv ( "MINIO_API_SELECT_PARQUET" , "off" )
2019-07-03 19:34:54 -04:00
var testTable = [ ] struct {
requestXML [ ] byte
expectedResult [ ] byte
} {
{
[ ] byte ( `
2019-01-08 19:53:04 -05:00
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT one , two , three from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< Parquet >
< / Parquet >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
2019-07-03 19:34:54 -04:00
` ) , [ ] byte {
0 , 0 , 0 , 137 , 0 , 0 , 0 , 85 , 194 , 213 , 168 , 241 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 24 , 97 , 112 , 112 , 108 , 105 , 99 , 97 , 116 , 105 , 111 , 110 , 47 , 111 , 99 , 116 , 101 , 116 , 45 , 115 , 116 , 114 , 101 , 97 , 109 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 7 , 82 , 101 , 99 , 111 , 114 , 100 , 115 , 45 , 49 , 44 , 102 , 111 , 111 , 44 , 116 , 114 , 117 , 101 , 10 , 44 , 98 , 97 , 114 , 44 , 102 , 97 , 108 , 115 , 101 , 10 , 50 , 46 , 53 , 44 , 98 , 97 , 122 , 44 , 116 , 114 , 117 , 101 , 10 , 75 , 182 , 193 , 80 , 0 , 0 , 0 , 235 , 0 , 0 , 0 , 67 , 213 , 243 , 57 , 141 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 8 , 116 , 101 , 120 , 116 , 47 , 120 , 109 , 108 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 83 , 116 , 97 , 116 , 115 , 60 , 63 , 120 , 109 , 108 , 32 , 118 , 101 , 114 , 115 , 105 , 111 , 110 , 61 , 34 , 49 , 46 , 48 , 34 , 32 , 101 , 110 , 99 , 111 , 100 , 105 , 110 , 103 , 61 , 34 , 85 , 84 , 70 , 45 , 56 , 34 , 63 , 62 , 60 , 83 , 116 , 97 , 116 , 115 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 45 , 49 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 45 , 49 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 51 , 54 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 60 , 47 , 83 , 116 , 97 , 116 , 115 , 62 , 128 , 96 , 253 , 66 , 0 , 0 , 0 , 56 , 0 , 0 , 0 , 40 , 193 , 198 , 132 , 212 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 3 , 69 , 110 , 100 , 207 , 151 , 211 , 146 ,
} ,
} ,
{
[ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT COUNT ( * ) AS total_record_count from S3Object < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< Parquet >
< / Parquet >
< / InputSerialization >
< OutputSerialization >
< CSV >
< / CSV >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ) , [ ] byte {
0 , 0 , 0 , 103 , 0 , 0 , 0 , 85 , 85 , 49 , 209 , 79 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 24 , 97 , 112 , 112 , 108 , 105 , 99 , 97 , 116 , 105 , 111 , 110 , 47 , 111 , 99 , 116 , 101 , 116 , 45 , 115 , 116 , 114 , 101 , 97 , 109 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 7 , 82 , 101 , 99 , 111 , 114 , 100 , 115 , 51 , 10 , 175 , 58 , 213 , 152 , 0 , 0 , 0 , 234 , 0 , 0 , 0 , 67 , 232 , 147 , 16 , 61 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 13 , 58 , 99 , 111 , 110 , 116 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 8 , 116 , 101 , 120 , 116 , 47 , 120 , 109 , 108 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 83 , 116 , 97 , 116 , 115 , 60 , 63 , 120 , 109 , 108 , 32 , 118 , 101 , 114 , 115 , 105 , 111 , 110 , 61 , 34 , 49 , 46 , 48 , 34 , 32 , 101 , 110 , 99 , 111 , 100 , 105 , 110 , 103 , 61 , 34 , 85 , 84 , 70 , 45 , 56 , 34 , 63 , 62 , 60 , 83 , 116 , 97 , 116 , 115 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 45 , 49 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 83 , 99 , 97 , 110 , 110 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 45 , 49 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 80 , 114 , 111 , 99 , 101 , 115 , 115 , 101 , 100 , 62 , 60 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 50 , 60 , 47 , 66 , 121 , 116 , 101 , 115 , 82 , 101 , 116 , 117 , 114 , 110 , 101 , 100 , 62 , 60 , 47 , 83 , 116 , 97 , 116 , 115 , 62 , 190 , 146 , 162 , 21 , 0 , 0 , 0 , 56 , 0 , 0 , 0 , 40 , 193 , 198 , 132 , 212 , 13 , 58 , 109 , 101 , 115 , 115 , 97 , 103 , 101 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 5 , 101 , 118 , 101 , 110 , 116 , 11 , 58 , 101 , 118 , 101 , 110 , 116 , 45 , 116 , 121 , 112 , 101 , 7 , 0 , 3 , 69 , 110 , 100 , 207 , 151 , 211 , 146 ,
} ,
} ,
}
2018-08-15 06:30:19 -04:00
2019-09-13 17:18:35 -04:00
for i , testCase := range testTable {
t . Run ( fmt . Sprint ( i ) , func ( t * testing . T ) {
getReader := func ( offset int64 , length int64 ) ( io . ReadCloser , error ) {
2021-04-03 11:25:19 -04:00
testdataFile := "testdata/testdata.parquet"
2019-09-13 17:18:35 -04:00
file , err := os . Open ( testdataFile )
if err != nil {
return nil , err
}
fi , err := file . Stat ( )
if err != nil {
return nil , err
}
if offset < 0 {
offset = fi . Size ( ) + offset
}
2020-05-18 12:59:45 -04:00
if _ , err = file . Seek ( offset , io . SeekStart ) ; err != nil {
2019-09-13 17:18:35 -04:00
return nil , err
}
return file , nil
2019-07-03 19:34:54 -04:00
}
2018-08-15 06:30:19 -04:00
2019-09-13 17:18:35 -04:00
s3Select , err := NewS3Select ( bytes . NewReader ( testCase . requestXML ) )
2019-07-03 19:34:54 -04:00
if err != nil {
2019-09-13 17:18:35 -04:00
t . Fatal ( err )
2019-07-03 19:34:54 -04:00
}
2018-11-14 18:55:10 -05:00
2019-09-13 17:18:35 -04:00
if err = s3Select . Open ( getReader ) ; err != nil {
t . Fatal ( err )
2019-07-03 19:34:54 -04:00
}
2018-11-14 18:55:10 -05:00
2019-09-13 17:18:35 -04:00
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
2018-11-14 18:55:10 -05:00
2019-09-13 17:18:35 -04:00
if ! reflect . DeepEqual ( w . response , testCase . expectedResult ) {
2020-02-13 17:03:52 -05:00
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
t . Errorf ( "received response does not match with expected reply\ngot: %#v\nwant:%#v\ndecoded:%s" , w . response , testCase . expectedResult , string ( got ) )
2019-09-13 17:18:35 -04:00
}
} )
2019-01-08 19:53:04 -05:00
}
2018-11-14 18:55:10 -05:00
}
2021-04-03 11:25:19 -04:00
func TestParquetInputSchema ( t * testing . T ) {
os . Setenv ( "MINIO_API_SELECT_PARQUET" , "on" )
defer os . Setenv ( "MINIO_API_SELECT_PARQUET" , "off" )
var testTable = [ ] struct {
requestXML [ ] byte
wantResult string
} {
{
requestXML : [ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT * FROM S3Object LIMIT 5 < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< Parquet >
< / Parquet >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ), wantResult: ` { "shipdate" : "1996-03-13T" }
{ "shipdate" : "1996-04-12T" }
{ "shipdate" : "1996-01-29T" }
{ "shipdate" : "1996-04-21T" }
{ "shipdate" : "1996-03-30T" } ` ,
} ,
{
requestXML : [ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT DATE_ADD ( day , 2 , shipdate ) as shipdate FROM S3Object LIMIT 5 < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< Parquet >
< / Parquet >
< / InputSerialization >
< OutputSerialization >
< JSON >
< / JSON >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ), wantResult: ` { "shipdate" : "1996-03-15T" }
{ "shipdate" : "1996-04-14T" }
{ "shipdate" : "1996-01-31T" }
{ "shipdate" : "1996-04-23T" }
{ "shipdate" : "1996-04T" } ` ,
} ,
}
for i , testCase := range testTable {
t . Run ( fmt . Sprint ( i ) , func ( t * testing . T ) {
getReader := func ( offset int64 , length int64 ) ( io . ReadCloser , error ) {
testdataFile := "testdata/lineitem_shipdate.parquet"
file , err := os . Open ( testdataFile )
if err != nil {
return nil , err
}
fi , err := file . Stat ( )
if err != nil {
return nil , err
}
if offset < 0 {
offset = fi . Size ( ) + offset
}
if _ , err = file . Seek ( offset , io . SeekStart ) ; err != nil {
return nil , err
}
return file , nil
}
s3Select , err := NewS3Select ( bytes . NewReader ( testCase . requestXML ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( getReader ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if ! reflect . DeepEqual ( gotS , testCase . wantResult ) {
t . Errorf ( "received response does not match with expected reply. Query: %s\ngot: %s\nwant:%s" , testCase . requestXML , gotS , testCase . wantResult )
}
} )
}
}
func TestParquetInputSchemaCSV ( t * testing . T ) {
os . Setenv ( "MINIO_API_SELECT_PARQUET" , "on" )
defer os . Setenv ( "MINIO_API_SELECT_PARQUET" , "off" )
var testTable = [ ] struct {
requestXML [ ] byte
wantResult string
} {
{
requestXML : [ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT * FROM S3Object LIMIT 5 < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< Parquet >
< / Parquet >
< / InputSerialization >
< OutputSerialization >
< CSV / >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ), wantResult: ` 1996 - 03 - 13 T
1996 - 04 - 12 T
1996 - 01 - 29 T
1996 - 04 - 21 T
1996 - 03 - 30 T ` ,
} ,
{
requestXML : [ ] byte ( `
< ? xml version = "1.0" encoding = "UTF-8" ? >
< SelectObjectContentRequest >
< Expression > SELECT DATE_ADD ( day , 2 , shipdate ) as shipdate FROM S3Object LIMIT 5 < / Expression >
< ExpressionType > SQL < / ExpressionType >
< InputSerialization >
< CompressionType > NONE < / CompressionType >
< Parquet >
< / Parquet >
< / InputSerialization >
< OutputSerialization >
< CSV / >
< / OutputSerialization >
< RequestProgress >
< Enabled > FALSE < / Enabled >
< / RequestProgress >
< / SelectObjectContentRequest >
` ), wantResult: ` 1996 - 03 - 15 T
1996 - 04 - 14 T
1996 - 01 - 31 T
1996 - 04 - 23 T
1996 - 04 T ` ,
} ,
}
for i , testCase := range testTable {
t . Run ( fmt . Sprint ( i ) , func ( t * testing . T ) {
getReader := func ( offset int64 , length int64 ) ( io . ReadCloser , error ) {
testdataFile := "testdata/lineitem_shipdate.parquet"
file , err := os . Open ( testdataFile )
if err != nil {
return nil , err
}
fi , err := file . Stat ( )
if err != nil {
return nil , err
}
if offset < 0 {
offset = fi . Size ( ) + offset
}
if _ , err = file . Seek ( offset , io . SeekStart ) ; err != nil {
return nil , err
}
return file , nil
}
s3Select , err := NewS3Select ( bytes . NewReader ( testCase . requestXML ) )
if err != nil {
t . Fatal ( err )
}
if err = s3Select . Open ( getReader ) ; err != nil {
t . Fatal ( err )
}
w := & testResponseWriter { }
s3Select . Evaluate ( w )
s3Select . Close ( )
resp := http . Response {
StatusCode : http . StatusOK ,
Body : ioutil . NopCloser ( bytes . NewReader ( w . response ) ) ,
ContentLength : int64 ( len ( w . response ) ) ,
}
res , err := minio . NewSelectResults ( & resp , "testbucket" )
if err != nil {
t . Error ( err )
return
}
got , err := ioutil . ReadAll ( res )
if err != nil {
t . Error ( err )
return
}
gotS := strings . TrimSpace ( string ( got ) )
if ! reflect . DeepEqual ( gotS , testCase . wantResult ) {
t . Errorf ( "received response does not match with expected reply. Query: %s\ngot: %s\nwant:%s" , testCase . requestXML , gotS , testCase . wantResult )
}
} )
}
}
