Line data    Source code

       1             : // Copyright (C) 2021 The Android Open Source Project
       2             : //
       3             : // Licensed under the Apache License, Version 2.0 (the "License");
       4             : // you may not use this file except in compliance with the License.
       5             : // You may obtain a copy of the License at
       6             : //
       7             : // http://www.apache.org/licenses/LICENSE-2.0
       8             : //
       9             : // Unless required by applicable law or agreed to in writing, software
      10             : // distributed under the License is distributed on an "AS IS" BASIS,
      11             : // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      12             : // See the License for the specific language governing permissions and
      13             : // limitations under the License.
      14             : 
      15             : package com.google.gerrit.index.testing;
      16             : 
      17             : import com.google.common.collect.Iterables;
      18             : import com.google.gerrit.index.Index;
      19             : import com.google.gerrit.index.IndexDefinition;
      20             : import com.google.gerrit.index.Schema;
      21             : import com.google.gerrit.server.config.GerritServerConfig;
      22             : import com.google.gerrit.server.config.SitePaths;
      23             : import com.google.gerrit.server.index.GerritIndexStatus;
      24             : import com.google.gerrit.server.index.OnlineUpgradeListener;
      25             : import com.google.gerrit.server.index.VersionManager;
      26             : import com.google.gerrit.server.plugincontext.PluginSetContext;
      27             : import com.google.inject.Inject;
      28             : import com.google.inject.Singleton;
      29             : import java.util.Collection;
      30             : import java.util.TreeMap;
      31             : import org.eclipse.jgit.lib.Config;
      32             : 
      33             : /** Fake version manager for {@link AbstractFakeIndex}. */
      34             : @Singleton
      35             : public class FakeIndexVersionManager extends VersionManager {
      36             : 
      37             :   @Inject
      38             :   FakeIndexVersionManager(
      39             :       @GerritServerConfig Config cfg,
      40             :       SitePaths sitePaths,
      41             :       PluginSetContext<OnlineUpgradeListener> listeners,
      42             :       Collection<IndexDefinition<?, ?, ?>> defs) {
      43         138 :     super(sitePaths, listeners, defs, VersionManager.getOnlineUpgrade(cfg));
      44         138 :   }
      45             : 
      46             :   @Override
      47             :   protected <K, V, I extends Index<K, V>> TreeMap<Integer, Version<V>> scanVersions(
      48             :       IndexDefinition<K, V, I> def, GerritIndexStatus cfg) {
      49         138 :     TreeMap<Integer, Version<V>> versions = new TreeMap<>();
      50         138 :     for (Schema<V> schema : def.getSchemas().values()) {
      51         138 :       int v = schema.getVersion();
      52         138 :       boolean exists = versions.containsKey(v);
      53         138 :       versions.put(v, new Version<>(schema, v, exists, cfg.getReady(def.getName(), v)));
      54         138 :     }
      55         138 :     return versions;
      56             :   }
      57             : 
      58             :   @Override
      59             :   protected <K, V, I extends Index<K, V>> void initIndex(
      60             :       IndexDefinition<K, V, I> def, GerritIndexStatus cfg) {
      61             :     // Set latest versions ready.
      62         138 :     if (def.getSchemas().isEmpty()) {
      63           0 :       super.initIndex(def, cfg);
      64           0 :       return;
      65             :     }
      66         138 :     Schema<V> schema = Iterables.getLast(def.getSchemas().values());
      67             :     try {
      68         138 :       cfg.setReady(def.getName(), schema.getVersion(), true);
      69         138 :       cfg.save();
      70           0 :     } catch (Exception e) {
      71           0 :       throw new IllegalStateException(e);
      72         138 :     }
      73         138 :     super.initIndex(def, cfg);
      74         138 :   }
      75             : }