001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2009 SonarSource SA 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.indexer; 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 027 public class ProjectPersister extends ResourcePersister<Project> { 028 029 public ProjectPersister(DatabaseSession session) { 030 super(session); 031 } 032 033 @Override 034 protected String generateEffectiveKey(Bucket<Project> bucket) { 035 return bucket.getResource().getKey(); 036 } 037 038 @Override 039 protected void prepareResourceModel(ResourceModel resourceModel, Bucket<Project> bucket) { 040 if (bucket.getProject() != null) { 041 resourceModel.setRootId(bucket.getProject().getResourceId()); 042 } 043 044 // LIMITATION : project.getLanguage() is set only in ProjectBatch, not in AggregatorBatch, so we 045 // have to explicitely use project.getLanguageKey() 046 resourceModel.setLanguageKey(bucket.getResource().getLanguageKey()); 047 } 048 049 @Override 050 protected Snapshot createSnapshot(Bucket<Project> bucket, ResourceModel resourceModel) { 051 Project project = bucket.getResource(); 052 Snapshot parentSnapshot = (bucket.getParent() != null ? bucket.getParent().getSnapshot() : null); 053 Snapshot snapshot = new Snapshot(resourceModel, parentSnapshot); 054 snapshot.setVersion(project.getAnalysisVersion()); 055 snapshot.setCreatedAt(project.getAnalysisDate()); 056 return snapshot; 057 } 058 }