TableCache.java

package com.zendesk.maxwell.replication;

import java.util.HashMap;

import com.google.code.or.binlog.impl.event.TableMapEvent;
import com.zendesk.maxwell.MaxwellFilter;
import com.zendesk.maxwell.schema.Database;
import com.zendesk.maxwell.schema.Schema;
import com.zendesk.maxwell.schema.Table;

public class TableCache {
	private final HashMap<Long, Table> tableMapCache = new HashMap<>();
	private final HashMap<Long, String> blacklistedTableCache = new HashMap<>();

	public void processEvent(Schema schema, MaxwellFilter filter, Long tableId, String dbName, String tblName) {
		if ( !tableMapCache.containsKey(tableId) ) {
			if ( filter != null && filter.isTableBlacklisted(dbName, tblName) ) {
				blacklistedTableCache.put(tableId, tblName);
				return;
			}

			Database db = schema.findDatabase(dbName);
			if ( db == null )
				throw new RuntimeException("Couldn't find database " + dbName);
			else {
				Table tbl = db.findTable(tblName);

				if (tbl == null)
					throw new RuntimeException("Couldn't find table " + tblName + " in database " + dbName);
				else
					tableMapCache.put(tableId, tbl);
			}
		}

	}

	// open-replicator keeps a very similar cache, but we can't get access to it.
	public void processEvent(Schema schema, MaxwellFilter filter, TableMapEvent event) {
		String dbName = new String(event.getDatabaseName().getValue());
		String tblName = new String(event.getTableName().getValue());

		processEvent(schema, filter, event.getTableId(), dbName, tblName);
	}

	public Table getTable(Long tableId) {
		return tableMapCache.get(tableId);
	}

	public boolean isTableBlacklisted(Long tableId) {
		return blacklistedTableCache.containsKey(tableId);
	}

	public String getBlacklistedTableName(Long tableId) {
		return blacklistedTableCache.get(tableId);
	}

	public void clear() {
		tableMapCache.clear();
		blacklistedTableCache.clear();
	}
}