com.zendesk.maxwell.schema.columndef.SetColumnDef 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 java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import com.zendesk.maxwell.producer.MaxwellOutputConfig;
import org.apache.commons.lang3.StringUtils;

public class SetColumnDef extends EnumeratedColumnDef {
	private SetColumnDef(String name, String type, short pos, String[] enumValues) {
		super(name, type, pos, enumValues);
	}

	public static SetColumnDef create(String name, String type, short pos, String[] enumValues) {
		SetColumnDef temp = new SetColumnDef(name, type, pos, enumValues);
		return (SetColumnDef) INTERNER.intern(temp);
	}

	@Override
	public String toSQL(Object value) throws ColumnDefCastException {
		return "'" + StringUtils.join(asList(value), "'") + "'";
	}

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

	private ArrayList asList(Object value) throws ColumnDefCastException {
		if ( value instanceof String ) {
			return new ArrayList<>(Arrays.asList((( String ) value).split(",")));
		} else if ( value instanceof Long ) {
			ArrayList values = new ArrayList<>();
			long v = (Long) value;
			List enumValues = getEnumValues();
			for (int i = 0; i < enumValues.size(); i++) {
				if (((v >> i) & 1) == 1) {
					values.add(enumValues.get(i));
				}
			}
			return values;
		} else {
			throw new ColumnDefCastException(this, value);
		}
	}
}