From d1855ee400066a6db051fa3c761569499d02ac08 Mon Sep 17 00:00:00 2001
From: Sergey Matvienko <smatvienko@thingsboard.io>
Date: Tue, 6 Jul 2021 20:51:11 +0300
Subject: [PATCH] javax.persistence.query.timeout with slowQuery repository for
 manual testing

---
 application/src/main/resources/thingsboard.yml            | 2 ++
 .../thingsboard/server/dao/sql/event/EventRepository.java | 6 ++++++
 .../thingsboard/server/dao/sql/event/JpaBaseEventDao.java | 8 ++++++++
 3 files changed, 16 insertions(+)

diff --git a/application/src/main/resources/thingsboard.yml b/application/src/main/resources/thingsboard.yml
index 05f3f1e825..5433aa2e0b 100644
--- a/application/src/main/resources/thingsboard.yml
+++ b/application/src/main/resources/thingsboard.yml
@@ -472,6 +472,8 @@ spring:
       repositories:
         enabled: "true"
   jpa:
+    properties:
+      javax.persistence.query.timeout: "${JAVAX_PERSISTENCE_QUERY_TIMEOUT:29000}"
     open-in-view: "false"
     hibernate:
       ddl-auto: "none"
diff --git a/dao/src/main/java/org/thingsboard/server/dao/sql/event/EventRepository.java b/dao/src/main/java/org/thingsboard/server/dao/sql/event/EventRepository.java
index 270ab26329..5f6cf91ab3 100644
--- a/dao/src/main/java/org/thingsboard/server/dao/sql/event/EventRepository.java
+++ b/dao/src/main/java/org/thingsboard/server/dao/sql/event/EventRepository.java
@@ -241,4 +241,10 @@ public interface EventRepository extends PagingAndSortingRepository<EventEntity,
                                            @Param("errorsOccurred") Integer errorsOccurred,
                                            Pageable pageable);
 
+    /**
+     *  Sleep 3 second on JPA connection
+     * */
+    @Query(value = "select count(*) from pg_sleep(3)", nativeQuery = true)
+    Long slowQuery();
+
 }
diff --git a/dao/src/main/java/org/thingsboard/server/dao/sql/event/JpaBaseEventDao.java b/dao/src/main/java/org/thingsboard/server/dao/sql/event/JpaBaseEventDao.java
index ee09bb7a90..383b4eac82 100644
--- a/dao/src/main/java/org/thingsboard/server/dao/sql/event/JpaBaseEventDao.java
+++ b/dao/src/main/java/org/thingsboard/server/dao/sql/event/JpaBaseEventDao.java
@@ -144,6 +144,14 @@ public class JpaBaseEventDao extends JpaAbstractDao<EventEntity, Event> implemen
 
     @Override
     public PageData<Event> findEvents(UUID tenantId, EntityId entityId, String eventType, TimePageLink pageLink) {
+        log.warn("going to run slow query");
+        try {
+            eventRepository.slowQuery();
+        } catch (Exception e) {
+            log.error("slowQuery" , e);
+            throw e;
+        }
+        log.warn("finished slow query");
         return DaoUtil.toPageData(
                 eventRepository
                         .findEventsByTenantIdAndEntityIdAndEventType(