com.arextest.web.api.service.controller.CheckHealthController Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arex-web-api Show documentation

There is a newer version: 0.7.0.1

package com.arextest.web.api.service.controller;

import com.arextest.common.model.response.Response;
import com.arextest.common.utils.ResponseUtils;
import com.arextest.web.core.business.ai.AIProvider;
import com.arextest.web.model.contract.contracts.ai.ModelInfo;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import lombok.Builder;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

@Slf4j
@Controller
@RequestMapping("/vi/")
@CrossOrigin(origins = "*", maxAge = 3600)
public class CheckHealthController {
  @Resource
  Optional aiController;

  @Value("${pom.version}")
  private String VERSION;

  @GetMapping("/health")
  @ResponseBody
  public Response checkHealth() {
    return ResponseUtils.successResponse(VERSION);
  }

  @Data
  @Builder
  public static class FeatureCheckRes {
    private boolean aiEnabled;
    private List modelInfos;
  }

  @GetMapping("/checkFeature")
  @ResponseBody
  public Response checkFeature() {
    return ResponseUtils.successResponse(FeatureCheckRes
        .builder()
        .aiEnabled(aiController.isPresent())
        .modelInfos(aiController
            .map(controller -> controller.getProviders()
                .stream()
                .map(AIProvider::getModelInfo)
                .collect(Collectors.toList()))
            .orElse(Collections.emptyList()))
        .build());
  }
}