|
@@ -2,6 +2,7 @@ package com.tzld.piaoquan.recommend.server.service.score;
|
|
|
|
|
|
|
|
|
import com.typesafe.config.Config;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
import org.apache.commons.lang.exception.ExceptionUtils;
|
|
|
import org.slf4j.Logger;
|
|
|
import org.slf4j.LoggerFactory;
|
|
@@ -12,6 +13,7 @@ import java.util.Map;
|
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
|
|
|
|
|
|
|
+@Slf4j
|
|
|
public final class ScorerUtils {
|
|
|
private static Logger LOGGER = LoggerFactory.getLogger(ScorerUtils.class);
|
|
|
|
|
@@ -19,8 +21,22 @@ public final class ScorerUtils {
|
|
|
|
|
|
public static String BASE_CONF = "feeds_score_config_baseline.conf";
|
|
|
|
|
|
+ public static void warmUp() {
|
|
|
+ log.info("scorer warm up ");
|
|
|
+ ScorerUtils.init(BASE_CONF);
|
|
|
+ }
|
|
|
+
|
|
|
private ScorerUtils() {
|
|
|
- scorerPipelineCache.put(BASE_CONF, getScorerPipeline(BASE_CONF));
|
|
|
+ // init(BASE_CONF);
|
|
|
+ }
|
|
|
+
|
|
|
+ public static void init(String configFile) {
|
|
|
+ ScorerConfig scorerConfig = new ScorerConfig();
|
|
|
+ scorerConfig.load(configFile);
|
|
|
+ List<AbstractScorer> scorers = ScorerUtils.constructScorers(scorerConfig);
|
|
|
+
|
|
|
+ initLoadModel(scorers);
|
|
|
+ scorerPipelineCache.put(configFile, new ScorerPipeline(scorers));
|
|
|
}
|
|
|
|
|
|
/**
|
|
@@ -36,6 +52,13 @@ public final class ScorerUtils {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ public static void initLoadModel(String configFile) {
|
|
|
+ ScorerConfig scorerConfig = new ScorerConfig();
|
|
|
+ scorerConfig.load(configFile);
|
|
|
+ List<AbstractScorer> scorers = ScorerUtils.constructScorers(scorerConfig);
|
|
|
+ initLoadModel(scorers);
|
|
|
+ }
|
|
|
+
|
|
|
public static void initLoadModel(Config config) {
|
|
|
ScorerConfig scorerConfig = new ScorerConfig();
|
|
|
scorerConfig.load(config);
|