com.senseidb.search.client.json.res.SenseiHitJsonHandler Maven / Gradle / Ivy

Go to download
package com.senseidb.search.client.json.res;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import com.senseidb.search.client.json.JsonDeserializer;
import com.senseidb.search.client.json.JsonHandler;

public class SenseiHitJsonHandler implements JsonHandler {

  private static final Set PREDEFINED_FIELDS= new HashSet(Arrays.asList("uid", "docid", "score", "srcdata", "grouphitscount", "groupHits", "stored", "termvectors", "explanation"));

  @Override
  public JSONObject serialize(SenseiHit bean) throws JSONException {
    throw new UnsupportedOperationException();
    // return (JSONObject) JsonSerializer.serialize(bean);

  }

  @Override
  public SenseiHit deserialize(JSONObject json) throws JSONException {
    if (json == null) {
      return null;
    }

    /*SenseiHit senseiHit = new SenseiHit();
    senseiHit.setDocid(json.optInt("docid"));
    senseiHit.setGrouphitscount(json.optInt("grouphitscount"));
    senseiHit.setSrcdata(json.optString("srcdata"));
    senseiHit.setUid(json.optInt("uid"));
    senseiHit.setScore(json.optInt("score"));
    JSONArray groupHitsArr = json.optJSONArray("groupHits");
    if (groupHitsArr != null){
      List groupHits = new ArrayList(groupHitsArr.length());
      for (int i = 0; i< groupHitsArr.length(); i++) {
        groupHits.add(deserialize(groupHitsArr.optJSONObject(i)));
      }
      senseiHit.setGroupHits(groupHits);
    }*/
    SenseiHit senseiHit =  JsonDeserializer.deserialize(SenseiHit.class, json, false);
    JSONArray storedFieldsArr = json.optJSONArray("stored");
    if (storedFieldsArr != null) {
      List storedFields = new ArrayList(storedFieldsArr.length());
      for (int i = 0; i< storedFieldsArr.length(); i++) {
        JSONObject storedJson = storedFieldsArr.optJSONObject(i);
        if (storedJson != null) {
          String fieldName = (String) storedJson.keys().next();
          storedFields.add(new FieldValue(fieldName, storedJson.optString(fieldName)));
        }
      }
      senseiHit.setStoredFields(storedFields);
    }
    Iterator iterator = json.keys();
    while (iterator.hasNext()) {
      String field = (String) iterator.next();
      if (PREDEFINED_FIELDS.contains(field)) {
        continue;
      }
      JSONArray jsonArr = json.optJSONArray(field);
      if (jsonArr != null) {
        List values = new ArrayList(jsonArr.length());
        for (int i = 0; i< jsonArr.length(); i++) {
          values.add(jsonArr.getString(i));
        }
        senseiHit.getFieldValues().put(field, values);
      }
    }

    return senseiHit;
  }
}