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 */ 020package org.sonar.core.resource; 021 022import com.google.common.collect.Lists; 023import org.apache.ibatis.session.SqlSession; 024import org.sonar.core.persistence.MyBatis; 025 026import java.util.List; 027 028public class ResourceDao { 029 private MyBatis mybatis; 030 031 public ResourceDao(MyBatis mybatis) { 032 this.mybatis = mybatis; 033 } 034 035 public List<ResourceDto> getResources(ResourceQuery query) { 036 SqlSession session = mybatis.openSession(); 037 try { 038 return session.getMapper(ResourceMapper.class).selectResources(query); 039 } finally { 040 MyBatis.closeQuietly(session); 041 } 042 } 043 044 public List<Long> getResourceIds(ResourceQuery query) { 045 SqlSession session = mybatis.openSession(); 046 try { 047 return session.getMapper(ResourceMapper.class).selectResourceIds(query); 048 } finally { 049 MyBatis.closeQuietly(session); 050 } 051 } 052 053 public ResourceDto getResource(long projectId) { 054 SqlSession session = mybatis.openSession(); 055 try { 056 return getResource(projectId, session); 057 } finally { 058 MyBatis.closeQuietly(session); 059 } 060 } 061 062 public ResourceDto getResource(long projectId, SqlSession session) { 063 return session.getMapper(ResourceMapper.class).selectResource(projectId); 064 } 065 066 public List<ResourceDto> getDescendantProjects(long projectId) { 067 SqlSession session = mybatis.openSession(); 068 try { 069 return getDescendantProjects(projectId, session); 070 } finally { 071 MyBatis.closeQuietly(session); 072 } 073 } 074 075 public List<ResourceDto> getDescendantProjects(long projectId, SqlSession session) { 076 ResourceMapper mapper = session.getMapper(ResourceMapper.class); 077 List<ResourceDto> resources = Lists.newArrayList(); 078 appendChildProjects(projectId, mapper, resources); 079 return resources; 080 } 081 082 private void appendChildProjects(long projectId, ResourceMapper mapper, List<ResourceDto> resources) { 083 List<ResourceDto> subProjects = mapper.selectDescendantProjects(projectId); 084 for (ResourceDto subProject : subProjects) { 085 resources.add(subProject); 086 appendChildProjects(subProject.getId(), mapper, resources); 087 } 088 } 089}