From 829196f92bc6c9bee1ebd0da080e54f9dd19d7e0 Mon Sep 17 00:00:00 2001
From: sonika-shah <58761340+sonika-shah@users.noreply.github.com>
Date: Wed, 14 May 2025 09:50:23 +0530
Subject: [PATCH] Support followers for Database (#21169)

---
 .../resources/databases/DatabaseResource.java | 63 ++++++++++++++++++-
 .../databases/DatabaseResourceTest.java       |  2 +-
 .../json/schema/entity/data/database.json     |  4 ++
 .../ui/src/generated/entity/data/database.ts  |  4 ++
 4 files changed, 71 insertions(+), 2 deletions(-)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/resources/databases/DatabaseResource.java b/openmetadata-service/src/main/java/org/openmetadata/service/resources/databases/DatabaseResource.java
index f91ad764f91..0d1dffc9354 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/resources/databases/DatabaseResource.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/resources/databases/DatabaseResource.java
@@ -79,7 +79,7 @@ public class DatabaseResource extends EntityResource<Database, DatabaseRepositor
   public static final String COLLECTION_PATH = "v1/databases/";
   private final DatabaseMapper mapper = new DatabaseMapper();
   static final String FIELDS =
-      "owners,databaseSchemas,usageSummary,location,tags,extension,domain,sourceHash";
+      "owners,databaseSchemas,usageSummary,location,tags,extension,domain,sourceHash,followers";
 
   @Override
   public Database addHref(UriInfo uriInfo, Database db) {
@@ -757,4 +757,65 @@ public class DatabaseResource extends EntityResource<Database, DatabaseRepositor
     Database database = repository.deleteDatabaseProfilerConfig(id);
     return addHref(uriInfo, database);
   }
+
+  @PUT
+  @Path("/{id}/followers")
+  @Operation(
+      operationId = "addFollowerToDatabase",
+      summary = "Add a follower",
+      description = "Add a user identified by `userId` as followed of this Database",
+      responses = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "OK",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = ChangeEvent.class))),
+        @ApiResponse(responseCode = "404", description = "Database for instance {id} is not found")
+      })
+  public Response addFollower(
+      @Context UriInfo uriInfo,
+      @Context SecurityContext securityContext,
+      @Parameter(description = "Id of the Database", schema = @Schema(type = "UUID"))
+          @PathParam("id")
+          UUID id,
+      @Parameter(
+              description = "Id of the user to be added as follower",
+              schema = @Schema(type = "string"))
+          UUID userId) {
+    return repository
+        .addFollower(securityContext.getUserPrincipal().getName(), id, userId)
+        .toResponse();
+  }
+
+  @DELETE
+  @Path("/{id}/followers/{userId}")
+  @Operation(
+      operationId = "deleteFollower",
+      summary = "Remove a follower",
+      description = "Remove the user identified `userId` as a follower of the entity.",
+      responses = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "OK",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = ChangeEvent.class)))
+      })
+  public Response deleteFollower(
+      @Context UriInfo uriInfo,
+      @Context SecurityContext securityContext,
+      @Parameter(description = "Id of the Entity", schema = @Schema(type = "UUID")) @PathParam("id")
+          UUID id,
+      @Parameter(
+              description = "Id of the user being removed as follower",
+              schema = @Schema(type = "string"))
+          @PathParam("userId")
+          String userId) {
+    return repository
+        .deleteFollower(securityContext.getUserPrincipal().getName(), id, UUID.fromString(userId))
+        .toResponse();
+  }
 }
diff --git a/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseResourceTest.java b/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseResourceTest.java
index 6a12032a3ce..433e6e94301 100644
--- a/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseResourceTest.java
+++ b/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/DatabaseResourceTest.java
@@ -333,7 +333,7 @@ public class DatabaseResourceTest extends EntityResourceTest<Database, CreateDat
         database.getUsageSummary(),
         database.getLocation());
 
-    fields = "owners,databaseSchemas,usageSummary,location,tags";
+    fields = "owners,databaseSchemas,usageSummary,location,tags,followers";
     database =
         byName
             ? getEntityByName(database.getFullyQualifiedName(), fields, ADMIN_AUTH_HEADERS)
diff --git a/openmetadata-spec/src/main/resources/json/schema/entity/data/database.json b/openmetadata-spec/src/main/resources/json/schema/entity/data/database.json
index 73485414e12..f863ffdc495 100644
--- a/openmetadata-spec/src/main/resources/json/schema/entity/data/database.json
+++ b/openmetadata-spec/src/main/resources/json/schema/entity/data/database.json
@@ -128,6 +128,10 @@
     "certification": {
       "$ref": "../../type/assetCertification.json"
     },
+    "followers": {
+      "description": "Followers of this entity.",
+      "$ref": "../../type/entityReferenceList.json"
+    },
     "sourceHash": {
       "description": "Source hash of the entity",
       "type": "string",
diff --git a/openmetadata-ui/src/main/resources/ui/src/generated/entity/data/database.ts b/openmetadata-ui/src/main/resources/ui/src/generated/entity/data/database.ts
index 6311de19791..1c6714415b0 100644
--- a/openmetadata-ui/src/main/resources/ui/src/generated/entity/data/database.ts
+++ b/openmetadata-ui/src/main/resources/ui/src/generated/entity/data/database.ts
@@ -59,6 +59,10 @@ export interface Database {
      * Entity extension data with custom attributes added to the entity.
      */
     extension?: any;
+    /**
+     * Followers of this entity.
+     */
+    followers?: EntityReference[];
     /**
      * Name that uniquely identifies a database in the format 'ServiceName.DatabaseName'.
      */