Java Config.get方法代码示例

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
public static IndexWriter configureWriter(Config config, PerfRunData runData, OpenMode mode, IndexCommit commit) throws IOException {
  IndexWriterConfig iwc = createWriterConfig(config, runData, mode, commit);
  String infoStreamVal = config.get("writer.info.stream", null);
  if (infoStreamVal != null) {
    if (infoStreamVal.equals("SystemOut")) {
      iwc.setInfoStream(System.out);
    } else if (infoStreamVal.equals("SystemErr")) {
      iwc.setInfoStream(System.err);
    } else {
      File f = new File(infoStreamVal).getAbsoluteFile();
      iwc.setInfoStream(new PrintStream(new BufferedOutputStream(new FileOutputStream(f)), false, Charset.defaultCharset().name()));
    }
  }
  IndexWriter writer = new IndexWriter(runData.getDirectory(), iwc);
  return writer;
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
public PerfTask(PerfRunData runData) {
  this();
  this.runData = runData;
  Config config = runData.getConfig();
  this.maxDepthLogStart = config.get("task.max.depth.log",0);

  String logStepAtt = "log.step";
  String taskLogStepAtt = "log.step." + name;
  if (config.get(taskLogStepAtt, null) != null) {
    logStepAtt = taskLogStepAtt;
  }

  // It's important to read this from Config, to support vals-by-round.
  logStep = config.get(logStepAtt, DEFAULT_LOG_STEP);
  // To avoid the check 'if (logStep > 0)' in tearDown(). This effectively
  // turns logging off.
  if (logStep <= 0) {
    logStep = Integer.MAX_VALUE;
  }
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
/**
 * Builds a SpatialStrategy from configuration options.
 */
protected SpatialStrategy makeSpatialStrategy(final Config config) {
  //A Map view of Config that prefixes keys with "spatial."
  Map<String, String> configMap = new AbstractMap<String, String>() {
    @Override
    public Set<Entry<String, String>> entrySet() {
      throw new UnsupportedOperationException();
    }

    @Override
    public String get(Object key) {
      return config.get("spatial." + key, null);
    }
  };

  SpatialContext ctx = SpatialContextFactory.makeSpatialContext(configMap, null);

  //Some day the strategy might be initialized with a factory but such a factory
  // is non-existent.
  return makeSpatialStrategy(config, configMap, ctx);
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setConfig(Config config) {
  super.setConfig(config);
  
  File workDir = new File(config.get("work.dir", "work"));
  String d = config.get("docs.dir", "dir-out");
  dataDir = new File(d);
  if (!dataDir.isAbsolute()) {
    dataDir = new File(workDir, d);
  }

  inputFiles = new Iterator(dataDir);

  if (inputFiles == null) {
    throw new RuntimeException("No txt files in dataDir: " + dataDir.getAbsolutePath());
  }
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
protected SpatialStrategy makeSpatialStrategy(final Config config, Map<String, String> configMap,
                                              SpatialContext ctx) {
  //A factory for the prefix tree grid
  SpatialPrefixTree grid = SpatialPrefixTreeFactory.makeSPT(configMap, null, ctx);

  RecursivePrefixTreeStrategy strategy = new RecursivePrefixTreeStrategy(grid, SPATIAL_FIELD) {
    {
      //protected field
      this.pointsOnly = config.get("spatial.docPointsOnly", false);
    }
  };

  int prefixGridScanLevel = config.get("query.spatial.prefixGridScanLevel", -4);
  if (prefixGridScanLevel < 0)
    prefixGridScanLevel = grid.getMaxLevels() + prefixGridScanLevel;
  strategy.setPrefixGridScanLevel(prefixGridScanLevel);

  double distErrPct = config.get("spatial.distErrPct", .025);//doc & query; a default
  strategy.setDistErrPct(distErrPct);
  return strategy;
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
public WriteLineDocTask(PerfRunData runData) throws Exception {
  super(runData);
  Config config = runData.getConfig();
  fname = config.get("line.file.out", null);
  if (fname == null) {
    throw new IllegalArgumentException("line.file.out must be set");
  }
  OutputStream out = StreamUtils.outputStream(new File(fname));
  lineFileOut = new PrintWriter(new BufferedWriter(new OutputStreamWriter(out, StandardCharsets.UTF_8), StreamUtils.BUFFER_SIZE));
  docMaker = runData.getDocMaker();
  
  // init fields 
  String f2r = config.get("line.fields",null);
  if (f2r == null) {
    fieldsToWrite = DEFAULT_FIELDS;
  } else {
    if (f2r.indexOf(SEP)>=0) {
      throw new IllegalArgumentException("line.fields "+f2r+" should not contain the separator char: "+SEP);
    }
    fieldsToWrite = f2r.split(","); 
  }
  
  // init sufficient fields
  sufficientFields = new boolean[fieldsToWrite.length];
  String suff = config.get("sufficient.fields",DEFAULT_SUFFICIENT_FIELDS);
  if (",".equals(suff)) {
    checkSufficientFields = false;
  } else {
    checkSufficientFields = true;
    HashSet<String> sf = new HashSet<>(Arrays.asList(suff.split(",")));
    for (int i=0; i<fieldsToWrite.length; i++) {
      if (sf.contains(fieldsToWrite[i])) {
        sufficientFields[i] = true;
      }
    }
  }
  
  writeHeader(lineFileOut);
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
public static IndexDeletionPolicy getIndexDeletionPolicy(Config config) {
  String deletionPolicyName = config.get("deletion.policy", "org.apache.lucene.index.KeepOnlyLastCommitDeletionPolicy");
  if (deletionPolicyName.equals(NoDeletionPolicy.class.getName())) {
    return NoDeletionPolicy.INSTANCE;
  } else {
    try {
      return Class.forName(deletionPolicyName).asSubclass(IndexDeletionPolicy.class).newInstance();
    } catch (Exception e) {
      throw new RuntimeException("unable to instantiate class '" + deletionPolicyName + "' as IndexDeletionPolicy", e);
    }
  }
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setup() throws Exception {
  super.setup();
  //check to make sure either the doc is being stored
  PerfRunData runData = getRunData();
  Config config = runData.getConfig();
  clnName = config.get("collector.class", "");
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setConfig(Config config) {
  super.setConfig(config);
  String fileName = config.get("docs.file", null);
  if (fileName == null) {
    throw new IllegalArgumentException("docs.file must be set");
  }
  file = new File(fileName).getAbsoluteFile();
  if (encoding == null) {
    encoding = IOUtils.UTF_8;
  }
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setConfig(Config config) {
  super.setConfig(config);
  File workDir = new File(config.get("work.dir", "work"));
  String d = config.get("docs.dir", "reuters-out");
  dataDir = new File(d);
  if (!dataDir.isAbsolute()) {
    dataDir = new File(workDir, d);
  }
  inputFiles.clear();
  collectFiles(dataDir, inputFiles);
  if (inputFiles.size() == 0) {
    throw new RuntimeException("No txt files in dataDir: "+dataDir.getAbsolutePath());
  }
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setConfig(Config config) {
  super.setConfig(config);
  keepImages = config.get("keep.image.only.docs", true);
  String fileName = config.get("docs.file", null);
  if (fileName != null) {
    file = new File(fileName).getAbsoluteFile();
  }
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
/**
 * Sets the {@link Config} for this content source. If you override this
 * method, you must call super.setConfig.
 */
public void setConfig(Config config) {
  this.config = config;
  forever = config.get("content.source.forever", true);
  logStep = config.get("content.source.log.step", 0);
  verbose = config.get("content.source.verbose", false);
  encoding = config.get("content.source.encoding", null);
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setConfig(Config config) {
  super.setConfig(config);
  random = new Random(config.get("rand.seed", 13));
  maxDocFacets = config.get("max.doc.facets", 10);
  maxDims = config.get("max.doc.facets.dims", 5);
  maxFacetDepth = config.get("max.facet.depth", 3);
  if (maxFacetDepth < 2) {
    throw new IllegalArgumentException("max.facet.depth must be at least 2; got: " + maxFacetDepth);
  }
  maxValue = maxDocFacets * maxFacetDepth;
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
/**
 * Optionally converts points to circles, and optionally bbox'es result.
 */
public static ShapeConverter makeShapeConverter(final SpatialStrategy spatialStrategy,
                                                Config config, String configKeyPrefix) {
  //by default does no conversion
  final double radiusDegrees = config.get(configKeyPrefix+"radiusDegrees", 0.0);
  final double plusMinus = config.get(configKeyPrefix+"radiusDegreesRandPlusMinus", 0.0);
  final boolean bbox = config.get(configKeyPrefix + "bbox", false);

  return new ShapeConverter() {
    @Override
    public Shape convert(Shape shape) {
      if (shape instanceof Point && (radiusDegrees != 0.0 || plusMinus != 0.0)) {
        Point point = (Point)shape;
        double radius = radiusDegrees;
        if (plusMinus > 0.0) {
          Random random = new Random(point.hashCode());//use hashCode so it's reproducibly random
          radius += random.nextDouble() * 2 * plusMinus - plusMinus;
          radius = Math.abs(radius);//can happen if configured plusMinus > radiusDegrees
        }
        shape = spatialStrategy.getSpatialContext().makeCircle(point, radius);
      }
      if (bbox)
        shape = shape.getBoundingBox();
      return shape;
    }
  };
}
 

import org.apache.lucene.benchmark.byTask.utils.Config; //导入方法依赖的package包/类
@Override
public void setConfig(Config config) throws Exception {
  strategy = SpatialDocMaker.getSpatialStrategy(config.getRoundNumber());
  shapeConverter = SpatialDocMaker.makeShapeConverter(strategy, config, "query.spatial.");

  distErrPct = config.get("query.spatial.distErrPct", Double.NaN);
  operation = SpatialOperation.get(config.get("query.spatial.predicate", "Intersects"));
  score = config.get("query.spatial.score", false);

  super.setConfig(config);//call last, will call prepareQueries()
}