All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.alibaba.hologres.client.model.Record Maven / Gradle / Ivy
/*
* Copyright (c) 2020. Alibaba Group Holding Limited
*/
package com.alibaba.hologres.client.model;
import com.alibaba.hologres.client.Put;
import com.alibaba.hologres.client.Trace;
import org.postgresql.jdbc.ArrayUtil;
import org.postgresql.jdbc.PgArray;
import org.postgresql.util.PGobject;
import java.io.Serializable;
import java.security.InvalidParameterException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.BitSet;
import java.util.List;
import java.util.concurrent.CompletableFuture;
/**
* Class to represent a record.
*/
public class Record implements Serializable {
/**
* schema never change.
*/
protected final TableSchema schema;
/**
* for partition table, the tableName is partition child table name.
*/
TableName tableName;
Object[] values;
BitSet bitSet;
BitSet onlyInsertColumnSet;
protected List attachmentList = null;
Put.MutationType type = Put.MutationType.INSERT;
int shardId = -1;
long byteSize = 0;
/**
* 只在put场景下使用，存储这个Record对应的所有put的future.
* 例：
* 当前后2个Put请求的主键相同时，那么两个Put对象的Record会合并成1个，那么合并后的Record的putFutures变量将会保留2个Put对象的future
*/
protected transient List> putFutures;
private Record(TableSchema schema,
TableName tableName,
Object[] values,
BitSet bitSet,
BitSet onlyInsertColumnSet,
List attachmentList,
Put.MutationType type,
long byteSize,
int shardId) {
this.schema = schema;
this.tableName = tableName;
this.values = values;
this.bitSet = bitSet;
this.onlyInsertColumnSet = onlyInsertColumnSet;
this.attachmentList = attachmentList;
this.type = type;
this.byteSize = byteSize;
this.shardId = shardId;
}
public Record clone() {
return new Record(schema,
tableName,
values.clone(),
(BitSet) bitSet.clone(),
(BitSet) onlyInsertColumnSet.clone(),
null,
type,
byteSize,
shardId);
}
Trace trace;
public void startTrace() {
trace = new Trace();
trace.begin();
}
public void stepTrace(String name) {
if (trace != null) {
trace.step(name);
}
}
public Trace getTrace() {
return trace;
}
public Object[] getValues() {
return values;
}
public Put.MutationType getType() {
return type;
}
public void setType(Put.MutationType type) {
this.type = type;
}
public TableSchema getSchema() {
return schema;
}
public boolean isAllColumnSet() {
return bitSet.stream().count() == schema.getColumnSchema().length;
}
public static Record build(TableSchema schema) {
return new Record(schema);
}
public Record(TableSchema schema) {
this.schema = schema;
this.tableName = schema.getTableNameObj();
bitSet = new BitSet(schema.getColumnSchema().length);
onlyInsertColumnSet = new BitSet(schema.getColumnSchema().length);
values = new Object[schema.getColumnSchema().length];
}
public boolean isSet(int index) {
return bitSet.get(index);
}
public long getByteSize() {
return byteSize;
}
private long getObjByteSize(int index, Object obj) {
if (obj == null) {
return 4;
}
long ret = 0;
Column column = schema.getColumnSchema()[index];
switch (column.getType()) {
case Types.BOOLEAN:
case Types.TINYINT:
case Types.BIT:
ret = 1;
break;
case Types.SMALLINT:
ret = 2;
break;
case Types.BIGINT:
case Types.DOUBLE:
ret = 8;
break;
case Types.TIMESTAMP:
case Types.TIME_WITH_TIMEZONE:
ret = 12;
break;
case Types.NUMERIC:
case Types.DECIMAL:
ret = 24;
break;
case Types.CHAR:
case Types.VARCHAR:
ret = String.valueOf(obj).length();
break;
case Types.ARRAY:
if (obj instanceof int[]) {
ret = ((int[]) obj).length * 4L;
} else if (obj instanceof long[]) {
ret = ((long[]) obj).length * 8L;
} else if (obj instanceof float[]) {
ret = ((float[]) obj).length * 4L;
} else if (obj instanceof double[]) {
ret = ((double[]) obj).length * 8L;
} else if (obj instanceof boolean[]) {
ret = ((boolean[]) obj).length;
} else if (obj instanceof String[]) {
ret = ArrayUtil.getArrayLength((String[]) obj);
} else if (obj instanceof Object[]) {
ret = ArrayUtil.getArrayLength((Object[]) obj, column.getTypeName());
} else if (obj instanceof List) {
ret = ArrayUtil.getArrayLength((List) obj, column.getTypeName());
} else if (obj instanceof PgArray) {
ret = ArrayUtil.getArrayLength((PgArray) obj);
} else {
ret = 1024;
}
break;
default:
if ("json".equalsIgnoreCase(column.getTypeName()) || "jsonb".equalsIgnoreCase(column.getTypeName())) { // json, jsonb 等类型
ret = String.valueOf(obj).length();
break;
}
if (obj instanceof PGobject) { // PGmoney 等类型
PGobject pObj = (PGobject) obj;
if (pObj.getValue() != null) {
ret = pObj.getValue().length();
}
} else if (obj instanceof byte[]) { // RoaringBitmap, bytea 等类型
ret = ((byte[]) obj).length;
} else {
ret = 4;
}
}
return ret;
}
public TableName getTableName() {
return tableName;
}
public void setObject(int index, Object obj) {
Object old = values[index];
long minus = 0L;
if (isSet(index)) {
minus = getObjByteSize(index, old);
}
long add = getObjByteSize(index, obj);
byteSize = byteSize + add - minus;
values[index] = obj;
bitSet.set(index);
}
public Object getObject(int index) {
return values[index];
}
public Object getObject(String columnName) {
Integer index = schema.getColumnIndex(columnName);
if (index == null) {
throw new InvalidParameterException("can not found column named " + columnName);
}
return values[index];
}
public int[] getKeyIndex() {
return schema.getKeyIndex();
}
public int getSize() {
return schema.getColumnSchema().length;
}
public int getLength() {
return (int) bitSet.stream().count();
}
public BitSet getBitSet() {
return bitSet;
}
public BitSet getOnlyInsertColumnSet() {
return onlyInsertColumnSet;
}
public void setPutFuture(CompletableFuture future) {
if (putFutures != null) {
throw new RuntimeException("setPutFuture should call ONLY ONCE");
}
putFutures = new ArrayList<>(2);
putFutures.add(future);
}
/**
* a.merge(b).
* a的值被b"覆盖"（只有b set过值会覆盖到a上），a.attachmentList=a.attachmentList+b.attachmentList，b盖在a上面.
*
* @param record
*/
public void merge(Record record) {
if (!schema.equals(record.schema)) {
throw new InvalidParameterException("schema not match");
}
for (int i = 0; i < record.getSize(); ++i) {
if (record.isSet(i) && !record.getOnlyInsertColumnSet().get(i)) {
setObject(i, record.getObject(i));
}
}
//merge attachment
this.addAttachmentList(record.attachmentList);
//merge putFutures
if (this.putFutures == null) {
if (record.putFutures != null) {
this.putFutures = new ArrayList<>(record.putFutures);
}
} else {
if (record.putFutures != null) {
this.putFutures.addAll(record.putFutures);
}
}
}
/**
* a.cover(b).
* a把B覆盖（彻彻底底的全覆盖,不需要b的任何字段）掉，a.attachmentList=b.attachmentList+a.attachmentList，a盖在b上面.
*
* @param record
*/
public void cover(Record record) {
if (!schema.equals(record.schema)) {
throw new InvalidParameterException("schema not match");
}
//cover attachment
if (record.getAttachmentList() != null) {
List origins = record.getAttachmentList();
if (this.getAttachmentList() != null) {
origins.addAll(this.getAttachmentList());
}
this.setAttachmentList(origins);
}
//cover putFutures
if (record.putFutures != null) {
List> origins = record.putFutures;
if (this.putFutures != null) {
origins.addAll(this.putFutures);
}
this.putFutures = origins;
}
}
public List> getPutFutures() {
return putFutures;
}
public List getAttachmentList() {
return attachmentList;
}
public void setAttachmentList(List origin) {
this.attachmentList = origin;
}
public void addAttachment(Object attachment) {
if (attachment == null) {
return;
}
if (this.attachmentList == null) {
this.attachmentList = new ArrayList<>(1);
}
this.attachmentList.add(attachment);
}
public void addAttachmentList(List list) {
if (list == null || list.size() == 0) {
return;
}
if (this.attachmentList == null) {
this.attachmentList = new ArrayList<>(list);
} else {
this.attachmentList.addAll(list);
}
}
public void changeToChildSchema(TableSchema schema) {
this.tableName = schema.getTableNameObj();
}
/**
* 设置Binlog Record所对应的shardId.
*/
public void setShardId(int shardId) {
this.shardId = shardId;
}
/*
* -1 表示未设置.
*
* @return
*/
public int getShardId() {
return shardId;
}
@Override
public String toString() {
return "Record{" +
"schema=" + schema +
", values=" + Arrays.toString(values) +
", bitSet=" + bitSet +
'}';
}
}
