Browse Source

model service

丁云鹏 6 months ago
parent
commit
2385a4b3c6

+ 3 - 0
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/WarmUpService.java

@@ -3,6 +3,7 @@ package com.tzld.piaoquan.recommend.model.service;
 import com.tzld.piaoquan.recommend.model.service.model.ModelEnum;
 import com.tzld.piaoquan.recommend.model.service.model.ModelManager;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.context.annotation.DependsOn;
 import org.springframework.stereotype.Component;
 
 import javax.annotation.PostConstruct;
@@ -12,10 +13,12 @@ import javax.annotation.PostConstruct;
  */
 @Component
 @Slf4j
+@DependsOn({"propertiesUtil"})
 public class WarmUpService {
     @PostConstruct
     public void warmup() {
         // ModelManager.getInstance().registerModel(ModelEnum.VIDEO_DSSM);
+
         ModelManager.getInstance().registerModel(ModelEnum.DEMO);
     }
 }

+ 1 - 2
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/util/PropertiesUtil.java

@@ -4,13 +4,12 @@ import org.springframework.context.EnvironmentAware;
 import org.springframework.core.env.Environment;
 import org.springframework.stereotype.Component;
 
-@Component
+@Component("propertiesUtil")
 public class PropertiesUtil implements EnvironmentAware {
 
 
     private static Environment environment;
 
-
     @Override
     public void setEnvironment(Environment environment) {
         this.environment = environment;