be/src/util/mysql_global.h

// 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.

#pragma once

#include <float.h>
#include <stdint.h>

#include <cstring>
#include <string>

namespace doris {

typedef unsigned char uchar;

#define int1store(T, A) *((uint8_t*)(T)) = (uint8_t)(A)
#define int2store(T, A) *((uint16_t*)(T)) = (uint16_t)(A)
#define int3store(T, A)                           \
    do {                                          \
        *(T) = (uchar)((A));                      \
        *(T + 1) = (uchar)(((uint32_t)(A) >> 8)); \
        *(T + 2) = (uchar)(((A) >> 16));          \
    } while (0)
#define int4store(T, A) *((uint32_t*)(T)) = (uint32_t)(A)
#define int8store(T, A) *((int64_t*)(T)) = (uint64_t)(A)
#define float4store(T, A) *((float*)(T)) = (float)(A)
#define float8store(T, A) *((double*)(T)) = (double)(A)

#define MY_ALIGN(A, L) (((A) + (L)-1) & ~((L)-1))
#define SIZEOF_CHARP 8

#define MAX_TINYINT_WIDTH 3     /* Max width for a TINY w.o. sign */
#define MAX_SMALLINT_WIDTH 5    /* Max width for a SHORT w.o. sign */
#define MAX_MEDIUMINT_WIDTH 8   /* Max width for a INT24 w.o. sign */
#define MAX_INT_WIDTH 10        /* Max width for a LONG w.o. sign */
#define MAX_BIGINT_WIDTH 20     /* Max width for a LONGLONG */
#define MAX_LARGEINT_WIDTH 39   /* Max width for a LARGEINT */
#define MAX_CHAR_WIDTH 255      /* Max length for a CHAR column */
#define MAX_BLOB_WIDTH 16777216 /* Default width for blob */
#define MAX_TIME_WIDTH 10       /* Max width for a TIME HH:MM:SS*/
#define MAX_DECPT_FOR_F_FORMAT DBL_DIG
#define MAX_DATETIME_WIDTH 27 /* YYYY-MM-DD HH:MM:SS.ssssss */
#define MAX_DECIMAL_WIDTH 29  /* Max width for a DECIMAL */

/* -[digits].E+## */
#define MAX_FLOAT_STR_LENGTH 24 // see gutil/strings/numbers.h kFloatToBufferSize
/* -[digits].E+### */
#define MAX_DOUBLE_STR_LENGTH 32 // see gutil/strings/numbers.h kDoubleToBufferSize

/* -[digits].[frac] */
#define MAX_DECIMAL_STR_LENGTH 29

inline void direct_write_to_mysql_result_string(std::string& mysql_rows, const char* str,
                                                size_t size) {
    // MySQL protocol length encoding:
    // <= 250: 1 byte for length
    // < 65536: 1 byte (252) + 2 bytes for length
    // < 16777216: 1 byte (253) + 3 bytes for length
    // >= 16777216: 1 byte (254) + 8 bytes for length

    char buf[16];
    if (size < 251ULL) {
        int1store(buf, size);
        mysql_rows.append(buf, 1);
    } else if (size < 65536ULL) {
        buf[0] = static_cast<char>(252);
        uint16_t temp16 = static_cast<uint16_t>(size);
        memcpy(buf + 1, &temp16, sizeof(temp16));
        mysql_rows.append(buf, 3);
    } else if (size < 16777216ULL) {
        buf[0] = static_cast<char>(253);
        int3store(buf + 1, size);
        mysql_rows.append(buf, 4);
    } else {
        buf[0] = static_cast<char>(254);
        uint64_t temp64 = static_cast<uint64_t>(size);
        memcpy(buf + 1, &temp64, sizeof(temp64));
        mysql_rows.append(buf, 9);
    }

    // Append string content
    mysql_rows.append(str, size);
}

inline void direct_write_to_mysql_result_null(std::string& mysql_rows) {
    // MySQL protocol for NULL value is a single byte with value 251
    mysql_rows.push_back(static_cast<char>(251));
}

} // namespace doris