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.batch.index; 021 022import com.google.common.collect.Sets; 023import org.sonar.api.database.DatabaseSession; 024import org.sonar.api.database.model.Snapshot; 025import org.sonar.api.database.model.SnapshotSource; 026import org.sonar.api.resources.DuplicatedSourceException; 027import org.sonar.api.resources.Resource; 028 029import java.util.Set; 030 031public final class SourcePersister { 032 033 private DatabaseSession session; 034 private Set<Integer> savedSnapshotIds = Sets.newHashSet(); 035 private ResourcePersister resourcePersister; 036 037 public SourcePersister(DatabaseSession session, ResourcePersister resourcePersister) { 038 this.session = session; 039 this.resourcePersister = resourcePersister; 040 } 041 042 public void saveSource(Resource resource, String source) { 043 Snapshot snapshot = resourcePersister.getSnapshotOrFail(resource); 044 if (isCached(snapshot)) { 045 throw new DuplicatedSourceException(resource); 046 } 047 session.save(new SnapshotSource(snapshot.getId(), source)); 048 session.commit(); 049 addToCache(snapshot); 050 } 051 052 public String getSource(Resource resource) { 053 SnapshotSource source = null; 054 Snapshot snapshot = resourcePersister.getSnapshot(resource); 055 if (snapshot!=null && snapshot.getId()!=null) { 056 source = session.getSingleResult(SnapshotSource.class, "snapshotId", snapshot.getId()); 057 } 058 return source!=null ? source.getData() : null; 059 } 060 061 private boolean isCached(Snapshot snapshot) { 062 return savedSnapshotIds.contains(snapshot.getId()); 063 } 064 065 private void addToCache(Snapshot snapshot) { 066 savedSnapshotIds.add(snapshot.getId()); 067 } 068 069 public void clear() { 070 savedSnapshotIds.clear(); 071 } 072}