org.sonar.server.search.ws.BaseMapping Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sonar-server Show documentation
There is a newer version: 7.2.1
/*
 * SonarQube
 * Copyright (C) 2009-2016 SonarSource SA
 * mailto:contact AT sonarsource DOT com
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package org.sonar.server.search.ws;

import com.google.common.collect.LinkedHashMultimap;
import com.google.common.collect.Multimap;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import javax.annotation.Nullable;
import org.sonar.api.server.ServerSide;
import org.sonar.api.utils.text.JsonWriter;
import org.sonar.server.search.BaseDoc;
import org.sonar.server.search.IndexUtils;
import org.sonar.server.search.QueryContext;

/**
 * Mapping of search documents (see {@link org.sonar.server.search.BaseDoc}) to WS JSON responses
 */
@ServerSide
public abstract class BaseMapping {

  private final Multimap indexFieldsByWsFields = LinkedHashMultimap.create();
  private final Multimap mappers = LinkedHashMultimap.create();

  protected BaseMapping() {
    // Nothing here
  }

  /**
   * All the WS supported fields
   */
  public Set supportedFields() {
    return mappers.keySet();
  }

  public org.sonar.server.es.SearchOptions newQueryOptions(SearchOptions options) {
    org.sonar.server.es.SearchOptions result = new org.sonar.server.es.SearchOptions();
    result.setPage(options.page(), options.pageSize());
    List optionFields = options.fields();
    if (optionFields != null) {
      for (String optionField : optionFields) {
        result.addFields(indexFieldsByWsFields.get(optionField));
      }
    }
    return result;
  }

  /**
   * Write only requested document fields
   */
  protected void doWrite(DOC doc, @Nullable CTX context, JsonWriter json, @Nullable QueryContext queryContext) {
    json.beginObject();
    json.prop("key", doc.keyField());
    if (queryContext == null || queryContext.getFieldsToReturn().isEmpty()) {
      // return all fields
      for (Mapper mapper : mappers.values()) {
        mapper.write(json, doc, context);
      }
    } else {
      for (String optionField : queryContext.getFieldsToReturn()) {
        for (Mapper mapper : mappers.get(optionField)) {
          mapper.write(json, doc, context);
        }
      }
    }
    json.endObject();
  }

  protected BaseMapping map(String key, String indexKey) {
    return map(key, new IndexStringMapper(key, indexKey));
  }

  protected BaseMapping mapBoolean(String key, String indexKey) {
    return map(key, new IndexBooleanMapper(key, indexKey));
  }

  protected BaseMapping mapDateTime(String key, String indexKey) {
    return map(key, new IndexDatetimeMapper(key, indexKey));
  }

  protected BaseMapping mapArray(String key, String indexKey) {
    return map(key, new IndexArrayMapper(key, indexKey));
  }

  protected BaseMapping map(String key, Mapper mapper) {
    mappers.put(key, mapper);
    if (mapper instanceof IndexMapper) {
      IndexMapper indexField = (IndexMapper) mapper;
      indexFieldsByWsFields.putAll(key, Arrays.asList(indexField.indexFields()));
    }
    return this;
  }

  public interface Mapper {
    void write(JsonWriter json, DOC doc, CTX context);
  }

  public abstract static class IndexMapper implements Mapper {
    protected final String[] indexFields;

    protected IndexMapper(String... indexFields) {
      this.indexFields = indexFields;
    }

    String[] indexFields() {
      return indexFields;
    }
  }

  /**
   * String field
   */
  public static class IndexStringMapper extends IndexMapper {
    protected final String key;

    public IndexStringMapper(String key, String indexKey) {
      super(indexKey);
      this.key = key;
    }

    @Override
    public void write(JsonWriter json, DOC doc, CTX context) {
      Object val = doc.getNullableField(indexFields[0]);
      if (val == null && indexFields.length == 2) {
        // There is an alternative value
        val = doc.getNullableField(indexFields[1]);
      }
      json.prop(key, val != null ? val.toString() : null);
    }
  }

  public static class IndexBooleanMapper extends IndexMapper {
    private final String key;

    public IndexBooleanMapper(String key, String indexKey) {
      super(indexKey);
      this.key = key;
    }

    @Override
    public void write(JsonWriter json, DOC doc, CTX context) {
      Boolean val = doc.getNullableField(indexFields[0]);
      json.prop(key, val != null ? val.booleanValue() : null);
    }
  }

  public static class IndexArrayMapper extends IndexMapper {
    private final String key;

    public IndexArrayMapper(String key, String indexKey) {
      super(indexKey);
      this.key = key;
    }

    @Override
    public void write(JsonWriter json, DOC doc, CTX context) {
      Iterable values = doc.getNullableField(indexFields[0]);
      if (values != null) {
        json.name(key).beginArray().values(values).endArray();
      }
    }
  }

  public static class IndexDatetimeMapper extends IndexMapper {
    private final String key;

    public IndexDatetimeMapper(String key, String indexKey) {
      super(indexKey);
      this.key = key;
    }

    @Override
    public void write(JsonWriter json, DOC doc, CTX context) {
      String val = doc.getNullableField(indexFields[0]);
      if (val != null) {
        json.propDateTime(key, IndexUtils.parseDateTime(val));
      }
    }
  }

}