org.yx.db.event.EventLane Maven / Gradle / Ivy

Go to download

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

A quick developing framewort for internet company

There is a newer version: 4.0.2

/**
 * Copyright (C) 2016 - 2030 youtongluan.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * 		http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.yx.db.event;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.yx.log.Logs;

public final class EventLane {
	private static final ThreadLocal> POOL = new ThreadLocal>() {
		@Override
		protected Map initialValue() {
			return new HashMap(4);
		}

	};
	private final List events = new ArrayList(8);

	public EventLane() {
	}

	private static EventLane pool(Connection conn) {
		return POOL.get().get(conn);
	}

	public static void pubuish(Connection conn, DBEvent event) {
		if (event == null) {
			return;
		}
		if (ModifyEvent.class.isInstance(event) && !isAutoCommit(conn)) {
			EventLane pool = pool(conn);
			if (pool == null) {
				pool = new EventLane();
				POOL.get().put(conn, pool);
			}
			pool.events.add(event);
		} else {
			DBEventPublisher.publish(event);
		}
	}

	private static boolean isAutoCommit(Connection conn) {
		try {
			return conn.getAutoCommit();
		} catch (SQLException e) {
			Logs.db().error("获取连接的autoCommit属性失败", e);
			return false;
		}
	}

	public static void realPubuish(Connection conn) {
		EventLane pool = pool(conn);
		if (pool == null) {
			return;
		}
		for (DBEvent event : pool.events) {
			DBEventPublisher.publish(event);
		}
		pool.events.clear();
	}

	public static void remove(Connection conn) {
		POOL.get().remove(conn);
	}

	public static void removeALL() {
		POOL.remove();
	}
}