1 files changed, 123 insertions, 0 deletions
diff --git a/org.eclipse.jdt.core/search/org/eclipse/jdt/internal/core/nd/field/FieldSearchKey.java b/org.eclipse.jdt.core/search/org/eclipse/jdt/internal/core/nd/field/FieldSearchKey.java
new file mode 100644
index 000000000..1b585cb5d
--- /dev/null
+++ b/org.eclipse.jdt.core/search/org/eclipse/jdt/internal/core/nd/field/FieldSearchKey.java
@@ -0,0 +1,123 @@
+/*******************************************************************************
+ * Copyright (c) 2015, 2016 Google, Inc and others.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *   Stefan Xenos (Google) - Initial implementation
+ *******************************************************************************/
+package org.eclipse.jdt.internal.core.nd.field;
+
+import org.eclipse.jdt.internal.core.nd.Nd;
+import org.eclipse.jdt.internal.core.nd.NdNode;
+import org.eclipse.jdt.internal.core.nd.db.BTree;
+import org.eclipse.jdt.internal.core.nd.db.Database;
+import org.eclipse.jdt.internal.core.nd.db.EmptyString;
+import org.eclipse.jdt.internal.core.nd.db.IString;
+
+/**
+ * Represents a search key into a global search index.
+ */
+public class FieldSearchKey<T> implements IField, IDestructableField {
+	private int offset;
+	FieldSearchIndex<?> searchIndex;
+
+	private FieldSearchKey(FieldSearchIndex<?> searchIndex) {
+		if (searchIndex != null) {
+			if (searchIndex.searchKey != null && searchIndex.searchKey != this) {
+				throw new IllegalArgumentException(
+					"Attempted to construct a FieldSearchKey referring to a search index that is " //$NON-NLS-1$
+					+ "already in use by a different key"); //$NON-NLS-1$
+			}
+			searchIndex.searchKey = this;
+		}
+		this.searchIndex = searchIndex;
+	}
+
+	/**
+	 * Creates a search key attribute in the given struct which stores an entry in the given global search index
+	 */
+	public static <T, B extends NdNode> FieldSearchKey<T> create(StructDef<B> builder,
+			FieldSearchIndex<B> searchIndex) {
+		FieldSearchKey<T> result = new FieldSearchKey<T>(searchIndex);
+
+		builder.add(result);
+		builder.addDestructableField(result);
+
+		return result;
+	}
+
+	public void put(Nd nd, long address, String newString) {
+		put(nd, address, newString.toCharArray());
+	}
+
+	/**
+	 * Sets the value of the key and inserts it into the index if it is not already present
+	 */
+	public void put(Nd nd, long address, char[] newString) {
+		cleanup(nd, address);
+
+		Database db = nd.getDB();
+		BTree btree = this.searchIndex.get(nd, Database.DATA_AREA_OFFSET);
+		db.putRecPtr(address + this.offset, db.newString(newString).getRecord());
+		btree.insert(address);
+	}
+
+	public IString get(Nd nd, long address) {
+		Database db = nd.getDB();
+		long namerec = db.getRecPtr(address + this.offset);
+
+		if (namerec == 0) {
+			return EmptyString.create();
+		}
+		return db.getString(namerec);
+	}
+
+	@Override
+	public void destruct(Nd nd, long address) {
+		cleanup(nd, address);
+	}
+
+	private void cleanup(Nd nd, long address) {
+		boolean isInIndex = isInIndex(nd, address);
+
+		if (isInIndex) {
+			// Remove this entry from the search index
+			this.searchIndex.get(nd, Database.DATA_AREA_OFFSET).delete(address);
+
+			get(nd, address).delete();
+			nd.getDB().putRecPtr(address + this.offset, 0);
+		}
+	}
+
+	/**
+	 * Clears this key and removes it from the search index
+	 */
+	public void removeFromIndex(Nd nd, long address) {
+		cleanup(nd, address);
+	}
+
+	/**
+	 * Returns true iff this key is currently in the index
+	 */
+	public boolean isInIndex(Nd nd, long address) {
+		long fieldAddress = address + this.offset;
+		Database db = nd.getDB();
+		long namerec = db.getRecPtr(fieldAddress);
+
+		boolean isInIndex = namerec != 0;
+		return isInIndex;
+	}
+
+	@Override
+	public void setOffset(int offset) {
+		this.offset = offset;
+	}
+
+	@Override
+	public int getRecordSize() {
+		return FieldString.RECORD_SIZE;
+	}
+}