001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2011 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.batch.index; 021 022 import org.sonar.api.database.DatabaseSession; 023 import org.sonar.api.database.model.ResourceModel; 024 import org.sonar.api.database.model.Snapshot; 025 import org.sonar.api.resources.Project; 026 import org.sonar.api.resources.ProjectLink; 027 028 public final class LinkPersister { 029 private DatabaseSession session; 030 private ResourcePersister resourcePersister; 031 032 public LinkPersister(DatabaseSession session, ResourcePersister resourcePersister) { 033 this.session = session; 034 this.resourcePersister = resourcePersister; 035 } 036 037 public void saveLink(Project project, ProjectLink link) { 038 Snapshot snapshot = resourcePersister.getSnapshotOrFail(project); 039 ResourceModel projectDao = session.reattach(ResourceModel.class, snapshot.getResourceId()); 040 ProjectLink dbLink = projectDao.getProjectLink(link.getKey()); 041 if (dbLink == null) { 042 link.setResource(projectDao); 043 projectDao.getProjectLinks().add(link); 044 session.save(link); 045 046 } else { 047 dbLink.copyFieldsFrom(link); 048 session.save(dbLink); 049 } 050 session.commit(); 051 052 } 053 054 public void deleteLink(Project project, String linkKey) { 055 Snapshot snapshot = resourcePersister.getSnapshot(project); 056 if (snapshot != null) { 057 ResourceModel model = session.reattach(ResourceModel.class, snapshot.getResourceId()); 058 ProjectLink dbLink = model.getProjectLink(linkKey); 059 if (dbLink != null) { 060 session.remove(dbLink); 061 model.getProjectLinks().remove(dbLink); 062 session.commit(); 063 } 064 065 } 066 } 067 }