com.dell.doradus.search.FieldSetCreator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of doradus-server Show documentation
The newest version!
/*
 * Copyright (C) 2014 Dell, Inc.
 * 
 * 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
 * 
 *     http://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 com.dell.doradus.search;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

import com.dell.doradus.common.CommonDefs;
import com.dell.doradus.common.TableDefinition;
import com.dell.doradus.search.aggregate.Entity;
import com.dell.doradus.search.aggregate.EntitySequence;
import com.dell.doradus.search.aggregate.SortOrder;
import com.dell.doradus.search.filter.Filter;
import com.dell.doradus.search.util.HeapList;

public class FieldSetCreator {
	/**
	 * limits the number of 
	 */
	public int limit;
	public Filter filter;
	public SortOrder[] orders;
	public TableDefinition tableDef;
	public List scalarFields;
	public List loadedFields;
	public TreeMap> links = new TreeMap>();
	public FieldSet fieldSet;
	
	public FieldSetCreator(FieldSet fieldSet, SortOrder[] orders) {
		tableDef = fieldSet.tableDef;
		limit = fieldSet.limit;
		if(limit == -1) limit = Integer.MAX_VALUE;
		scalarFields = fieldSet.ScalarFields;
		loadedFields = scalarFields;
		this.orders = orders;
		this.fieldSet = fieldSet;
		
		Set flds = new HashSet(scalarFields.size());
		flds.addAll(scalarFields);
		if(fieldSet.filter != null) {
			filter = new QueryExecutor(tableDef).filter(fieldSet.filter);
			filter.addFields(flds);
		}
		//if(orders != null) {
		//	if(order.items.size() != 1) throw new IllegalArgumentException("Link paths not supported in sort order");
		//	if(!order.items.get(0).isLink) {
		//		flds.add(order.items.get(0).name);
		//	}
		//}
		if(flds.contains("*")) {
			flds.clear();
			flds.add("*");
		}
		loadedFields = new ArrayList(flds);
		
		for(String link: fieldSet.getLinks()) {
			List list = new ArrayList();
			links.put(link, list);
			for(FieldSet fs: fieldSet.getLinks(link)) {
				list.add(new FieldSetCreator(fs, null));
			}
		}
		
	}
	
	public SearchResultList create(EntitySequence sequence, int skip) {
		int maxSize = limit;
		if (limit < Integer.MAX_VALUE - skip) {
			maxSize += skip;
		}
		
		SearchResultList resultList = new SearchResultList();
		
		if(orders == null || limit == Integer.MAX_VALUE) {
	        for(Entity entity: sequence) {
	        	if(filter != null && !filter.check(entity)) continue;
	        	SearchResult result = createResult(entity);
	            resultList.results.add(result);
	            if(resultList.results.size() >= maxSize) break;
	        }
		}
		else {
			HeapList results = new HeapList(maxSize);
	        for(Entity entity: sequence) {
	        	if(filter != null && !filter.check(entity)) continue;
	        	SearchResult result = createResult(entity);
	            results.Add(result);
	        }
	        SearchResult[] arr = results.GetValues(SearchResult.class);
	        for(SearchResult result : arr) resultList.results.add(result);
		}
        if(limit == Integer.MAX_VALUE && orders != null) Collections.sort(resultList.results);
        if (skip > 0) {
        	if (skip > resultList.results.size()) {
        		resultList.results.clear();
        	} else {
        		resultList.results.subList(0, skip).clear();
        	}
        }
        if(resultList.results.size() >= limit) resultList.continuation_token = resultList.results.get(resultList.results.size() - 1).id.toString();
        return resultList;
	}

	private SearchResult createResult(Entity entity) {
    	SearchResult result = new SearchResult();
        result.scalars.put(CommonDefs.ID_FIELD, entity.id().toString());
        result.id = entity.id();
        result.orders = orders;
        result.fieldSet = fieldSet;
        for(String f: scalarFields) {
            if(f.equals(CommonDefs.ID_FIELD)) continue;
            else if(f.equals("*")) {
            	Iterable allFields = entity.getAllFields();
            	if(allFields != null) {
	                for(String field: allFields) {
	                    String value = entity.get(field);
	                    if(value != null) result.scalars.put(field, value);
	                }
            	}
            }else {
                String v = entity.get(f);
                if(v != null) result.scalars.put(f, v);
            }
        }
        for(Map.Entry> entry: links.entrySet()) {
            String linkName = entry.getKey();
            List linkedSetList = entry.getValue();
            List searchResultList = new ArrayList();
            result.links.put(linkName, searchResultList);
            for(FieldSetCreator linkedSet: linkedSetList) {
	            EntitySequence linkedSequence = entity.getLinkedEntities(linkName, linkedSet.loadedFields);
	            SearchResultList linkedResultList = linkedSet.create(linkedSequence, 0);
	            searchResultList.add(linkedResultList);
            }
        }
    	return result;
	}
}