org.tarantool.SqlProtoUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of connector Show documentation
Tarantool client for java
The newest version!
package org.tarantool;

import org.tarantool.jdbc.type.TarantoolSqlType;
import org.tarantool.jdbc.type.TarantoolType;
import org.tarantool.protocol.TarantoolPacket;

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public abstract class SqlProtoUtils {
    public static List> readSqlResult(TarantoolPacket pack) {
        List> data = getSQLData(pack);
        List metaData = getSQLMetadata(pack);

        List> values = new ArrayList<>(data.size());
        for (List row : data) {
            LinkedHashMap value = new LinkedHashMap<>();
            for (int i = 0; i < row.size(); i++) {
                value.put(metaData.get(i).getName(), row.get(i));
            }
            values.add(value);
        }
        return values;
    }

    public static List> getSQLData(TarantoolPacket pack) {
        return (List>) pack.getBody().get(Key.DATA.getId());
    }

    public static List getSQLMetadata(TarantoolPacket pack) {
        List> meta = (List>) pack.getBody().get(Key.SQL_METADATA.getId());
        List values = new ArrayList<>(meta.size());
        for (Map item : meta) {
            values.add(new SQLMetaData(
                (String) item.get(Key.SQL_FIELD_NAME.getId()),
                (String) item.get(Key.SQL_FIELD_TYPE.getId()))
            );
        }
        return values;
    }

    public static Long getSQLRowCount(TarantoolPacket pack) {
        Map info = (Map) pack.getBody().get(Key.SQL_INFO.getId());
        Number rowCount;
        if (info != null && (rowCount = ((Number) info.get(Key.SQL_ROW_COUNT.getId()))) != null) {
            return rowCount.longValue();
        }
        return null;
    }

    public static List getSQLAutoIncrementIds(TarantoolPacket pack) {
        Map info = (Map) pack.getBody().get(Key.SQL_INFO.getId());
        if (info != null) {
            List generatedIds = (List) info.get(Key.SQL_INFO_AUTOINCREMENT_IDS.getId());
            return generatedIds == null ? Collections.emptyList() : generatedIds;
        }
        return Collections.emptyList();
    }

    public static class SQLMetaData {
        private String name;
        private TarantoolSqlType type;

        /**
         * Constructs new SQL metadata based on a raw Tarantool
         * type.
         *
         * Tarantool returns a raw type instead of SQL one.
         * This leads a type mapping ambiguity between raw and
         * SQL types and a default SQL type will be chosen.
         *
         * @param name column name
         * @param tarantoolType raw Tarantool type name
         *
         * @see TarantoolSqlType#getDefaultSqlType(TarantoolType)
         */
        public SQLMetaData(String name, String tarantoolType) {
            this(
                name,
                TarantoolSqlType.getDefaultSqlType(TarantoolType.of(tarantoolType))
            );
        }

        public SQLMetaData(String name, TarantoolSqlType type) {
            this.name = name;
            this.type = type;
        }

        public String getName() {
            return name;
        }

        public TarantoolSqlType getType() {
            return type;
        }

        @Override
        public String toString() {
            return "SQLMetaData{" +
                "name='" + name + '\'' +
                ", type=" + type +
                '}';
        }

    }
}