Fix LazyInitializationException in agent WebSocket proxy (#169)

* Initial plan

* Fix LazyInitializationException in agent proxy communication

Co-authored-by: phrocker <[email protected]>

---------

Co-authored-by: copilot-swe-agent[bot] <[email protected]>
Co-authored-by: phrocker <[email protected]>

Author: Copilot

dataplane/src/main/java/io/sentrius/sso/core/repository/OpsApprovalRepository.java

@@ -4,11 +4,14 @@
 import java.util.UUID;
 import io.sentrius.sso.core.model.zt.OpsApproval;
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
 
 public interface OpsApprovalRepository extends JpaRepository<OpsApproval, Long> {
     void deleteByZtatRequestId(Long ztatRequestId);
 
     Optional<OpsApproval> findByZtatRequestId(Long id);
 
-    Optional<OpsApproval> findByToken(UUID token);
+    @Query("SELECT oa FROM OpsApproval oa LEFT JOIN FETCH oa.ztatRequest WHERE oa.token = :token")
+    Optional<OpsApproval> findByToken(@Param("token") UUID token);
 }

dataplane/src/main/java/io/sentrius/sso/core/services/security/ZeroTrustAccessTokenService.java

@@ -19,11 +19,11 @@
 import io.sentrius.sso.core.model.zt.OpsZeroTrustAcessTokenRequest;
 import io.sentrius.sso.core.repository.OpsJITRequestRepository;
 import io.sentrius.sso.core.utils.ZTATUtils;
-import jakarta.transaction.Transactional;
 import lombok.NonNull;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 @Slf4j
 @Service
@@ -323,6 +323,7 @@ public void addCommunicationLink(RequestCommunicationLink link) {
     ztatRequestService.addCommunicationLink(link);
   }
 
+  @Transactional(readOnly = true)
   public boolean isOpsActive(String ztat) {
     var status = ztatRequestService.getOpsTokenStatus(ztat);
     if (status.isPresent()) {
@@ -346,6 +347,7 @@ public boolean isOpsActive(String ztat) {
 
   }
 
+  @Transactional
   public boolean incremenOpsUses(String ztat) {
     var status = ztatRequestService.getOpsTokenStatus(ztat);
     if (status.isPresent()) {

dataplane/src/main/java/io/sentrius/sso/core/services/security/ZeroTrustRequestService.java

@@ -192,6 +192,7 @@ public void revokeJIT(ZeroTrustAccessTokenRequest ztatRequest, Long userId) {
         }
     }
 
+    @Transactional(readOnly = true)
     public Optional<OpsApproval> getOpsTokenStatus(String token ) {
         try {
             return opsApprovalRepository.findByToken(UUID.fromString(token));