model service

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/PredictService.java

@@ -7,10 +7,12 @@ import com.tzld.piaoquan.recommend.model.grpc.model.common.Result;
 import com.tzld.piaoquan.recommend.model.service.model.Model;
 import com.tzld.piaoquan.recommend.model.service.model.ModelEnum;
 import com.tzld.piaoquan.recommend.model.service.model.ModelManager;
+import org.springframework.stereotype.Service;
 
 /**
  * @author dyp
  */
+@Service
 public class PredictService {
     public PredictResponse predict(PredictRequest request) {
         String modelName = request.getModelName();

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/web/DemoController.java

@@ -0,0 +1,25 @@
+package com.tzld.piaoquan.recommend.model.web;
+
+import com.tzld.piaoquan.recommend.model.service.PredictService;
+import com.tzld.piaoquan.recommend.model.service.model.Model;
+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.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * @author dyp
+ */
+@RestController
+@Slf4j
+public class DemoController {
+
+    @GetMapping("/demo")
+    public String demo() {
+        Model model = ModelManager.getInstance().getModel(ModelEnum.DEMO);
+        String date = model.predict("");
+        return date;
+    }
+}