fix: AI chat not working when use PostgreSQL as database (OD-2657)

Author: robinshine

server-core/src/main/java/io/onedev/server/data/migration/DataMigrator.java

@@ -8442,4 +8442,12 @@ private void migrate217(File dataDir, Stack<Integer> versions) {
 		}
 	}
 
+	private void migrate218(File dataDir, Stack<Integer> versions) {
+		for (File file : dataDir.listFiles()) {
+			if (file.getName().startsWith("Chats.xml") || file.getName().startsWith("ChatMessages.xml")) {
+				FileUtils.deleteFile(file);
+			}
+		}
+	}
+
 }

server-core/src/main/java/io/onedev/server/model/ChatMessage.java

@@ -1,13 +1,10 @@
 package io.onedev.server.model;
 
-import java.util.LinkedHashMap;
-
 import javax.persistence.Column;
 import javax.persistence.Entity;
 import javax.persistence.FetchType;
 import javax.persistence.Index;
 import javax.persistence.JoinColumn;
-import javax.persistence.Lob;
 import javax.persistence.ManyToOne;
 import javax.persistence.Table;
 
@@ -29,14 +26,9 @@ public class ChatMessage extends AbstractEntity {
 
 	private boolean request;
 
-	@Lob
 	@Column(nullable=false, length=MAX_CONTENT_LEN)
     private String content;
 
-	@Lob
-    @Column(nullable=false, length=65535)
-    private LinkedHashMap<String, String> attachments = new LinkedHashMap<>();
-
 	public Chat getChat() {
 		return chat;
 	}
@@ -69,12 +61,4 @@ public void setContent(String content) {
 		this.content = StringUtils.abbreviate(content, MAX_CONTENT_LEN);
 	}
 
-	public LinkedHashMap<String, String> getAttachments() {
-		return attachments;
-	}
-
-	public void setAttachments(LinkedHashMap<String, String> attachments) {
-		this.attachments = attachments;
-	}
-	
 }

server-core/src/main/java/io/onedev/server/model/User.java

@@ -734,11 +734,7 @@ public boolean isSystem() {
 	public boolean isUnknown() {
 		return UNKNOWN_ID.equals(getId());
 	}
-	
-	public boolean isOrdinary() {
-		return getId() > 0;
-	}
-	
+		
 	public Collection<UserAuthorization> getProjectAuthorizations() {
 		return projectAuthorizations;
 	}

server-core/src/main/java/io/onedev/server/notification/CodeCommentNotificationManager.java

@@ -85,7 +85,7 @@ public void on(CodeCommentEvent event) {
 			}
 
 			Set<String> emailAddresses = notifyUsers.stream()
-					.filter(it -> it.isOrdinary()
+					.filter(it -> it.getId() > 0
 							&& it.getType() != User.Type.AI
 							&& (!it.equals(event.getUser()) || it.isNotifyOwnEvents())
 							&& it.getPrimaryEmailAddress() != null
@@ -118,7 +118,7 @@ public void on(CodeCommentEvent event) {
 	}
 
 	private boolean isAiEntitled(@Nullable User user, CodeComment comment, User ai) {
-		if (user != null && user.isOrdinary()) {
+		if (user != null && user.getId() > 0) {
 			if (user.isEntitledToAi(ai)) {
 				return true;
 			} else {

server-core/src/main/java/io/onedev/server/notification/IssueNotificationManager.java

@@ -436,7 +436,7 @@ public void on(IssuesMoved event) {
 	}
 
 	private boolean isAiEntitled(@Nullable User user, Issue issue, User ai) {
-		if (user != null && user.isOrdinary()) {
+		if (user != null && user.getId() > 0) {
 			if (user.isEntitledToAi(ai)) {
 				return true;
 			} else {

server-core/src/main/java/io/onedev/server/notification/PullRequestNotificationManager.java

@@ -415,7 +415,7 @@ Review current pull request for major issues (ignore styling/format/documentatio
 	}
 
 	private boolean isAiEntitled(@Nullable User user, PullRequest request, User ai) {
-		if (user != null && user.isOrdinary()) {
+		if (user != null && user.getId() > 0) {
 			if (user.isEntitledToAi(ai)) {
 				return true;
 			} else {

server-core/src/main/java/io/onedev/server/web/page/base/base.js

@@ -352,27 +352,42 @@ onedev.server = {
 			pageUnloading = true;
 		});
 
-		var keepAlive;
+		var keepAliveTimer;
+		var lastKeepAliveMessageReceived = true;
 		Wicket.Event.subscribe("/websocket/open", function(jqEvent) {
 			$(".connection-error").hide();
-			keepAlive = setInterval(function() {
-				Wicket.WebSocket.send("KeepAlive");
-			}, 30000); // 30 seconds to tolerate Nginx default timeout (60 seconds)		
+			keepAliveTimer = setInterval(function() {
+				if (!lastKeepAliveMessageReceived) {
+					console.debug(new Date().toLocaleTimeString() + ": Last websocket keep alive message not received");
+					onConnectionError();
+				} else {
+					console.debug(new Date().toLocaleTimeString() + ": Sending websocket keep alive message");
+					Wicket.WebSocket.send("KeepAlive");
+					lastKeepAliveMessageReceived = false;
+				}
+			}, 30000); // 30 seconds to tolerate Nginx default timeout (60 seconds)
 		});
-		Wicket.Event.subscribe("/websocket/closed", function(jqEvent) {
+		function onConnectionError() {
 			if (!pageUnloading) {
 				$("body>.error").hide();
 				$(".connection-error").show();
 			}
-			if (keepAlive) {
-				clearInterval(keepAlive);
-				keepAlive = undefined;
+			if (keepAliveTimer) {
+				clearInterval(keepAliveTimer);
+				keepAliveTimer = undefined;
 			}
+		}
+		Wicket.Event.subscribe("/websocket/closed", function(jqEvent) {
+			console.debug(new Date().toLocaleTimeString() + ": Websocket closed");
+			onConnectionError();
 		});
 		Wicket.Event.subscribe("/websocket/message", function(jqEvent, message) {
 			if (message == "ErrorMessage") {
 				$("body>.error").hide();
 				$(".page-error").show();
+			} else if (message == "KeepAlive") {
+				lastKeepAliveMessageReceived = true;
+				console.debug(new Date().toLocaleTimeString() + ": Received websocket keep alive message");
 			}
 		});		
 	},