org.hibernate.hql.internal.ast.util.ASTReferencedTablesPrinter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of hibernate-core Show documentation

Hibernate's core ORM functionality

There is a newer version: 7.0.0.Alpha1

/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.hql.internal.ast.util;

import java.util.Arrays;
import java.util.LinkedHashMap;

import org.hibernate.hql.internal.ast.tree.DotNode;
import org.hibernate.hql.internal.ast.tree.FromElement;
import org.hibernate.hql.internal.ast.tree.FromReferenceNode;
import org.hibernate.hql.internal.ast.tree.IdentNode;
import org.hibernate.hql.internal.ast.tree.SelectClause;
import org.hibernate.persister.entity.AbstractEntityPersister;
import org.hibernate.persister.entity.EntityPersister;

import antlr.collections.AST;

public class ASTReferencedTablesPrinter extends ASTPrinter {

	public ASTReferencedTablesPrinter(Class tokenTypeConstants) {
		super( tokenTypeConstants );
	}

	@Override
	public String nodeToString(AST ast) {
		if ( ast == null ) {
			return "{node:null}";
		}
		return ast.getClass().getSimpleName();
	}

	@Override
	public LinkedHashMap createNodeProperties(AST node) {
		LinkedHashMap props = new LinkedHashMap<>();
		if ( node instanceof FromReferenceNode ) {
			FromReferenceNode frn = (FromReferenceNode) node;
			FromElement fromElement = frn.getFromElement();
			EntityPersister entityPersister = fromElement != null ? fromElement.getEntityPersister() : null;
			String entityPersisterStr = entityPersister != null ? entityPersister.toString() : null;
			props.put( "persister", entityPersisterStr );
		}
		if ( node instanceof DotNode ) {
			DotNode dn = (DotNode) node;
			props.put( "path", dn.getPath() );
		}
		if ( node instanceof IdentNode ) {
			IdentNode in = (IdentNode) node;
			props.put( "originalText", in.getOriginalText() );
		}
		if ( node instanceof SelectClause ) {
			SelectClause sc = (SelectClause) node;
			for ( Object element : sc.getFromElementsForLoad() ) {
				FromElement fromElement = (FromElement) element;
				EntityPersister entityPersister = fromElement.getEntityPersister();
				if ( entityPersister != null && entityPersister instanceof AbstractEntityPersister ) {
					AbstractEntityPersister aep = (AbstractEntityPersister) entityPersister;
					String entityClass = aep.getMappedClass().getSimpleName();
					String tables = Arrays.toString( aep.getTableNames() );
					props.put( String.format( "referencedTables(entity %s)", entityClass ), tables );
				}
			}
		}
		return props;
	}
}