tickettracker feature

src/main/java/love/sola/netsupport/sql/SQLCore.java

@@ -2,14 +2,20 @@ package love.sola.netsupport.sql;
 
 import com.google.gson.*;
 import com.google.gson.annotations.Expose;
+import com.google.gson.reflect.TypeToken;
+import com.google.gson.stream.JsonReader;
+import com.google.gson.stream.JsonWriter;
 import love.sola.netsupport.enums.ISP;
+import org.hibernate.Hibernate;
 import org.hibernate.SessionFactory;
 import org.hibernate.boot.MetadataSources;
 import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
+import org.hibernate.proxy.HibernateProxy;
 import org.hibernate.service.ServiceRegistry;
 
 import javax.naming.InitialContext;
 import javax.sql.DataSource;
+import java.io.IOException;
 import java.util.Date;
 
 /**
@@ -50,6 +56,7 @@ public class SQLCore {
 			.registerTypeAdapter(Date.class, (JsonSerializer<Date>) (src, typeOfSrc, context) -> new JsonPrimitive(src.getTime()))
 			.registerTypeAdapter(ISP.class, (JsonDeserializer<ISP>) (json, typeOfT, context) -> ISP.fromId(json.getAsJsonPrimitive().getAsInt()))
 			.registerTypeAdapter(ISP.class, (JsonSerializer<ISP>) (src, typeOfSrc, context) -> new JsonPrimitive(src.id))
+			.registerTypeAdapterFactory(HibernateProxyTypeAdapter.FACTORY)
 			.create();
 	public static SessionFactory sf;
 	public static ServiceRegistry sr;
@@ -69,4 +76,43 @@ public class SQLCore {
 		}
 	}
 
+	public static class HibernateProxyTypeAdapter extends TypeAdapter<HibernateProxy> {
+
+		public static final TypeAdapterFactory FACTORY = new TypeAdapterFactory() {
+			@Override
+			@SuppressWarnings("unchecked")
+			public <T> TypeAdapter<T> create(Gson gson, TypeToken<T> type) {
+				return (HibernateProxy.class.isAssignableFrom(type.getRawType()) ? (TypeAdapter<T>) new HibernateProxyTypeAdapter(gson) : null);
+			}
+		};
+		private final Gson context;
+
+		private HibernateProxyTypeAdapter(Gson context) {
+			this.context = context;
+		}
+
+		@Override
+		public HibernateProxy read(JsonReader in) throws IOException {
+			throw new UnsupportedOperationException("Not supported");
+		}
+
+		@SuppressWarnings({"rawtypes", "unchecked"})
+		@Override
+		public void write(JsonWriter out, HibernateProxy value) throws IOException {
+			if (value == null) {
+				out.nullValue();
+				return;
+			}
+			// Retrieve the original (not proxy) class
+			Class<?> baseType = Hibernate.getClass(value);
+			// Get the TypeAdapter of the original class, to delegate the serialization
+			TypeAdapter delegate = context.getAdapter(TypeToken.get(baseType));
+			// Get a filled instance of the original class
+			Object unproxiedValue = ((HibernateProxy) value).getHibernateLazyInitializer()
+					.getImplementation();
+			// Serialize the value
+			delegate.write(out, unproxiedValue);
+		}
+	}
+
 }

src/main/java/love/sola/netsupport/sql/TableTicket.java

@@ -7,6 +7,9 @@ import org.hibernate.Session;
 import org.hibernate.criterion.Order;
 import org.hibernate.criterion.Projections;
 import org.hibernate.criterion.Restrictions;
+import org.hibernate.envers.AuditReader;
+import org.hibernate.envers.AuditReaderFactory;
+import org.hibernate.envers.query.AuditEntity;
 
 import java.util.List;
 
@@ -70,4 +73,21 @@ public class TableTicket extends SQLCore {
 		}
 	}
 
+	@SuppressWarnings("unchecked")
+	public static List<Object[]> track(int tid) {
+		try (Session s = SQLCore.sf.openSession()) {
+			AuditReader reader = getAuditReader(s);
+			return reader.createQuery()
+					.forRevisionsOfEntity(Ticket.class, false, true)
+					.addOrder(AuditEntity.revisionNumber().desc())
+					.add(AuditEntity.id().eq(tid))
+					.getResultList()
+					;
+		}
+	}
+
+	protected static AuditReader getAuditReader(Session session) {
+		return AuditReaderFactory.get(session);
+	}
+
 }