plugins/org.eclipse.mat.api/src/org/eclipse/mat/inspections/FindStringsQuery.java

/*******************************************************************************
 * Copyright (c) 2008, 2010 SAP AG.
 * 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:
 *    SAP AG - initial API and implementation
 *******************************************************************************/
package org.eclipse.mat.inspections;

import java.util.Collection;
import java.util.regex.Pattern;

import org.eclipse.mat.collect.ArrayInt;
import org.eclipse.mat.internal.Messages;
import org.eclipse.mat.query.IQuery;
import org.eclipse.mat.query.IResult;
import org.eclipse.mat.query.annotations.Argument;
import org.eclipse.mat.query.annotations.CommandName;
import org.eclipse.mat.query.annotations.Icon;
import org.eclipse.mat.snapshot.ISnapshot;
import org.eclipse.mat.snapshot.model.IClass;
import org.eclipse.mat.snapshot.model.IObject;
import org.eclipse.mat.snapshot.query.IHeapObjectArgument;
import org.eclipse.mat.snapshot.query.ObjectListResult;
import org.eclipse.mat.util.IProgressListener;

@CommandName("find_strings")
@Icon("/META-INF/icons/find_strings.gif")
public class FindStringsQuery implements IQuery
{
    @Argument
    public ISnapshot snapshot;

    @Argument(isMandatory = false)
    public IHeapObjectArgument objects;

    @Argument
    public Pattern pattern;

    public IResult execute(IProgressListener listener) throws Exception
    {
        ArrayInt result = new ArrayInt();

        Collection<IClass> classes = snapshot.getClassesByName("java.lang.String", false); //$NON-NLS-1$
        if (objects == null)
        {
            if (classes != null)
                ClassesLoop: for (IClass clasz : classes)
                {
                    int[] objectIds = clasz.getObjectIds();

                    listener.beginTask(Messages.FindStringsQuery_SearchingStrings, objectIds.length);

                    for (int id : objectIds)
                    {
                        if (listener.isCanceled())
                            break ClassesLoop;

                        String value = snapshot.getObject(id).getClassSpecificName();
                        if (value != null && pattern.matcher(value).matches())
                            result.add(id);

                        listener.worked(1);
                    }

                    listener.done();
                }
        }
        else
        {
            if (classes != null && !classes.isEmpty())
            {
                IClass javaLangString = classes.iterator().next();

                int totalWork = 0;
                for (int[] objectIds : objects)
                {
                    totalWork += objectIds.length;
                }

                listener.beginTask(Messages.FindStringsQuery_SearchingStrings, totalWork);

                ObjectsLoop: for (int[] objectIds : objects)
                {
                    for (int id : objectIds)
                    {
                        if (listener.isCanceled())
                            break ObjectsLoop;

                        if (snapshot.isArray(id) || snapshot.isClass(id) || snapshot.isClassLoader(id))
                            continue;

                        IObject instance = snapshot.getObject(id);
                        // if (!classes.contains(instance.getClazz()))
                        if (!javaLangString.equals(instance.getClazz()))
                            continue;

                        String value = instance.getClassSpecificName();
                        if (value != null && pattern.matcher(value).matches())
                        {
                            result.add(id);
                        }

                        listener.worked(1);
                    }
                }

                listener.done();
            }
        }

        if (listener.isCanceled() && result.isEmpty())
            throw new IProgressListener.OperationCanceledException();

        return new ObjectListResult.Outbound(snapshot, result.toArray());
    }

}