com.huaweicloud.dws.client.util.RecordUtil Maven / Gradle / Ivy

Go to download
package com.huaweicloud.dws.client.util;

import com.huaweicloud.dws.client.model.Record;

import java.util.HashMap;
import java.util.Map;

/**
 * @ProjectName: dws-connector
 * @ClassName: RecordUtil
 * @Description: Record 工具
 * @Date: 2023/2/8 10:38
 * @Version: 1.0
 */
public class RecordUtil {

    /**
     * 提取 record中的主键值
     */
    public static Map getRecordPrimaryKeyValue(Record record) {
        Map result = new HashMap<>(record.getColumnBit().size());
        for (String name : record.getTableSchema().getPrimaryKeyNames()) {
            result.put(name, record.getValue(name));
        }
        return result;
    }

    /**
     * 将数据转成json
     */
    public static Map toMap(Record record) {
        Map result = new HashMap<>(record.getColumnBit().size());
        for (String name : record.getTableSchema().getColumnNames()) {
            result.put(name, record.getValue(name));
        }
        return result;
    }
}