org.openrewrite.maven.cache.InMemoryMavenPomCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rewrite-maven Show documentation
Eliminate tech-debt. Automatically.
The newest version!
/*
 * Copyright 2020 the original author or authors.
 * 
 * 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
 * 

 * https://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 org.openrewrite.maven.cache;

import com.github.benmanes.caffeine.cache.Cache;
import com.github.benmanes.caffeine.cache.Caffeine;
import io.micrometer.core.instrument.Metrics;
import io.micrometer.core.instrument.binder.cache.CaffeineCacheMetrics;
import lombok.Value;
import org.jspecify.annotations.Nullable;
import org.openrewrite.maven.tree.*;

import java.net.URI;
import java.util.Optional;

public class InMemoryMavenPomCache implements MavenPomCache {
    @Value
    public static class MetadataKey {
        URI repository;
        GroupArtifactVersion gav;
    }

    private final Cache> pomCache;
    private final Cache> mavenMetadataCache;
    private final Cache> repositoryCache;
    private final Cache dependencyCache;

    public InMemoryMavenPomCache() {
        this(
                Caffeine.newBuilder()
                        .recordStats()
                        .maximumSize(100_000)
                        .build(),
                Caffeine.newBuilder()
                        .recordStats()
                        .maximumSize(100_000)
                        .build(),
                Caffeine.newBuilder()
                        .recordStats()
                        .maximumSize(10_000)
                        .build(),
                Caffeine.newBuilder()
                        .recordStats()
                        .maximumSize(100_000)
                        .build()
        );
    }

    public InMemoryMavenPomCache(String cacheNickname,
                                 Cache> pomCache,
                                 Cache> mavenMetadataCache,
                                 Cache> repositoryCache,
                                 Cache dependencyCache) {

        this.pomCache = CaffeineCacheMetrics.monitor(Metrics.globalRegistry, pomCache, "Maven POMs - " + cacheNickname);
        this.mavenMetadataCache = CaffeineCacheMetrics.monitor(Metrics.globalRegistry, mavenMetadataCache, "Maven metadata - " + cacheNickname);
        this.repositoryCache = CaffeineCacheMetrics.monitor(Metrics.globalRegistry, repositoryCache, "Maven repositories - " + cacheNickname);
        this.dependencyCache = CaffeineCacheMetrics.monitor(Metrics.globalRegistry, dependencyCache, "Resolved dependency POMs - " + cacheNickname);
    }

    public InMemoryMavenPomCache(Cache> pomCache,
                                 Cache> mavenMetadataCache,
                                 Cache> repositoryCache,
                                 Cache dependencyCache) {
        this("default", pomCache, mavenMetadataCache, repositoryCache, dependencyCache);
    }

    @Override
    public @Nullable ResolvedPom getResolvedDependencyPom(ResolvedGroupArtifactVersion dependency) {
        return dependencyCache.getIfPresent(dependency);
    }

    @Override
    public void putResolvedDependencyPom(ResolvedGroupArtifactVersion dependency, ResolvedPom resolved) {
        dependencyCache.put(dependency, resolved.deduplicate());
    }

    @Override
    public @Nullable Optional getMavenMetadata(URI repo, GroupArtifactVersion gav) {
        return mavenMetadataCache.getIfPresent(new MetadataKey(repo, gav));
    }

    @Override
    public void putMavenMetadata(URI repo, GroupArtifactVersion gav, @Nullable MavenMetadata metadata) {
        mavenMetadataCache.put(new MetadataKey(repo, gav), Optional.ofNullable(metadata));
    }

    @Override
    public @Nullable Optional getPom(ResolvedGroupArtifactVersion gav) {
        return pomCache.getIfPresent(gav);
    }

    @Override
    public void putPom(ResolvedGroupArtifactVersion gav, @Nullable Pom pom) {
        pomCache.put(gav, Optional.ofNullable(pom));
    }

    @Override
    public @Nullable Optional getNormalizedRepository(MavenRepository repository) {
        return repositoryCache.getIfPresent(repository);
    }

    @Override
    public void putNormalizedRepository(MavenRepository repository, MavenRepository normalized) {
        repositoryCache.put(repository, Optional.ofNullable(normalized));
    }
}