org.openrewrite.java.spring.data.MigrateQuerydslJpaRepository Maven / Gradle / Ivy
/*
* Copyright 2021 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.java.spring.data;
import org.openrewrite.ExecutionContext;
import org.openrewrite.Recipe;
import org.openrewrite.TreeVisitor;
import org.openrewrite.internal.lang.Nullable;
import org.openrewrite.java.ChangeType;
import org.openrewrite.java.JavaParser;
import org.openrewrite.java.JavaTemplate;
import org.openrewrite.java.JavaVisitor;
import org.openrewrite.java.search.UsesType;
import org.openrewrite.java.tree.*;
import java.util.List;
public class MigrateQuerydslJpaRepository extends Recipe {
@Override
public String getDisplayName() {
return "Use `QuerydslPredicateExecutor`";
}
@Override
public String getDescription() {
return "`QuerydslJpaRepository` was deprecated in Spring Data 2.1.";
}
@Nullable
@Override
protected TreeVisitor getSingleSourceApplicableTest() {
return new UsesType<>("org.springframework.data.jpa.repository.support.QuerydslJpaRepository");
}
@Override
protected JavaVisitor getVisitor() {
return new JavaVisitor() {
final String originalFqn = "org.springframework.data.jpa.repository.support.QuerydslJpaRepository";
final String targetFqn = "org.springframework.data.jpa.repository.support.QuerydslJpaPredicateExecutor";
@Override
public J visitCompilationUnit(J.CompilationUnit cu, ExecutionContext ctx) {
J.CompilationUnit c = (J.CompilationUnit)super.visitCompilationUnit(cu, ctx);
doAfterVisit(new ChangeType(originalFqn, targetFqn));
return c;
}
@Override
public J visitParameterizedType(J.ParameterizedType type, ExecutionContext context) {
J.ParameterizedType t = (J.ParameterizedType) super.visitParameterizedType(type, context);
if (t.getClazz() instanceof J.Identifier && TypeUtils.isOfType(TypeUtils.asFullyQualified(t.getClazz().getType()),
TypeUtils.asFullyQualified(JavaType.Class.build(originalFqn))) &&
t.getTypeParameters() != null && t.getTypeParameters().size() == 2) {
t = t.withTypeParameters(t.getTypeParameters().subList(0, 1));
}
return t;
}
@Override
public J visitNewClass(J.NewClass newClass, ExecutionContext context) {
if (newClass.getClazz() != null && TypeUtils.isOfClassType(newClass.getClazz().getType(), originalFqn) &&
newClass.getArguments() != null) {
String template = "new QuerydslJpaPredicateExecutor(#{any(org.springframework.data.jpa.repository.support.JpaEntityInformation)}, " +
"#{any(javax.persistence.EntityManager)}, " +
"#{any(org.springframework.data.querydsl.EntityPathResolver)}, null)";
J.FieldAccess entityPathResolver = TypeTree.build("SimpleEntityPathResolver.INSTANCE");
return newClass.withTemplate(
JavaTemplate.builder(this::getCursor, template)
.imports(originalFqn)
.javaParser(() -> JavaParser.fromJavaVersion()
.dependsOn(
"package javax.persistence;" +
"public interface EntityManager {}",
"package org.springframework.data.querydsl;" +
"public interface EntityPathResolver {}",
"package org.springframework.data.querydsl;" +
"public interface QuerydslPredicateExecutor {}",
"package org.springframework.data.repository.core;" +
"public interface EntityMetadata {}",
"package org.springframework.data.repository.core;" +
"public interface EntityInformation extends EntityMetadata {}",
"package org.springframework.data.jpa.repository.query;" +
"import org.springframework.data.repository.core.EntityMetadata;" +
"public interface JpaEntityMetadata extends EntityMetadata {",
"package org.springframework.data.jpa.repository.support;" +
"import org.springframework.data.repository.core.EntityInformation;" +
"import org.springframework.data.jpa.repository.query.JpaEntityMetadata;" +
"public interface JpaEntityInformation extends EntityInformation, JpaEntityMetadata {}",
"package org.springframework.data.jpa.repository.support;" +
"public interface CrudMethodMetadata {}",
"package org.springframework.data.jpa.repository.support;" +
"import javax.persistence.EntityManager;" +
"import org.springframework.data.querydsl.EntityPathResolver;" +
"import org.springframework.data.querydsl.QuerydslPredicateExecutor;" +
"public class QuerydslJpaPredicateExecutor implements QuerydslPredicateExecutor {" +
"public QuerydslJpaPredicateExecutor(" +
"JpaEntityInformation entityInformation," +
"EntityManager entityManager," +
"EntityPathResolver resolver," +
"@Nullable CrudMethodMetadata metadata) {}" +
"}",
"package org.springframework.data.querydsl;" +
"public class SimpleEntityPathResolver implements EntityPathResolver {" +
"public static final SimpleEntityPathResolver INSTANCE = new SimpleEntityPathResolver(\"\")" +
"public SimpleEntityPathResolver(String querySuffix) {" +
"Assert.notNull(querySuffix, \"Query suffix must not be null!\");" +
"this.querySuffix = querySuffix;}}"
).build())
.build(),
newClass.getCoordinates().replace(),
newClass.getArguments().get(0),
newClass.getArguments().get(1),
newClass.getArguments().size() == 3 ? newClass.getArguments().get(2) : entityPathResolver);
}
return super.visitNewClass(newClass, context);
}
};
}
}