From: Jacek <jacek.poreda@sonarsource.com>
Date: Fri, 20 Aug 2021 14:58:53 +0000 (+0200)
Subject: SONAR-15259 Use portfolios realted tables in DefineAction WS
X-Git-Tag: 9.1.0.47736~56
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=a79ac0071eb12dc0ac595d7faa374c620891ff27;p=sonarqube.git

SONAR-15259 Use portfolios realted tables in DefineAction WS
---

diff --git a/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioDao.java b/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioDao.java
index d9c49ae2178..d779c6845fa 100644
--- a/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioDao.java
+++ b/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioDao.java
@@ -64,9 +64,6 @@ public class PortfolioDao implements Dao {
   }
 
   public void insert(DbSession dbSession, PortfolioDto portfolio) {
-    if (portfolio.getUuid() == null) {
-      portfolio.setUuid(uuidFactory.create());
-    }
     mapper(dbSession).insert(portfolio);
   }
 
@@ -160,4 +157,22 @@ public class PortfolioDao implements Dao {
     return executeLargeInputs(uuids, uuids1 -> mapper(dbSession).selectByUuids(uuids1)).stream()
       .collect(Collectors.toMap(PortfolioDto::getUuid, PortfolioDto::getKey));
   }
+
+  public void deleteAllDescendantPortfolios(DbSession dbSession, String rootUuid) {
+    mapper(dbSession).deleteAllDescendantPortfolios(rootUuid);
+  }
+
+  public void deleteAllReferences(DbSession dbSession) {
+    mapper(dbSession).deleteAllReferences();
+
+  }
+
+  public void deleteAllProjects(DbSession dbSession) {
+    mapper(dbSession).deleteAllProjects();
+  }
+
+  public List<PortfolioProjectDto> selectAllProjectsOfPortfolios(DbSession dbSession) {
+    return mapper(dbSession).selectAllProjectsOfPortfolios();
+  }
+
 }
diff --git a/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioMapper.java b/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioMapper.java
index f019dbaed05..6587896849b 100644
--- a/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioMapper.java
+++ b/server/sonar-db-dao/src/main/java/org/sonar/db/portfolio/PortfolioMapper.java
@@ -76,4 +76,15 @@ public interface PortfolioMapper {
   void deleteProjects(String portfolioUuid);
 
   void deleteProject(@Param("portfolioUuid") String portfolioUuid, @Param("projectUuid") String projectUuid);
+
+  void deleteAllDescendantPortfolios(String rootUuid);
+
+  void deleteAllReferences();
+
+  void deleteAllProjects();
+
+  List<PortfolioProjectDto> selectAllProjectsOfPortfolios();
+
+  List<ReferenceDto> selectAllReferencesOfPortfolios();
+
 }
diff --git a/server/sonar-db-dao/src/main/resources/org/sonar/db/portfolio/PortfolioMapper.xml b/server/sonar-db-dao/src/main/resources/org/sonar/db/portfolio/PortfolioMapper.xml
index 3e581205004..938e314e129 100644
--- a/server/sonar-db-dao/src/main/resources/org/sonar/db/portfolio/PortfolioMapper.xml
+++ b/server/sonar-db-dao/src/main/resources/org/sonar/db/portfolio/PortfolioMapper.xml
@@ -55,18 +55,18 @@
       p.kee=#{kee,jdbcType=VARCHAR}
   </select>
 
-  <select id="selectAllRoots" resultType="Portfolio">
+  <select id="selectAll" resultType="Portfolio">
     SELECT
-      <include refid="portfolioColumns"/>
+    <include refid="portfolioColumns"/>
     FROM portfolios p
-    where
-      p.parent_uuid is null
   </select>
 
-  <select id="selectAll" resultType="Portfolio">
+  <select id="selectAllRoots" resultType="Portfolio">
     SELECT
       <include refid="portfolioColumns"/>
     FROM portfolios p
+    where
+      p.parent_uuid is null
   </select>
 
    <select id="selectProjects" resultType="Project">
@@ -78,6 +78,17 @@
       pp.portfolio_uuid=#{portfolioUuid,jdbcType=VARCHAR}
   </select>
 
+  <select id="selectAllProjectsOfPortfolios" resultType="org.sonar.db.portfolio.PortfolioProjectDto">
+    SELECT
+       pp.uuid,
+       pp.project_uuid as projectUuid,
+       pp.portfolio_uuid as portfolioUuid,
+       pp.project_uuid as projectUuid,
+       pp.created_at as createdAt
+    FROM portfolio_projects pp
+    INNER JOIN projects p on p.uuid = pp.project_uuid
+  </select>
+
   <select id="selectAllProjectsInHierarchy" resultType="org.sonar.db.portfolio.PortfolioProjectDto">
     SELECT
       pp.uuid,
@@ -208,6 +219,18 @@
     <foreach collection="uuids" open="(" close=")" item="uuid" separator=",">#{uuid,jdbcType=VARCHAR}</foreach>
   </delete>
 
+  <delete id="deleteAllDescendantPortfolios" parameterType="String">
+    DELETE FROM portfolios WHERE root_uuid = #{rootUuid,jdbcType=VARCHAR} and parent_uuid is not null
+  </delete>
+
+  <delete id="deleteAllReferences" parameterType="String">
+    DELETE FROM portfolio_references
+  </delete>
+
+  <delete id="deleteAllProjects" parameterType="String">
+    DELETE FROM portfolio_projects
+  </delete>
+
   <insert id="insertReference" parameterType="PortfolioReference">
     INSERT INTO portfolio_references (
       uuid,
diff --git a/server/sonar-db-dao/src/test/java/org/sonar/db/portfolio/PortfolioDaoTest.java b/server/sonar-db-dao/src/test/java/org/sonar/db/portfolio/PortfolioDaoTest.java
index d921b1477de..afbd38a25b7 100644
--- a/server/sonar-db-dao/src/test/java/org/sonar/db/portfolio/PortfolioDaoTest.java
+++ b/server/sonar-db-dao/src/test/java/org/sonar/db/portfolio/PortfolioDaoTest.java
@@ -237,6 +237,61 @@ public class PortfolioDaoTest {
     assertThat(portfolioDao.getAllProjectsInHierarchy(db.getSession(), "nonexisting")).isEmpty();
   }
 
+  @Test
+  public void deleteAllDescendantPortfolios() {
+    createPortfolio("root", null, "root");
+    createPortfolio("child1", null, "root");
+    createPortfolio("child11", "child1", "root");
+
+    createPortfolio("root2", null, "root2");
+
+    portfolioDao.deleteAllDescendantPortfolios(db.getSession(), "root");
+    portfolioDao.deleteAllDescendantPortfolios(db.getSession(), "root2");
+
+    assertThat(db.countSql(db.getSession(), "select count(*) from portfolios where parent_uuid is not null")).isZero();
+  }
+
+  @Test
+  public void deleteAllReferences() {
+    createPortfolio("root", null, "root");
+    createPortfolio("child1", null, "root");
+    createPortfolio("child11", "child1", "root");
+
+    createPortfolio("root2", null, "root2");
+    createPortfolio("root3", null, "root3");
+    createPortfolio("root4", null, "root4");
+
+    portfolioDao.addReference(db.getSession(), "child1", "root2");
+    portfolioDao.addReference(db.getSession(), "root3", "root4");
+
+    portfolioDao.deleteAllReferences(db.getSession());
+
+    assertThat(db.countSql(db.getSession(), "select count(*) from portfolio_references")).isZero();
+  }
+
+  @Test
+  public void deleteAllProjects() {
+    db.components().insertPrivateProject("p1");
+    db.components().insertPrivateProject("p2");
+    db.components().insertPrivateProject("p3");
+    db.components().insertPrivateProject("p4");
+
+    createPortfolio("root", null, "root");
+    createPortfolio("child1", null, "root");
+    createPortfolio("child11", "child1", "root");
+
+    createPortfolio("root2", null, "root2");
+
+    portfolioDao.addProject(db.getSession(), "root", "p1");
+    portfolioDao.addProject(db.getSession(), "child1", "p2");
+    portfolioDao.addProject(db.getSession(), "child11", "p3");
+    portfolioDao.addProject(db.getSession(), "root2", "p4");
+
+    portfolioDao.deleteAllProjects(db.getSession());
+
+    assertThat(db.countSql(db.getSession(), "select count(*) from portfolio_projects")).isZero();
+  }
+
   private PortfolioDto createPortfolio(String uuid, @Nullable String parentUuid, String rootUuid) {
     PortfolioDto p = new PortfolioDto()
       .setName("name_" + uuid)
diff --git a/server/sonar-db-dao/src/testFixtures/java/org/sonar/db/component/ComponentDbTester.java b/server/sonar-db-dao/src/testFixtures/java/org/sonar/db/component/ComponentDbTester.java
index 905d3297731..79e61335dd6 100644
--- a/server/sonar-db-dao/src/testFixtures/java/org/sonar/db/component/ComponentDbTester.java
+++ b/server/sonar-db-dao/src/testFixtures/java/org/sonar/db/component/ComponentDbTester.java
@@ -236,6 +236,13 @@ public class ComponentDbTester {
     return insertComponentAndPortfolio(ComponentTesting.newPortfolio().setPrivate(true), true, dtoPopulator, portfolioPopulator);
   }
 
+  public void addPortfolioReference(ComponentDto portfolio, String... referencerUuids) {
+    for (String uuid : referencerUuids) {
+      dbClient.portfolioDao().addReference(dbSession, portfolio.uuid(), uuid);
+    }
+    db.commit();
+  }
+
   public void addPortfolioProject(ComponentDto portfolio, String... projectUuids) {
     for (String uuid : projectUuids) {
       dbClient.portfolioDao().addProject(dbSession, portfolio.uuid(), uuid);