Cache project metadata and tokens

it/server/src/test/java/com/linecorp/centraldogma/it/WriteQuotaTestBase.java

@@ -18,6 +18,7 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.awaitility.Awaitility.await;
 
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
@@ -64,8 +65,8 @@ void updateWriteQuota() throws Exception {
         CompletableFutures.allAsList(futures).join();
 
         /// update write quota to 2qps
-        QuotaConfig writeQuota = new QuotaConfig(2, 1);
-        QuotaConfig updated = updateWriteQuota(webClient(), repositoryName, writeQuota);
+        final QuotaConfig writeQuota = new QuotaConfig(2, 1);
+        final QuotaConfig updated = updateWriteQuota(webClient(), repositoryName, writeQuota);
         assertThat(updated).isEqualTo(writeQuota);
 
         // Wait for releasing previously acquired locks
@@ -86,9 +87,9 @@ void updateWriteQuota() throws Exception {
         Thread.sleep(1000);
 
         // Increase write quota
-        writeQuota = new QuotaConfig(5, 1);
-        updated = updateWriteQuota(webClient(), repositoryName, writeQuota);
-        assertThat(updated).isEqualTo(writeQuota);
+        final QuotaConfig writeQuota1 = new QuotaConfig(5, 1);
+        await().untilAsserted(() -> assertThat(updateWriteQuota(webClient(), repositoryName, writeQuota1))
+                .isEqualTo(writeQuota1));
 
         final List<CompletableFuture<PushResult>> futures4 = parallelPush(dogmaClient(), repositoryName, 4);
         assertThat(CompletableFutures.allAsList(futures4).join()).hasSize(8);

server/src/main/java/com/linecorp/centraldogma/server/CentralDogma.java

@@ -668,7 +668,8 @@ private Server startServer(ProjectManager pm, CommandExecutor executor,
         cfg.gracefulShutdownTimeout().ifPresent(
                 t -> sb.gracefulShutdownTimeoutMillis(t.quietPeriodMillis(), t.timeoutMillis()));
 
-        final MetadataService mds = new MetadataService(pm, executor);
+        final MetadataService mds = new MetadataService(pm, executor, projectInitializer);
+        executor.setRepositoryMetadataSupplier(mds::getRepo);
         final WatchService watchService = new WatchService(meterRegistry);
         final AuthProvider authProvider = createAuthProvider(executor, sessionManager, mds);
         final ProjectApiManager projectApiManager = new ProjectApiManager(pm, executor, mds);
@@ -796,7 +797,7 @@ private CommandExecutor newZooKeeperCommandExecutor(
                 new StandaloneCommandExecutor(pm, repositoryWorker, serverStatusManager, sessionManager,
                         /* onTakeLeadership */ null, /* onReleaseLeadership */ null,
                         /* onTakeZoneLeadership */ null, /* onReleaseZoneLeadership */ null),
-                meterRegistry, pm, config().writeQuotaPerRepository(), zone,
+                meterRegistry, config().writeQuotaPerRepository(), zone,
                 onTakeLeadership, onReleaseLeadership,
                 onTakeZoneLeadership, onReleaseZoneLeadership);
     }

server/src/main/java/com/linecorp/centraldogma/server/command/AbstractCommandExecutor.java

@@ -22,6 +22,7 @@
 import java.util.concurrent.CompletionStage;
 import java.util.concurrent.ForkJoinPool;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.function.BiFunction;
 import java.util.function.Consumer;
 
 import javax.annotation.Nullable;
@@ -34,6 +35,7 @@
 import com.linecorp.armeria.common.util.Exceptions;
 import com.linecorp.armeria.common.util.StartStopSupport;
 import com.linecorp.centraldogma.common.ReadOnlyException;
+import com.linecorp.centraldogma.server.metadata.RepositoryMetadata;
 
 /**
  * Helps to implement a concrete {@link CommandExecutor}.
@@ -57,6 +59,9 @@ public abstract class AbstractCommandExecutor implements CommandExecutor {
     private final AtomicInteger numPendingStopRequests = new AtomicInteger();
     private final CommandExecutorStatusManager statusManager;
 
+    @Nullable
+    private BiFunction<String, String, CompletableFuture<RepositoryMetadata>> repositoryMetadataSupplier;
+
     /**
      * Creates a new instance.
      *
@@ -120,6 +125,23 @@ public final void setWritable(boolean writable) {
         this.writable = writable;
     }
 
+    @Override
+    public void setRepositoryMetadataSupplier(
+            BiFunction<String, String, CompletableFuture<RepositoryMetadata>> supplier) {
+        repositoryMetadataSupplier = requireNonNull(supplier, "supplier");
+    }
+
+    /**
+     * Returns the {@link RepositoryMetadata} of the specified repository.
+     */
+    @Nullable
+    protected CompletableFuture<RepositoryMetadata> repositoryMetadata(String projectName, String repoName) {
+        if (repositoryMetadataSupplier == null) {
+            return null;
+        }
+        return repositoryMetadataSupplier.apply(projectName, repoName);
+    }
+
     @Override
     public final <T> CompletableFuture<T> execute(Command<T> command) {
         requireNonNull(command, "command");

server/src/main/java/com/linecorp/centraldogma/server/command/CommandExecutor.java

@@ -17,10 +17,12 @@
 package com.linecorp.centraldogma.server.command;
 
 import java.util.concurrent.CompletableFuture;
+import java.util.function.BiFunction;
 
 import javax.annotation.Nullable;
 
 import com.linecorp.centraldogma.server.QuotaConfig;
+import com.linecorp.centraldogma.server.metadata.RepositoryMetadata;
 
 /**
  * An executor interface which executes {@link Command}s.
@@ -66,6 +68,12 @@ public interface CommandExecutor {
      */
     void setWriteQuota(String projectName, String repoName, @Nullable QuotaConfig writeQuota);
 
+    /**
+     * Sets the {@link RepositoryMetadata} supplier.
+     */
+    void setRepositoryMetadataSupplier(
+            BiFunction<String, String, CompletableFuture<RepositoryMetadata>> supplier);
+
     /**
      * Executes the specified {@link Command}.
      *

server/src/main/java/com/linecorp/centraldogma/server/command/ForwardingCommandExecutor.java

@@ -19,9 +19,11 @@
 import static java.util.Objects.requireNonNull;
 
 import java.util.concurrent.CompletableFuture;
+import java.util.function.BiFunction;
 
 import com.linecorp.centraldogma.internal.Util;
 import com.linecorp.centraldogma.server.QuotaConfig;
+import com.linecorp.centraldogma.server.metadata.RepositoryMetadata;
 
 /**
  * A {@link CommandExecutor} which forwards all its method calls to another {@link CommandExecutor}.
@@ -73,6 +75,12 @@ public void setWriteQuota(String projectName, String repoName, QuotaConfig write
         delegate().setWriteQuota(projectName, repoName, writeQuota);
     }
 
+    @Override
+    public void setRepositoryMetadataSupplier(
+            BiFunction<String, String, CompletableFuture<RepositoryMetadata>> supplier) {
+        delegate().setRepositoryMetadataSupplier(supplier);
+    }
+
     @Override
     public <T> CompletableFuture<T> execute(Command<T> command) {
         return delegate().execute(command);