Package org.apache.solr.common

Examples of org.apache.solr.common.SolrInputField

818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
                }
            }
            if(fieldBoost != null){ //we need still to do part (2) of setting the correct boost
                for(Entry<String,int[]> entry : fieldsToBoost.entrySet()){
                    if(entry.getValue()[0] > 1) { //adapt the boost only for multi valued fields
                        SolrInputField solrField = inputDocument.getField(entry.getKey());
                        //the correct bosst is baseBoost (representing entity boost with field
                        //boost) multiplied with the sqrt(fieldValues). The 2nd part aims to
                        //compensate the Solr lengthNorm (1/sqrt(fieldTokens))
                        //see STANBOL-1027 for details
                        solrField.setBoost(baseBoost*(float)Math.sqrt(entry.getValue()[0]));
                    }
                }
            }
        }
        return inputDocument;
View Full Code Here
141
142
143
144
145
146
147
148
149
150
151
152
153
    for (SolrInputDocument document : docs) {
      updateUnmarshalled.add(document);
    }

    SolrInputDocument outDoc = updateUnmarshalled.getDocuments().get(0);
    SolrInputField iter = outDoc.getField("iter");
    Assert.assertNotNull("iter field is null", iter);
    Object iterVal = iter.getValue();
    Assert.assertTrue("iterVal is not a Collection",
                      iterVal instanceof Collection);
    Assert.assertEquals("iterVal contents", values, iterVal);

  }
View Full Code Here
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
                           SolrInputDocument actualDoc) {
    Assert.assertEquals(expectedDoc.getDocumentBoost(),
                        actualDoc.getDocumentBoost());

    for (String s : expectedDoc.getFieldNames()) {
      SolrInputField expectedField = expectedDoc.getField(s);
      SolrInputField actualField = actualDoc.getField(s);
      Assert.assertEquals(m + ": diff boosts for field: " + s,
                          expectedField.getBoost(), actualField.getBoost());
      Object expectedVal = expectedField.getValue();
      Object actualVal = actualField.getValue();
      if (expectedVal instanceof Set &&
          actualVal instanceof Collection) {
        // unmarshaled documents never contain Sets, they are just a
        // List in an arbitrary order based on what the iterator of
        // hte original Set returned, so we need a comparison that is
View Full Code Here
55
56
57
58
59
60
61
62
63
64
65
66
    public void testAdd_IdAlreadyPresent() {
        IdAddingSolrUpdateWriter updateWriter = new IdAddingSolrUpdateWriter(
                UNIQUE_KEY_FIELD, DOCUMENT_ID, null, TABLE_NAME, updateCollector);
       
       
        SolrInputField solrIdField = new SolrInputField(DOCUMENT_ID);
        solrIdField.setValue(DOCUMENT_ID, 1.0f);
       
        when(solrDoc.getField(UNIQUE_KEY_FIELD)).thenReturn(solrIdField);

        updateWriter.add(solrDoc);
View Full Code Here
88
89
90
91
92
93
94
95
96
97
98
99
100
101
                UNIQUE_KEY_FIELD, DOCUMENT_ID, null, TABLE_NAME, updateCollector);
       
        SolrInputDocument docA = mock(SolrInputDocument.class);
        SolrInputDocument docB = mock(SolrInputDocument.class);
       
        SolrInputField keyFieldA = new SolrInputField(UNIQUE_KEY_FIELD);
        keyFieldA.setValue(idA, 1.0f);
        SolrInputField keyFieldB = new SolrInputField(UNIQUE_KEY_FIELD);
        keyFieldB.setValue(idB, 1.0f);
       
       
        when(docA.getField(UNIQUE_KEY_FIELD)).thenReturn(keyFieldA);
        when(docB.getField(UNIQUE_KEY_FIELD)).thenReturn(keyFieldB);
View Full Code Here
60
61
62
63
64
65
66
67
68
     * @param inputDocument document to be added
     * @param prefix prefix to be added to field names
     */
    public void add(SolrInputDocument inputDocument, String prefix) {
        for (Entry<String, SolrInputField> entry : inputDocument.entrySet()) {
            SolrInputField inputField = entry.getValue();
            document.addField(prefix + entry.getKey(), inputField.getValues(), inputField.getBoost());
        }
    }
View Full Code Here
89
90
91
92
93
94
95
96
97
98
99
       
        SolrInputDocument solrDocument = solrInputDocCaptor.getValue();

        assertEquals(Sets.newHashSet("fieldA", "fieldB"), solrDocument.keySet());

        SolrInputField fieldA = solrDocument.get("fieldA");
        SolrInputField fieldB = solrDocument.get("fieldB");

        assertEquals(Lists.newArrayList(42), fieldA.getValues());
        assertEquals(Lists.newArrayList("A", "B", "C"), fieldB.getValues());
    }
View Full Code Here
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
     * Adding multiple documents without ids will result in an IllegalStateException being thrown.
     */
    @Override
    public void add(SolrInputDocument solrDocument) {
        String docId = documentId;
        SolrInputField uniqueKeySolrField = solrDocument.getField(uniqueKeyField);
        if (uniqueKeySolrField == null) {
            if (idUsed) {
                throw new IllegalStateException("Document id '" + documentId + "' has already been used by this record");
            }
            solrDocument.addField(uniqueKeyField, documentId);
            idUsed = true;
        } else {
            docId = uniqueKeySolrField.getValue().toString();
        }
       
        if (tableNameField != null) {
            solrDocument.addField(tableNameField, tableName);
        }
View Full Code Here
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
          Class<?> rawMapType = mapTypeInformation.getType();

          Map<?, ?> map = (Map<?, ?>) fieldValue;
          for (Map.Entry<?, ?> entry : map.entrySet()) {
            String mappedFieldName = entry.getKey().toString();
            SolrInputField field = new SolrInputField(mappedFieldName);
            if (entry.getValue() instanceof Iterable) {
              for (Object o : (Iterable<?>) entry.getValue()) {
                field.addValue(convertToSolrType(rawMapType, o), 1f);
              }
            } else {
              if (rawMapType.isArray()) {
                for (Object o : (Object[]) entry.getValue()) {
                  field.addValue(convertToSolrType(rawMapType, o), 1f);
                }
              } else {
                field.addValue(convertToSolrType(rawMapType, entry.getValue()), 1f);
              }
            }
            target.put(mappedFieldName, field);
          }
          return;
        }

        SolrInputField field = new SolrInputField(persistentProperty.getFieldName());
        if (persistentProperty.isCollectionLike()) {
          Collection<?> collection = asCollection(fieldValue);
          for (Object o : collection) {
            if (o != null) {
              field.addValue(convertToSolrType(persistentProperty.getType(), o), 1f);
            }
          }
        } else {
          field.setValue(convertToSolrType(persistentProperty.getType(), fieldValue), 1f);
        }
        target.put(persistentProperty.getFieldName(), field);

        if (persistentProperty.isBoosted()) {
          field.setBoost(persistentProperty.getBoost());
        }

      }
    });
View Full Code Here
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
    this.fieldName = fieldName;
  }

  @Override
  public int compare(SolrInputDocument doc1, SolrInputDocument doc2) {
    SolrInputField f1 = doc1.getField(fieldName);
    SolrInputField f2 = doc2.getField(fieldName);
    if (f1 == f2) {
      return 0;
    } else if (f1 == null) {
      return -1;
    } else if (f2 == null) {
      return 1;
    }
   
    Object v1 = f1.getFirstValue();
    Object v2 = f2.getFirstValue();         
    return child.compare(v1, v2);
  }
View Full Code Here
TOP

Related Classes of org.apache.solr.common.SolrInputField

  • com.ngdata.hbaseindexer.indexer.IdAddingSolrUpdateWriter
  • com.ngdata.hbaseindexer.indexer.IdAddingSolrUpdateWriterTest
  • com.ngdata.hbaseindexer.parse.DefaultResultToSolrMapperTest
  • com.ngdata.hbaseindexer.parse.SolrInputDocumentBuilder
  • com.tamingtext.classifier.bayes.BayesUpdateRequestProcessor
  • gc.solr.publish.SolrDocSaxHandler
  • org.apache.blur.slur.RowMutationHelper
  • org.apache.jetspeed.search.solr.SolrSearchEngineImpl
  • org.apache.solr.client.solrj.beans.TestDocumentObjectBinder
  • org.apache.solr.client.solrj.request.JavaBinUpdateRequestCodec
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.