com.zendesk.maxwell.schema.columndef.ColumnDefWithLength Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of maxwell Show documentation

Maxwell's daemon. Watches mysql, outputs to JSON.

There is a newer version: 1.8.0

package com.zendesk.maxwell.schema.columndef;

import com.zendesk.maxwell.producer.MaxwellOutputConfig;

import java.util.Objects;

public abstract class ColumnDefWithLength extends ColumnDef {
	private Long columnLength;

	protected static ThreadLocal threadLocalBuilder = new ThreadLocal() {
		@Override
		protected StringBuilder initialValue() {
			return new StringBuilder();
		}

		@Override
		public StringBuilder get() {
			StringBuilder b = super.get();
			b.setLength(0);
			return b;
		}
	};

	protected ColumnDefWithLength(String name, String type, short pos, Long columnLength) {
		super(name, type, pos);
		if ( columnLength == null )
			this.columnLength = 0L;
		else
			this.columnLength = columnLength;
	}

	@Override
	public boolean equals(Object o) {
		if (o.getClass() == getClass()) {
			ColumnDefWithLength other = (ColumnDefWithLength)o;
			return super.equals(o)
					&& Objects.equals(columnLength, other.columnLength);
		}
		return false;
	}

	@Override
	public int hashCode() {
		int hash = super.hashCode();
		return 31 * hash + Objects.hash(columnLength);
	}

	@Override
	public String toSQL(Object value) throws ColumnDefCastException {
		return "'" + formatValue(value, new MaxwellOutputConfig()) + "'";
	}

	@Override
	public Object asJSON(Object value, MaxwellOutputConfig config) throws ColumnDefCastException {
		return formatValue(value, config);
	}

	public Long getColumnLength() { return columnLength ; }

	public ColumnDefWithLength withColumnLength(long length) {
		return cloneSelfAndSet(clone -> {
			clone.columnLength = length;
		});
	}

	protected abstract String formatValue(Object value, MaxwellOutputConfig config) throws ColumnDefCastException;

	protected static String appendFractionalSeconds(String value, int nanos, Long columnLength) {
		if ( columnLength == 0L )
			return value;

		// 6 is the max precision of datetime2/time6/timestamp2 in MysQL
		// 3: we go from nano (10^9) to micro (10^6)
		int divideBy = (int) Math.pow(10, 6 + 3 - columnLength);
		int fractional = nanos / divideBy;

		StringBuilder result = threadLocalBuilder.get();
		result.append(value);
		result.append(".");

		String asStr = String.valueOf(fractional);
		for ( int i = 0 ; i < columnLength - asStr.length(); i++ )
			result.append("0");

		result.append(asStr);
		return result.toString();
	}
}