hibernate 监听器记录数据变化信息

package com.ezparking.web;

import com.ezparking.model.CustomerServe;
import com.ezparking.model.UserDataLog;
import com.ezparking.util.WebUtils;
import org.hibernate.event.internal.DefaultLoadEventListener;
import org.hibernate.event.service.spi.EventListenerRegistry;
import org.hibernate.event.spi.*;
import org.hibernate.internal.SessionFactoryImpl;
import org.hibernate.persister.entity.EntityPersister;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import javax.annotation.PostConstruct;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.servlet.http.HttpServletRequest;
import java.sql.Timestamp;
import java.util.Optional;

@Component
public class UserDataLogListener extends DefaultLoadEventListener implements PostUpdateEventListener, PostInsertEventListener, PostDeleteEventListener {

	private static final String[] OBJECTS_SKIPPED = {"PoiStatusLog", "CustomerLog", "UserLog", "UserDataLog", "UserCoordinate"};

	private static final Logger logger = LoggerFactory.getLogger(UserDataLogListener.class);

	@Autowired
	private EntityManagerFactory entityManagerFactory;

	@PostConstruct
	public void registerListeners() {
		logger.info("register listeners ******");
		SessionFactoryImpl sessionFactoryImpl = entityManagerFactory.unwrap(SessionFactoryImpl.class);
		EventListenerRegistry eventListenerRegistry = sessionFactoryImpl.getServiceRegistry().getService(EventListenerRegistry.class);
		eventListenerRegistry.getEventListenerGroup(EventType.POST_INSERT).appendListener(this);
		eventListenerRegistry.getEventListenerGroup(EventType.POST_UPDATE).appendListener(this);
		eventListenerRegistry.getEventListenerGroup(EventType.POST_DELETE).appendListener(this);
	}

	@Override
	public void onPostInsert(PostInsertEvent event) {
		logger.info("insert start***********");

		if (!requiresLog(event.getEntity().getClass())) {
			return;
		}

		if (StringUtils.isEmpty(getUserId()) || StringUtils.isEmpty(getLogId())) {
			return;
		}
		logger.debug(event.getEntity() + " inserted");

		long startTime = System.currentTimeMillis();
		StringBuilder fields = new StringBuilder();
		StringBuilder afterValues = new StringBuilder();
		StringBuilder remark = new StringBuilder();

		for (int i = 0; i < event.getState().length; i++) {
			fields.append(event.getPersister().getPropertyNames()[i] + "|");
			afterValues.append(event.getState()[i] + "|");
			remark.append(event.getPersister().getPropertyNames()[i] + ": ");
			remark.append(event.getState()[i] + "\r\n");
		}

		UserDataLog userDataLog = new UserDataLog();
		userDataLog.setUserId(getUserId());
		userDataLog.setLogId(getLogId());
		userDataLog.setFields(fields.toString());
		userDataLog.setAfterValues(afterValues.toString());
		userDataLog.setRemark(remark.toString());
		userDataLog.setObject(event.getEntity().getClass().getSimpleName());
		userDataLog.setObjectKey(event.getId().toString());
		userDataLog.setOperation("INSERT");
		userDataLog.setLogTime(new Timestamp(System.currentTimeMillis()));

		this.persistLog(userDataLog);

		long endTime = System.currentTimeMillis();
		logger.debug("日志时间： " + (endTime - startTime) + "ms");
	}

	@Override
	public void onPostUpdate(PostUpdateEvent event) {
		logger.info("update start***********");
		if (!requiresLog(event.getEntity().getClass())) {
			return;
		}

		if (StringUtils.isEmpty(getUserId()) || StringUtils.isEmpty(getLogId())) {
			return;
		}
		logger.debug(event.getEntity() + " updated");

		long startTime = System.currentTimeMillis();
		StringBuilder fields = new StringBuilder();
		StringBuilder beforeValues = new StringBuilder();
		StringBuilder afterValues = new StringBuilder();
		StringBuilder remark = new StringBuilder();

		for (int i = 0; i < event.getState().length; i++) {
			// Object oldValue = event.getOldState()[i];
			// Object newValue = event.getState()[i];
			// String propertyName = event.getPersister().getPropertyNames()[i];
			// logger.info(propertyName + ": " + oldValue + " -> " + newValue);
			fields.append(event.getPersister().getPropertyNames()[i] + "|");
			beforeValues.append(event.getOldState()[i] + "|");
			afterValues.append(event.getState()[i] + "|");
		}

		for (int i = 0; i < event.getDirtyProperties().length; i++) {
			int j = event.getDirtyProperties()[i];
			remark.append(event.getPersister().getPropertyNames()[j] + ": ");
			remark.append(event.getOldState()[j] + " -> ");
			remark.append(event.getState()[j] + "\r\n");
		}

		UserDataLog userDataLog = new UserDataLog();
		userDataLog.setUserId(getUserId());
		userDataLog.setLogId(getLogId());
		userDataLog.setFields(fields.toString());
		userDataLog.setBeforeValues(beforeValues.toString());
		userDataLog.setAfterValues(afterValues.toString());
		//dl.setFields(serialize(event.getPersister().getPropertyNames()));
		//dl.setBeforeValues(serialize(event.getOldState()));
		//dl.setAfterValues(serialize(event.getState()));
		userDataLog.setRemark(remark.toString());
		userDataLog.setObject(event.getEntity().getClass().getSimpleName());
		userDataLog.setObjectKey(event.getId().toString());
		userDataLog.setOperation("UPDATE");
		userDataLog.setLogTime(new Timestamp(System.currentTimeMillis()));

		this.persistLog(userDataLog);

		long endTime = System.currentTimeMillis();
		logger.debug("日志时间： " + (endTime - startTime) + "ms");
	}

	@Override
	public void onPostDelete(PostDeleteEvent event) {
		logger.info("delete start***********");
		if (!requiresLog(event.getEntity().getClass())) {
			return;
		}

		if (StringUtils.isEmpty(getUserId()) || StringUtils.isEmpty(getLogId())) {
			return;
		}
		logger.debug(event.getEntity() + " removed");

		long startTime = System.currentTimeMillis();
		StringBuilder fields = new StringBuilder();
		StringBuilder beforeValues = new StringBuilder();
		UserDataLog userDataLog = new UserDataLog();
		userDataLog.setUserId(getUserId());
		userDataLog.setLogId(getLogId());
		userDataLog.setFields(fields.toString());
		userDataLog.setBeforeValues(beforeValues.toString());
		userDataLog.setObject(event.getEntity().getClass().getSimpleName());
		userDataLog.setObjectKey(event.getId().toString());
		userDataLog.setOperation("REMOVE");
		userDataLog.setLogTime(new Timestamp(System.currentTimeMillis()));
		this.persistLog(userDataLog);
		long endTime = System.currentTimeMillis();
		logger.debug("日志时间： " + (endTime - startTime) + "ms");
	}

	private void persistLog(UserDataLog userDataLog) {
		EntityManager em = entityManagerFactory.createEntityManager();
		try {
			em.getTransaction().begin();
			em.persist(userDataLog);
			em.getTransaction().commit();
		} finally {
			em.close();
		}
	}

	@Override
	public boolean requiresPostCommitHanding(EntityPersister persister) {
		return false;
	}

	private boolean requiresLog(Class<?> clazz) {
		for (String clsName : OBJECTS_SKIPPED) {
			if (clsName.equals(clazz.getSimpleName())) {
				return false;
			}
		}
		return true;
	}

	private Integer getLogId() {
		HttpServletRequest request = WebUtils.getHttpRequest();
		if (request == null) {
			return null;
		}
		Integer logId = (Integer) request.getSession().getAttribute("logId");
		return logId;
	}

	private String getUserId() {
		HttpServletRequest request = WebUtils.getHttpRequest();
		if (request == null) {
			return null;
		}
		CustomerServe customerServe = (CustomerServe) request.getAttribute("serve");
		if (customerServe == null || StringUtils.isEmpty(customerServe.getApp())) {
			return null;
		}

		return Optional.ofNullable(customerServe.getDeviceId()).orElse(customerServe.getLinkedKey());
	}
}