Thanks to visit codestin.com
Credit goes to github.com

Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,12 @@ public void setBinlogConf(BinlogConf binlogConf) {
}

public void setRowConverter(AbstractCDCRowConverter rowConverter) {
setRowConverter(rowConverter, false);
}

public void setRowConverter(AbstractCDCRowConverter rowConverter, boolean useAbstractColumn) {
this.format.setRowConverter(rowConverter);
format.setUseAbstractColumn(useAbstractColumn);
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ public DataStream<RowData> createSource() {
: TimestampFormat.ISO_8601;
rowConverter = new BinlogRowConverter(rowType, format);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createInput(builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,8 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
List<FieldConf> fieldConfList = sinkConf.getColumn();

final RowType rowType = TableUtil.createRowType(fieldConfList, getRawTypeConverter());
builder.setRowConverter(new CassandraColumnConverter(rowType, fieldConfList));
builder.setRowConverter(
new CassandraColumnConverter(rowType, fieldConfList), useAbstractBaseColumn);

return createOutput(dataSet, builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,8 @@ public DataStream<RowData> createSource() {
fieldConfList.forEach(fieldConf -> columnNameList.add(fieldConf.getName()));

final RowType rowType = TableUtil.createRowType(fieldConfList, getRawTypeConverter());
builder.setRowConverter(new CassandraRowConverter(rowType, columnNameList));
builder.setRowConverter(
new CassandraRowConverter(rowType, columnNameList), useAbstractBaseColumn);

return createInput(builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
} else {
rowConverter = new ElasticsearchRowConverter(rowType);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ public DataStream<RowData> createSource() {
} else {
rowConverter = new ElasticsearchRowConverter(rowType);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createInput(builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
}
EmqxOutputFormatBuilder builder = new EmqxOutputFormatBuilder();
builder.setEmqxConf(emqxConf);
builder.setRowConverter(new EmqxColumnConverter(emqxConf));
builder.setRowConverter(new EmqxColumnConverter(emqxConf), useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ public DataStream<RowData> createSource() {
}
EmqxInputFormatBuilder builder = new EmqxInputFormatBuilder();
builder.setEmqxConf(emqxConf);
builder.setRowConverter(new EmqxColumnConverter(emqxConf));
builder.setRowConverter(new EmqxColumnConverter(emqxConf), useAbstractBaseColumn);
return createInput(builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ protected ISerializationConverter<List<String>> wrapIntoNullableExternalConverte
ISerializationConverter serializationConverter, FieldConf fieldConf) {
return (rowData, index, list) -> {
if (rowData == null || rowData.isNullAt(index)) {
list.set(index, null);
list.add(index, null);
} else {
serializationConverter.serialize(rowData, index, list);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
ftpConfig.setColumn(fieldConfList);
final RowType rowType =
TableUtil.createRowType(ftpConfig.getColumn(), getRawTypeConverter());
builder.setRowConverter(new FtpColumnConverter(rowType, ftpConfig));
builder.setRowConverter(new FtpColumnConverter(rowType, ftpConfig), useAbstractBaseColumn);

return createOutput(dataSet, builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ public DataStream<RowData> createSource() {
ftpConfig.setColumn(fieldConfList);
final RowType rowType =
TableUtil.createRowType(ftpConfig.getColumn(), getRawTypeConverter());
builder.setRowConverter(new FtpColumnConverter(rowType, ftpConfig));
builder.setRowConverter(new FtpColumnConverter(rowType, ftpConfig), useAbstractBaseColumn);

return createInput(builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
rowConverter = new HbaseRowConverter(hbaseSchema, nullStringLiteral);
}

builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter,useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ public DataStream<RowData> createSource() {
rowConverter = new HbaseRowConverter(hbaseSchema, nullStringLiteral);
}

builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter,useAbstractBaseColumn);
return createInput(builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,6 @@
import com.dtstack.chunjun.connector.hdfs.util.HdfsUtil;
import com.dtstack.chunjun.enums.ColumnType;
import com.dtstack.chunjun.enums.SizeUnitType;
import com.dtstack.chunjun.metrics.RowSizeCalculator;
import com.dtstack.chunjun.throwable.ChunJunRuntimeException;
import com.dtstack.chunjun.throwable.WriteRecordException;
import com.dtstack.chunjun.util.ColumnTypeUtil;
Expand Down Expand Up @@ -77,11 +76,6 @@ public class HdfsOrcOutputFormat extends BaseHdfsOutputFormat {

protected int[] colIndices;

/** 初始化对象大小计算器 */
protected void initRowSizeCalculator() {
rowSizeCalculator = RowSizeCalculator.getRowSizeCalculator();
}

@Override
@SuppressWarnings("unchecked")
protected void openSource() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
hdfsConf.getColumn(),
getRawTypeConverter());

builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ public DataStream<RowData> createSource() {
hdfsConf.getColumn(),
getRawTypeConverter());

builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createInput(builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,8 @@ private BaseHdfsOutputFormat createHdfsOutputFormat(
useAbstractBaseColumn,
copyHiveConf.getFileType(),
fieldConfList,
HdfsRawTypeConverter::apply));
HdfsRawTypeConverter::apply),
useAbstractBaseColumn);
builder.setInitAccumulatorAndDirty(false);

BaseHdfsOutputFormat outputFormat = (BaseHdfsOutputFormat) builder.finish();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ public DataStream<RowData> createSource() {
builder.setMetaHeaders(httpRestConfig.getHeader());
builder.setMetaParams(httpRestConfig.getParam());
builder.setMetaBodys(httpRestConfig.getBody());
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createInput(builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
builder.setDatabase(influxdbConfig.getDatabase());
builder.setMeasurement(influxdbConfig.getMeasurement());
builder.setEnableBatch(influxdbConfig.isEnableBatch());
builder.setRowConverter(null, useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,7 @@ public RawTypeConverter getRawTypeConverter() {
public DataStream<RowData> createSource() {
InfluxdbInputFormatBuilder builder = new InfluxdbInputFormatBuilder();
builder.setInfluxdbConfig(config);
builder.setRowConverter(null, useAbstractBaseColumn);
return createInput(builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
TableUtil.createRowType(jdbcConf.getColumn(), getRawTypeConverter());
rowConverter = jdbcDialect.getRowConverter(rowType);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);

return createOutput(dataSet, builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ public DataStream<RowData> createSource() {
TableUtil.createRowType(jdbcConf.getColumn(), getRawTypeConverter());
rowConverter = jdbcDialect.getRowConverter(rowType);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);

return createInput(builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,8 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
fieldConfList.forEach(field -> columnNames.add(field.getName()));

final RowType rowType = TableUtil.createRowType(fieldConfList, getRawTypeConverter());
builder.setRowConverter(new KuduColumnConverter(rowType, columnNames));
builder.setRowConverter(
new KuduColumnConverter(rowType, columnNames), useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,8 @@ public DataStream<RowData> createSource() {
List<String> columnNameList = new ArrayList<>();
fieldConfList.forEach(fieldConf -> columnNameList.add(fieldConf.getName()));

builder.setRowConverter(new KuduRowConverter(rowType, columnNameList));
builder.setRowConverter(
new KuduRowConverter(rowType, columnNameList), useAbstractBaseColumn);

return createInput(builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
} else {
converter = mongoConverterFactory.createRowConverter();
}
builder.setRowConverter(converter);
builder.setRowConverter(converter, useAbstractBaseColumn);
builder.setConfig(mongodbDataSyncConf);
return createOutput(dataSet, builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ public DataStream<RowData> createSource() {
} else {
converter = mongoConverterFactory.createRowConverter();
}
builder.setRowConverter(converter);
builder.setRowConverter(converter, useAbstractBaseColumn);
builder.setConfig(mongodbDataSyncConf);
return createInput(builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,13 @@ public void setConf(PGWalConf conf) {
}

public void setRowConverter(AbstractCDCRowConverter rowConverter) {
setRowConverter(rowConverter, false);
}

public void setRowConverter(
AbstractCDCRowConverter rowConverter, boolean useAbstractBaseColumn) {
this.format.setRowConverter(rowConverter);
this.format.setUseAbstractColumn(useAbstractBaseColumn);
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ public DataStream<RowData> createSource() {
TableUtil.createRowType(conf.getColumn(), getRawTypeConverter());
rowConverter = new PGWalRowConverter(rowType, TimestampFormat.SQL);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createInput(builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
}
RedisOutputFormatBuilder builder = new RedisOutputFormatBuilder();
builder.setRedisConf(redisConf);
builder.setRowConverter(new RedisColumnConverter(redisConf));
builder.setRowConverter(new RedisColumnConverter(redisConf), useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
} else {
converter = converterFactory.createRowConverter();
}
builder.setRowConverter(converter);
builder.setRowConverter(converter, useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ public DataStream<RowData> createSource() {
} else {
converter = converterFactory.createRowConverter();
}
builder.setRowConverter(converter);
builder.setRowConverter(converter, useAbstractBaseColumn);
return createInput(builder.finish());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,13 @@ public void setSqlServerCdcConf(SqlServerCdcConf sqlServerCdcConf) {
}

public void setRowConverter(AbstractCDCRowConverter rowConverter) {
setRowConverter(rowConverter, false);
}

public void setRowConverter(
AbstractCDCRowConverter rowConverter, boolean useAbstractBaseColumn) {
this.format.setRowConverter(rowConverter);
this.format.setUseAbstractColumn(useAbstractBaseColumn);
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ public DataStream<RowData> createSource() {
: TimestampFormat.ISO_8601;
rowConverter = new SqlServerCdcRowConverter(rowType, format);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);
return createInput(builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
List<FieldConf> fieldList = syncConf.getWriter().getFieldList();
typeInformation = TableUtil.getTypeInformation(fieldList, getRawTypeConverter());
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);

builder.finish().initializeGlobal(1);

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ public DataStreamSink<RowData> createSink(DataStream<RowData> dataSet) {
converter = new StreamRowConverter(rowType);
}

builder.setRowConverter(converter);
builder.setRowConverter(converter, useAbstractBaseColumn);
return createOutput(dataSet, builder.finish());
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ public DataStream<RowData> createSource() {
TableUtil.createRowType(streamConf.getColumn(), getRawTypeConverter());
rowConverter = new StreamRowConverter(rowType);
}
builder.setRowConverter(rowConverter);
builder.setRowConverter(rowConverter, useAbstractBaseColumn);

return createInput(builder.finish());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,11 @@
public abstract class AbstractBaseColumn implements Serializable {
private static final long serialVersionUID = 1L;
protected Object data;
protected int byteSize;

public AbstractBaseColumn(final Object data) {
public AbstractBaseColumn(final Object data, int byteSize) {
this.data = data;
this.byteSize = byteSize;
}

/**
Expand Down Expand Up @@ -181,6 +183,10 @@ public void setData(Object data) {
this.data = data;
}

public int getByteSize() {
return byteSize;
};

@Override
public String toString() {
return this.data == null ? "" : this.asString();
Expand Down
Loading