package com.example.springboot_hikari.controller;

import com.alibaba.fastjson.JSON;
import com.example.springboot_hikari.domain.DataRecordBean;
import com.example.springboot_hikari.service.DealDataService;
import com.example.springboot_hikari.util.WebResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

/**
 * @author Simon
 */
@Slf4j
@RestController
@RequestMapping("/monitor")
public class MonitorController {

    @Autowired
    private DealDataService dealDataService;

    @RequestMapping("/queryImpalaCount")
    public WebResult queryImpalaCount(@RequestParam(value = "tableName", required = false) String tableName) {
        List<DataRecordBean> countList = dealDataService.queryHiveCount(tableName);
        String jsonListStr = JSON.toJSONString(countList);
        return new WebResult(String.valueOf(HttpStatus.OK.value()), HttpStatus.OK.getReasonPhrase(), jsonListStr);
    }

    @RequestMapping("/queryOracleCount")
    public WebResult queryOracleCount(@RequestParam(value = "tableName", required = false) String tableName) {
        Integer count = dealDataService.queryOracleCount(tableName);
        return new WebResult(String.valueOf(HttpStatus.OK.value()), HttpStatus.OK.getReasonPhrase(), count.toString());
    }

    @RequestMapping("/queryMysqlCount")
    public WebResult queryMySQLCount(@RequestParam(value = "tableName", required = false) String tableName) {
        Integer count = dealDataService.queryMysqlCount(tableName);
        return new WebResult(String.valueOf(HttpStatus.OK.value()), HttpStatus.OK.getReasonPhrase(), count.toString());
    }
}