Java DocTransformer类代码示例

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
public final void writeDocuments(String name, ResultContext res, ReturnFields fields ) throws IOException {
  DocList ids = res.docs;
  TransformContext context = new TransformContext();
  context.query = res.query;
  context.wantsScores = fields.wantsScore() && ids.hasScores();
  context.req = req;
  writeStartDocumentList(name, ids.offset(), ids.size(), ids.matches(), 
      context.wantsScores ? new Float(ids.maxScore()) : null );
  
  DocTransformer transformer = fields.getTransformer();
  context.searcher = req.getSearcher();
  context.iterator = ids.iterator();
  if( transformer != null ) {
    transformer.setContext( context );
  }
  int sz = ids.size();
  Set<String> fnames = fields.getLuceneFieldNames();
  for (int i=0; i<sz; i++) {
    int id = context.iterator.nextDoc();
    Document doc = context.searcher.doc(id, fnames);
    SolrDocument sdoc = toSolrDocument( doc );
    if( transformer != null ) {
      transformer.transform( sdoc, id);
    }
    writeSolrDocument( null, sdoc, returnFields, i );
  }
  if( transformer != null ) {
    transformer.setContext( null );
  }
  writeEndDocumentList();
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
protected void writeResultsBody( ResultContext res, JavaBinCodec codec ) throws IOException 
{
  DocList ids = res.docs;
  int sz = ids.size();
  codec.writeTag(JavaBinCodec.ARR, sz);
  if(searcher == null) searcher = solrQueryRequest.getSearcher();
  if(schema == null) schema = solrQueryRequest.getSchema();

  DocTransformer transformer = returnFields.getTransformer();
  TransformContext context = new TransformContext();
  context.query = res.query;
  context.wantsScores = returnFields.wantsScore() && ids.hasScores();
  context.req = solrQueryRequest;
  context.searcher = searcher;
  if( transformer != null ) {
    transformer.setContext( context );
  }
  
  Set<String> fnames = returnFields.getLuceneFieldNames();
  context.iterator = ids.iterator();
  for (int i = 0; i < sz; i++) {
    int id = context.iterator.nextDoc();
    Document doc = searcher.doc(id, fnames);
    SolrDocument sdoc = getDoc(doc);
    if( transformer != null ) {
      transformer.transform(sdoc, id);
    }
    codec.writeSolrDocument(sdoc);
  }
  if( transformer != null ) {
    transformer.setContext( null );
  }
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
@Override
public void prepare(ResponseBuilder rb) throws IOException {
  QueryDocAuthorizationComponent docComponent =
      (QueryDocAuthorizationComponent)rb.req.getCore().getSearchComponent("queryDocAuthorization");
  if (docComponent != null) {
    String userName = sentryInstance.getUserName(rb.req);
    String superUser = (System.getProperty("solr.authorization.superuser", "solr"));
    // security is never applied to the super user; for example, if solr internally is using
    // real time get for replica synchronization, we need to return all the documents.
    if (docComponent.getEnabled() && !superUser.equals(userName)) {
      Set<String> roles = sentryInstance.getRoles(userName);
      if (roles != null && roles.size() > 0) {
        SolrReturnFields savedReturnFields = (SolrReturnFields)rb.rsp.getReturnFields();
        if (savedReturnFields == null) {
          throw new SolrException(SolrException.ErrorCode.SERVER_ERROR,
              "Not able to authorize request because ReturnFields is invalid: " + savedReturnFields);
        }
        DocTransformer savedTransformer = savedReturnFields.getTransformer();
        Query filterQuery = docComponent.getFilterQuery(roles);
        if (filterQuery != null) {
          SolrReturnFields solrReturnFields = new AddDocIdReturnFields(rb.req, savedTransformer, filterQuery);
          rb.rsp.setReturnFields(solrReturnFields);
        } else {
          throw new SolrException(SolrException.ErrorCode.UNAUTHORIZED,
            "Request from user: " + userName +
            "rejected because filter query was unable to be generated");
        }
      } else {
        throw new SolrException(SolrException.ErrorCode.UNAUTHORIZED,
          "Request from user: " + userName +
          " rejected because user is not associated with any roles");
      }
    }
  } else {
    throw new SolrException(SolrException.ErrorCode.UNAUTHORIZED,
      "RealTimeGetRequest request " +
      " rejected because \"queryDocAuthorization\" component not defined");
  }
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
public AddDocIdReturnFields(SolrQueryRequest req, DocTransformer docTransformer,
    Query filterQuery) {
  super(req);
  this.originalTransformer = docTransformer;
  this.filterQuery = filterQuery;
  final DocTransformers docTransformers = new DocTransformers();
  if (originalTransformer != null) docTransformers.addTransformer(originalTransformer);
  docTransformers.addTransformer(new DocIdAugmenter(ID_FIELD_NAME));
  this.transformer = docTransformers;
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
protected void writeResultsBody( ResultContext res, JavaBinCodec codec ) throws IOException 
{
  DocList ids = res.docs;
  int sz = ids.size();
  codec.writeTag(JavaBinCodec.ARR, sz);
  if(searcher == null) searcher = solrQueryRequest.getSearcher();
  if(schema == null) schema = solrQueryRequest.getSchema();

  DocTransformer transformer = returnFields.getTransformer();
  TransformContext context = new TransformContext();
  context.query = res.query;
  context.wantsScores = returnFields.wantsScore() && ids.hasScores();
  context.req = solrQueryRequest;
  context.searcher = searcher;
  if( transformer != null ) {
    transformer.setContext( context );
  }
  
  Set<String> fnames = returnFields.getLuceneFieldNames();
  boolean onlyPseudoFields = (fnames == null && !returnFields.wantsAllFields() && !returnFields.hasPatternMatching())
      || (fnames != null && fnames.size() == 1 && SolrReturnFields.SCORE.equals(fnames.iterator().next()));
  context.iterator = ids.iterator();
  for (int i = 0; i < sz; i++) {
    int id = context.iterator.nextDoc();
    SolrDocument sdoc;
    if (onlyPseudoFields) {
      // no need to get stored fields of the document, see SOLR-5968
      sdoc = new SolrDocument();
    } else {
      Document doc = searcher.doc(id, fnames);
      sdoc = getDoc(doc);
    }
    if( transformer != null ) {
      transformer.transform(sdoc, id);
    }
    codec.writeSolrDocument(sdoc);
  }
  if( transformer != null ) {
    transformer.setContext( null );
  }
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
/** Returns the DocTransformer used to modify documents, or <code>null</code> */
public abstract DocTransformer getTransformer();
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
@Override
public DocTransformer getTransformer()
{
  return transformer;
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
@Override
public DocTransformer create(String field, SolrParams params, SolrQueryRequest req)
{
    return new CachedDocTransformer();
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
@Override
public DocTransformer create(String field, SolrParams params, SolrQueryRequest req)
{
    return new DocValueDocTransformer();
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
@Override
public DocTransformer getTransformer() {
  return transformer;
}
 

import org.apache.solr.response.transform.DocTransformer; //导入依赖的package包/类
public DocTransformer getOriginalTransformer() {
  return originalTransformer;
}