001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2012 SonarSource 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020 package org.sonar.core.resource; 021 022 import com.google.common.collect.Lists; 023 import org.apache.ibatis.session.SqlSession; 024 import org.sonar.core.persistence.MyBatis; 025 026 import java.util.List; 027 028 public class ResourceDao { 029 private MyBatis mybatis; 030 031 public ResourceDao(MyBatis mybatis) { 032 this.mybatis = mybatis; 033 } 034 035 public ResourceDto getResource(long projectId) { 036 SqlSession session = mybatis.openSession(); 037 try { 038 return getResource(projectId, session); 039 } finally { 040 MyBatis.closeQuietly(session); 041 } 042 } 043 044 public ResourceDto getResource(long projectId, SqlSession session) { 045 return session.getMapper(ResourceMapper.class).selectResource(projectId); 046 } 047 048 public List<ResourceDto> getDescendantProjects(long projectId) { 049 SqlSession session = mybatis.openSession(); 050 try { 051 return getDescendantProjects(projectId, session); 052 } finally { 053 MyBatis.closeQuietly(session); 054 } 055 } 056 057 public List<ResourceDto> getDescendantProjects(long projectId, SqlSession session) { 058 ResourceMapper mapper = session.getMapper(ResourceMapper.class); 059 List<ResourceDto> resources = Lists.newArrayList(); 060 appendChildProjects(projectId, mapper, resources); 061 return resources; 062 } 063 064 private void appendChildProjects(long projectId, ResourceMapper mapper, List<ResourceDto> resources) { 065 List<ResourceDto> subProjects = mapper.selectDescendantProjects(projectId); 066 for (ResourceDto subProject : subProjects) { 067 resources.add(subProject); 068 appendChildProjects(subProject.getId(), mapper, resources); 069 } 070 } 071 }