ed3ff470ce
This is part of the array type support and has not been fully completed. The following functions are implemented 1. fe array type support and implementation of array function, support array syntax analysis and planning 2. Support import array type data through insert into 3. Support select array type data 4. Only the array type is supported on the value lie of the duplicate table this pr merge some code from #4655 #4650 #4644 #4643 #4623 #2979
126 lines
4.1 KiB
C++
126 lines
4.1 KiB
C++
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you 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
|
|
//
|
|
// http://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.
|
|
|
|
#ifndef DORIS_BE_SRC_QUERY_MYSQL_MYSQL_ROW_BUFFER_H
|
|
#define DORIS_BE_SRC_QUERY_MYSQL_MYSQL_ROW_BUFFER_H
|
|
|
|
#include <stdint.h>
|
|
|
|
namespace doris {
|
|
|
|
/**
|
|
// Now only support text protocol
|
|
* helper for construct MySQL send row
|
|
* The MYSQL protocal:
|
|
*
|
|
* | flag | (length) | value | flag | (length) | value | ......
|
|
* <--------A column--------><--------A column--------><-.....->
|
|
*
|
|
* The flag means value's length or null value:
|
|
* If value is NULL, flag is 251
|
|
* If value's length < 251, flag is the value's length
|
|
* If 251 <= value's length < 65536, flag is 252 and the next two byte is length
|
|
* If 65536 <= value's length < 16777216 , flag is 253 and the next three byte is length
|
|
* If 16777216 <= value's length, flag is 254 and the next eighth byte is length
|
|
*
|
|
* the code example:
|
|
* mrb.push_null();
|
|
* mrb.push_tinyint(5);
|
|
* mrb.push_int(120);
|
|
* mrb.push_string("...my length is 65536...");
|
|
*
|
|
* the protocol contents:
|
|
*
|
|
* 251-1-'5'-3-'120'-253-65536-"...my length is 65536..."
|
|
*
|
|
*/
|
|
class MysqlRowBuffer {
|
|
public:
|
|
MysqlRowBuffer();
|
|
~MysqlRowBuffer();
|
|
|
|
void reset() { _pos = _buf; }
|
|
|
|
// TODO(zhaochun): add signed/unsigned support
|
|
int push_tinyint(int8_t data);
|
|
int push_smallint(int16_t data);
|
|
int push_int(int32_t data);
|
|
int push_bigint(int64_t data);
|
|
int push_unsigned_bigint(uint64_t data);
|
|
int push_float(float data);
|
|
int push_double(double data);
|
|
int push_string(const char* str, int length);
|
|
int push_null();
|
|
|
|
// this function reserved size, change the pos step size, return old pos
|
|
// Becareful when use the returned pointer.
|
|
char* reserved(int size);
|
|
|
|
const char* buf() const { return _buf; }
|
|
const char* pos() const { return _pos; }
|
|
int length() const { return _pos - _buf; }
|
|
|
|
/**
|
|
* Why?
|
|
* Because the Nested-Type's data need pushed multiple times, but mysql protocol don't
|
|
* support nested type and each push will decide a column data.
|
|
*
|
|
* How?
|
|
* Dynamic mode allow we push data in a column multiple times, and allow recursive push.
|
|
* We will think that the length of the next column is uncertain when open dynamic
|
|
* mode, so we will set the flag to 254(longest flag) and skip 8 bytes which used for
|
|
* record length, then compute the actual data length when close dynamic mode.
|
|
* In a recursive call(special for nested type), the mode will open multiple times, but
|
|
* the data is actually written in one column, so we only need to deal it at the beginning
|
|
* and at the end.
|
|
*
|
|
* the code:
|
|
* mrb.push_tinyint(5);
|
|
* mrb.push_smallint(120);
|
|
* mrb.push_int(-30000);
|
|
*
|
|
* In normal mode, the buffer contains three column:
|
|
* 1-'5'-3-'120'-6-'-30000'
|
|
*
|
|
* Same code in dynamic mode, the buffer contains a column:
|
|
* 254-48-'5'-'120'-'-30000'
|
|
*
|
|
* NOTE: The open_dynamic_mode() and close_dynamic_mode() need appear in pairs
|
|
*/
|
|
void open_dynamic_mode();
|
|
|
|
/**
|
|
* NOTE: The open_dynamic_mode() and close_dynamic_mode() need appear in pairs
|
|
*/
|
|
void close_dynamic_mode();
|
|
|
|
private:
|
|
int reserve(int size);
|
|
|
|
char* _pos;
|
|
char* _buf;
|
|
int _buf_size;
|
|
char _default_buf[4096];
|
|
|
|
int _dynamic_mode;
|
|
char* _len_pos;
|
|
};
|
|
|
|
} // namespace doris
|
|
|
|
#endif // DORIS_BE_SRC_QUERY_MYSQL_MYSQL_ROW_BUFFER_H
|