@Entity
@Indexed
public class ElementSet {
	private int id;

	private Element next;

	@Id
	@GeneratedValue
	public int getId() {
		return id;
	}

	public void setId(int id) {
		this.id = id;
	}

	@IndexedEmbedded(prefix = "next_", depth = 2)
	@ManyToOne(fetch = FetchType.EAGER)
	public Element getNext() {
		return next;
	}

	public void setNext(Element next) {
		this.next = next;
	}

	@Override
	public String toString() {
		final StringBuilder sb = new StringBuilder( "ElementSet{" );
		sb.append( "id=" ).append( id );
		sb.append( ", next=" ).append( next );
		sb.append( '}' );
		return sb.toString();
	}
}

@Entity
public class Element {
	private int id;
	private String name;
	private Set<Attribute> elementAttributes = new HashSet<Attribute>();

	@Id
	@GeneratedValue
	public int getId() {
		return id;
	}

	public void setId(int id) {
		this.id = id;
	}

	@IndexedEmbedded(depth = 1, prefix = "attribute_")
	@ManyToMany(fetch = FetchType.LAZY)
	public Set<Attribute> getElementAttributes() {
		return elementAttributes;
	}

	public void setElementAttributes(Set<Attribute> elementAttributes) {
		this.elementAttributes = elementAttributes;
	}

	public void addAttribute(Attribute attribute) {
		elementAttributes.add( attribute );
	}

	@Field(store = Store.YES)
	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	@Override
	public String toString() {
		final StringBuilder sb = new StringBuilder( "Element{" );
		sb.append( "id=" ).append( id );
		sb.append( ", name='" ).append( name ).append( '\'' );
		sb.append( ", elementAttributes=" ).append( elementAttributes );
		sb.append( '}' );
		return sb.toString();
	}
}

@Entity
public class Attribute {
	private int id;

	private String value;

	@Id
	@GeneratedValue
	public int getId() {
		return id;
	}

	public void setId(int id) {
		this.id = id;
	}

	@Field(store = Store.YES)
	public String getValue() {
		return value;
	}

	public void setValue(String value) {
		this.value = value;
	}

	@Override
	public String toString() {
		final StringBuilder sb = new StringBuilder( "Attribute{" );
		sb.append( "id=" ).append( id );
		sb.append( ", value='" ).append( value ).append( '\'' );
		sb.append( '}' );
		return sb.toString();
	}
}

public class IndexedEmbeddedWithPrefixAndDepthTest extends SearchTestCaseJUnit4 {

	@Override
	@Before
	public void setUp() throws Exception {
		super.setUp();
		Attribute fooAttribute = new Attribute();
		fooAttribute.setValue( "foo" );

		Attribute barAttribute = new Attribute();
		barAttribute.setValue( "bar" );

		Element element = new Element();
		element.setName( "snafu" );
		element.addAttribute( fooAttribute );
		element.addAttribute( barAttribute );

		ElementSet set = new ElementSet();
		set.setNext( element );

		persist( fooAttribute, barAttribute, element, set );
	}

	@Test
	public void testAllExpectedFieldsInLuceneDocument() throws Exception {
		SearchFactory searchFactory = getSearchFactory();
		IndexReader indexReader = searchFactory.getIndexReaderAccessor().open( ElementSet.class );
		assertEquals( "There should be a single document in the index", 1, indexReader.numDocs() );
		Document document = indexReader.document( 0 );
		assertNotNull( document );

		assertDocumentContainsField( document, "id" );
		assertDocumentContainsField( document, "next_id" );
		assertDocumentContainsField( document, "next_name" );
		assertDocumentContainsField( document, "next_attribute_id" );
		assertDocumentContainsField( document, "next_attribute_value" );
	}

	private void assertDocumentContainsField(Document document, String fieldName) {
		String fieldValue = document.get( fieldName );
		assertNotNull( "The field '" + fieldName + "' should exist in the document", fieldValue );
	}

	@Override
	protected Class<?>[] getAnnotatedClasses() {
		return new Class<?>[] {
				ElementSet.class, Element.class, Attribute.class
		};
	}
}