Add and manage announcements

src/main/java/org/gridsuite/useradmin/server/UserAdminController.java

@@ -7,20 +7,21 @@
 package org.gridsuite.useradmin.server;
 
 import io.swagger.v3.oas.annotations.Operation;
-import io.swagger.v3.oas.annotations.Parameter;
 import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.responses.ApiResponses;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.validation.constraints.NotEmpty;
 import org.gridsuite.useradmin.server.dto.UserConnection;
 import org.gridsuite.useradmin.server.dto.UserInfos;
+import org.gridsuite.useradmin.server.repository.AnnouncementEntity;
 import org.gridsuite.useradmin.server.service.UserAdminService;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.UUID;
 
 /**
  * @author Etienne Homer <etienne.homer at rte-france.com>
@@ -109,27 +110,38 @@ public ResponseEntity<List<UserConnection>> getConnections(@RequestHeader("userI
         return ResponseEntity.ok().contentType(MediaType.APPLICATION_JSON).body(service.getConnections(userId));
     }
 
-    @PostMapping(value = "/messages/maintenance")
-    @Operation(summary = "send a message to all users connected")
+    @PostMapping(value = "/announcements")
+    @Operation(summary = "Send a message to all the connected users")
     @ApiResponses(value = {
         @ApiResponse(responseCode = "200", description = "message sent"),
         @ApiResponse(responseCode = "403", description = "user is not an admin")
     })
     public ResponseEntity<Void> sendMaintenanceMessage(@RequestHeader("userId") String userId,
-                                                       @Parameter(description = "the display time of the message in seconds") @RequestParam(value = "durationInSeconds", required = false) Integer duration,
-                                                       @Parameter(description = "the message to display") @RequestBody String message) {
-        service.sendMaintenanceMessage(userId, duration, message);
+                                                       @RequestBody AnnouncementEntity announcement) {
+        service.sendAnnouncement(announcement, userId);
         return ResponseEntity.ok().build();
     }
 
-    @PostMapping(value = "/messages/cancel-maintenance")
-    @Operation(summary = "send a message to all users connected")
+    @DeleteMapping(value = "/announcements/{announcementId}")
+    @Operation(summary = "Cancel and delete a message")
     @ApiResponses(value = {
-        @ApiResponse(responseCode = "200", description = "message sent"),
-        @ApiResponse(responseCode = "403", description = "user is not an admin")
+        @ApiResponse(responseCode = "200", description = "message canceled"),
+        @ApiResponse(responseCode = "403", description = "user is not an admin"),
+        @ApiResponse(responseCode = "404", description = "message not found")
     })
-    public ResponseEntity<Void> sendCancelMaintenanceMessage(@RequestHeader("userId") String userId) {
-        service.sendCancelMaintenanceMessage(userId);
+    public ResponseEntity<Void> sendCancelMaintenanceMessage(@RequestHeader("userId") String userId,
+                                                             @PathVariable("announcementId") UUID announcementId) {
+        service.cancelAnnouncement(announcementId, userId);
         return ResponseEntity.ok().build();
     }
+
+    @GetMapping(value = "/announcements")
+    @Operation(summary = "Get all the messages")
+    @ApiResponses(value = {
+        @ApiResponse(responseCode = "200", description = "messages retrieved"),
+        @ApiResponse(responseCode = "403", description = "user is not an admin")
+    })
+    public ResponseEntity<List<AnnouncementEntity>> sendCancelMaintenanceMessage(@RequestHeader("userId") String userId) {
+        return ResponseEntity.ok(service.getAllAnnouncements(userId));
+    }
 }

src/main/java/org/gridsuite/useradmin/server/repository/AnnouncementEntity.java

@@ -0,0 +1,50 @@
+/**
+ * Copyright (c) 2024, RTE (http://www.rte-france.com)
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+package org.gridsuite.useradmin.server.repository;
+
+import jakarta.persistence.*;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+import java.time.Duration;
+import java.time.Instant;
+import java.util.UUID;
+
+/**
+ * @author Florent MILLOT <florent.millot at rte-france.com>
+ */
+@NoArgsConstructor
+@AllArgsConstructor
+@Getter
+@Setter
+@Entity
+@Table(name = "announcement")
+public class AnnouncementEntity {
+
+    public AnnouncementEntity(String message, Duration duration) {
+        this.message = message;
+        this.duration = duration;
+    }
+
+    @Id
+    @GeneratedValue
+    @Column
+    private UUID id;
+
+    @Column(nullable = false)
+    private Instant creationDate = Instant.now();
+
+    @Column(nullable = false)
+    private String message;
+
+    @Column
+    private Duration duration;
+
+}

src/main/java/org/gridsuite/useradmin/server/repository/AnnouncementRepository.java

@@ -0,0 +1,20 @@
+/**
+ * Copyright (c) 2024, RTE (http://www.rte-france.com)
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+package org.gridsuite.useradmin.server.repository;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+import java.util.UUID;
+
+/**
+ * @author Florent MILLOT <florent.millot at rte-france.com>
+ */
+@Repository
+public interface AnnouncementRepository extends JpaRepository<AnnouncementEntity, UUID> {
+}

src/main/java/org/gridsuite/useradmin/server/service/AnnouncementService.java

@@ -0,0 +1,61 @@
+/**
+ * Copyright (c) 2024, RTE (http://www.rte-france.com)
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+package org.gridsuite.useradmin.server.service;
+
+import org.gridsuite.useradmin.server.repository.AnnouncementEntity;
+import org.gridsuite.useradmin.server.repository.AnnouncementRepository;
+import org.springframework.http.HttpStatus;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.server.ResponseStatusException;
+
+import java.util.List;
+import java.util.Objects;
+import java.util.UUID;
+
+/**
+ * @author Florent MILLOT <florent.millot at rte-france.com>
+ */
+@Service
+public class AnnouncementService {
+
+    private final AnnouncementRepository announcementRepository;
+    private final NotificationService notificationService;
+
+    public AnnouncementService(final AnnouncementRepository announcementRepository,
+                               final NotificationService notificationService) {
+        this.announcementRepository = Objects.requireNonNull(announcementRepository);
+        this.notificationService = Objects.requireNonNull(notificationService);
+    }
+
+    @Transactional
+    public void sendAnnouncement(AnnouncementEntity announcement) {
+        this.announcementRepository.deleteAll(); // for now, only one message at a time
+        this.announcementRepository.save(announcement);
+        if (announcement.getDuration() == null) {
+            notificationService.emitMaintenanceMessage(announcement.getMessage());
+        } else {
+            notificationService.emitMaintenanceMessage(announcement.getMessage(), announcement.getDuration().toSeconds());
+        }
+    }
+
+    @Transactional
+    public void cancelAnnouncement(UUID announcementId) {
+        if (!announcementRepository.existsById(announcementId)) {
+            throw new ResponseStatusException(HttpStatus.NOT_FOUND, "Unable to find announcement");
+        }
+        this.announcementRepository.deleteById(announcementId);
+        notificationService.emitCancelMaintenanceMessage();
+    }
+
+    @Transactional(readOnly = true)
+    public List<AnnouncementEntity> getAllAnnouncements() {
+        return this.announcementRepository.findAll();
+    }
+
+}

src/main/java/org/gridsuite/useradmin/server/service/NotificationService.java

@@ -42,7 +42,7 @@ private void sendMessage(Message<String> message) {
         updatePublisher.send("publishMessage-out-0", message);
     }
 
-    public void emitMaintenanceMessage(String message, int duration) {
+    public void emitMaintenanceMessage(String message, long duration) {
         sendMessage(MessageBuilder.withPayload(message)
                 .setHeader(HEADER_MESSAGE_TYPE, MESSAGE_TYPE_MAINTENANCE)
                 .setHeader(HEADER_DURATION, duration)

src/main/java/org/gridsuite/useradmin/server/service/UserAdminService.java

@@ -10,16 +10,14 @@
 import org.gridsuite.useradmin.server.UserAdminException;
 import org.gridsuite.useradmin.server.dto.UserConnection;
 import org.gridsuite.useradmin.server.dto.UserInfos;
+import org.gridsuite.useradmin.server.repository.AnnouncementEntity;
 import org.gridsuite.useradmin.server.repository.UserAdminRepository;
 import org.gridsuite.useradmin.server.repository.UserInfosEntity;
 import org.springframework.lang.NonNull;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
-import java.util.Collection;
-import java.util.List;
-import java.util.Objects;
-import java.util.Optional;
+import java.util.*;
 
 import static org.gridsuite.useradmin.server.UserAdminException.Type.FORBIDDEN;
 
@@ -30,17 +28,17 @@
 public class UserAdminService {
     private final UserAdminRepository userAdminRepository;
     private final ConnectionsService connectionsService;
+    private final AnnouncementService announcementService;
     private final UserAdminApplicationProps applicationProps;
-    private final NotificationService notificationService;
 
     public UserAdminService(final UserAdminApplicationProps applicationProps,
                             final UserAdminRepository userAdminRepository,
                             final ConnectionsService connectionsService,
-                            final NotificationService notificationService) {
+                            final AnnouncementService announcementService) {
         this.applicationProps = Objects.requireNonNull(applicationProps);
         this.userAdminRepository = Objects.requireNonNull(userAdminRepository);
         this.connectionsService = Objects.requireNonNull(connectionsService);
-        this.notificationService = Objects.requireNonNull(notificationService);
+        this.announcementService = Objects.requireNonNull(announcementService);
     }
 
     private boolean isAdmin(@lombok.NonNull String sub) {
@@ -109,21 +107,18 @@ public boolean userIsAdmin(@NonNull String userId) {
         return isAdmin(userId);
     }
 
-    public void sendMaintenanceMessage(String userId, Integer durationInSeconds, String message) {
-        if (!isAdmin(userId)) {
-            throw new UserAdminException(FORBIDDEN);
-        }
-        if (durationInSeconds == null) {
-            notificationService.emitMaintenanceMessage(message);
-        } else {
-            notificationService.emitMaintenanceMessage(message, durationInSeconds);
-        }
+    public void sendAnnouncement(AnnouncementEntity announcement, String userId) {
+        assertIsAdmin(userId);
+        announcementService.sendAnnouncement(announcement);
     }
 
-    public void sendCancelMaintenanceMessage(String userId) {
-        if (!isAdmin(userId)) {
-            throw new UserAdminException(FORBIDDEN);
-        }
-        notificationService.emitCancelMaintenanceMessage();
+    public void cancelAnnouncement(UUID announcementId, String userId) {
+        assertIsAdmin(userId);
+        announcementService.cancelAnnouncement(announcementId);
+    }
+
+    public List<AnnouncementEntity> getAllAnnouncements(String userId) {
+        assertIsAdmin(userId);
+        return announcementService.getAllAnnouncements();
     }
 }

src/main/resources/db/changelog/changesets/changelog_20240329T134444Z.xml

@@ -0,0 +1,17 @@
+<?xml version="1.1" encoding="UTF-8" standalone="no"?>
+<databaseChangeLog xmlns="http://www.liquibase.org/xml/ns/dbchangelog" xmlns:ext="http://www.liquibase.org/xml/ns/dbchangelog-ext" xmlns:pro="http://www.liquibase.org/xml/ns/pro" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.liquibase.org/xml/ns/dbchangelog-ext http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-ext.xsd http://www.liquibase.org/xml/ns/pro http://www.liquibase.org/xml/ns/pro/liquibase-pro-latest.xsd http://www.liquibase.org/xml/ns/dbchangelog http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-latest.xsd">
+    <changeSet author="florent (generated)" id="1711719892005-1">
+        <createTable tableName="announcement">
+            <column name="id" type="UUID">
+                <constraints nullable="false" primaryKey="true" primaryKeyName="announcementPK"/>
+            </column>
+            <column name="creation_date" type="TIMESTAMP(6)">
+                <constraints nullable="false"/>
+            </column>
+            <column name="duration" type="numeric(21, 0)"/>
+            <column name="message" type="VARCHAR(255)">
+                <constraints nullable="false"/>
+            </column>
+        </createTable>
+    </changeSet>
+</databaseChangeLog>

src/main/resources/db/changelog/db.changelog-master.yaml

@@ -8,3 +8,6 @@ databaseChangeLog:
   - include:
       file: changesets/changelog_20240202T000000Z.xml
       relativeToChangelogFile: true
+  - include:
+      file: changesets/changelog_20240329T134444Z.xml
+      relativeToChangelogFile: true