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.resources.Library; 024 import org.sonar.api.resources.Project; 025 import org.sonar.api.resources.Resource; 026 027 import java.util.HashMap; 028 import java.util.Map; 029 030 public final class ResourcePersisters { 031 032 private Map<Class<? extends Resource>, ResourcePersister> persistersByClass; 033 private ResourcePersister defaultPersister; 034 035 public ResourcePersisters(DatabaseSession session) { 036 defaultPersister = new DefaultPersister(session); 037 persistersByClass = new HashMap<Class<? extends Resource>, ResourcePersister>(); 038 persistersByClass.put(Project.class, new ProjectPersister(session)); 039 persistersByClass.put(Library.class, new LibraryPersister(session)); 040 } 041 042 public ResourcePersister get(Bucket bucket) { 043 return get(bucket.getResource()); 044 } 045 046 public ResourcePersister get(Resource resource) { 047 ResourcePersister persister = persistersByClass.get(resource.getClass()); 048 return persister != null ? persister : defaultPersister; 049 } 050 }