| // Copyright (C) 2016 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package com.google.gerrit.server.schema; |
| |
| import com.google.common.annotations.VisibleForTesting; |
| import com.google.gerrit.extensions.registration.DynamicItem; |
| import com.google.gerrit.lifecycle.LifecycleModule; |
| import com.google.gerrit.server.change.AccountPatchReviewStore; |
| import com.google.gerrit.server.config.GerritServerConfig; |
| import com.google.gerrit.server.config.SitePaths; |
| import com.google.gwtorm.server.OrmDuplicateKeyException; |
| import com.google.gwtorm.server.OrmException; |
| import com.google.inject.Inject; |
| import com.google.inject.Singleton; |
| |
| import org.eclipse.jgit.lib.Config; |
| |
| import java.sql.SQLException; |
| |
| @Singleton |
| public class H2AccountPatchReviewStore extends JdbcAccountPatchReviewStore { |
| |
| @VisibleForTesting |
| public static class InMemoryModule extends LifecycleModule { |
| @Override |
| protected void configure() { |
| H2AccountPatchReviewStore inMemoryStore = new H2AccountPatchReviewStore(); |
| DynamicItem.bind(binder(), AccountPatchReviewStore.class) |
| .toInstance(inMemoryStore); |
| listener().toInstance(inMemoryStore); |
| } |
| } |
| |
| @Inject |
| H2AccountPatchReviewStore(@GerritServerConfig Config cfg, |
| SitePaths sitePaths) { |
| super(cfg, sitePaths); |
| } |
| |
| /** |
| * Creates an in-memory H2 database to store the reviewed flags. |
| * This should be used for tests only. |
| */ |
| @VisibleForTesting |
| private H2AccountPatchReviewStore() { |
| // DB_CLOSE_DELAY=-1: By default the content of an in-memory H2 database is |
| // lost at the moment the last connection is closed. This option keeps the |
| // content as long as the vm lives. |
| super(createDataSource("jdbc:h2:mem:account_patch_reviews;DB_CLOSE_DELAY=-1")); |
| } |
| |
| @Override |
| public OrmException convertError(String op, SQLException err) { |
| switch (getSQLStateInt(err)) { |
| case 23001: // UNIQUE CONSTRAINT VIOLATION |
| case 23505: // DUPLICATE_KEY_1 |
| return new OrmDuplicateKeyException("account_patch_reviews", err); |
| |
| default: |
| if (err.getCause() == null && err.getNextException() != null) { |
| err.initCause(err.getNextException()); |
| } |
| return new OrmException(op + " failure on account_patch_reviews", err); |
| } |
| } |
| } |