com.redhat.lightblue.mindex.SimpleKeySpec Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lightblue-core-crud Show documentation
There is a newer version: 2.18.0
 /*
 Copyright 2013 Red Hat, Inc. and/or its affiliates.

 This file is part of lightblue.

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see .
 */
package com.redhat.lightblue.mindex;

import java.util.Set;
import java.util.HashSet;
import java.util.Comparator;

import com.fasterxml.jackson.databind.JsonNode;

import com.redhat.lightblue.metadata.Type;
import com.redhat.lightblue.metadata.FieldTreeNode;
import com.redhat.lightblue.metadata.ArrayField;

import com.redhat.lightblue.assoc.QueryFieldInfo;

import com.redhat.lightblue.util.JsonDoc;
import com.redhat.lightblue.util.Path;
import com.redhat.lightblue.util.KeyValueCursor;

/**
 * Simple key specification, includes a single field
 */
public class SimpleKeySpec implements KeySpec,Comparator {
    final FieldTreeNode fieldMd;
    final Path fullName;
    final Type type;

    public SimpleKeySpec(QueryFieldInfo finfo) {
        this.fieldMd=finfo.getFieldMd();
        this.fullName=finfo.getEntityRelativeFieldName();
        this.type=fieldMd.getType();
    }


    @Override
    public int compareKeys(Key k1,Key k2) {
        return compare( (SimpleKey)k1,(SimpleKey)k2);
    }
    
    @Override
    public int compare(SimpleKey v1,SimpleKey v2) {
        return type.compare(v1.value,v2.value);
    }
    
    @Override
    public Set extract(JsonDoc doc,Set set) {
        if(set==null)
            set=new HashSet<>();
        KeyValueCursor cursor=doc.getAllNodes(fullName);
        boolean hasData=false;
        while(cursor.hasNext()) {
            cursor.next();
            set.add(new SimpleKey(type.fromJson(cursor.getCurrentValue())));
            hasData=true;
        }
        if(!hasData) {
        	// No value in doc: insert null
        	set.add(SimpleKey.NULL_KEY);
        }
        return set;
    }
}