28 lines
879 B
Go
28 lines
879 B
Go
package system
|
|
|
|
import (
|
|
"ai4m/global"
|
|
"ai4m/model/system/response"
|
|
)
|
|
|
|
type AiService struct {
|
|
}
|
|
|
|
func (aiService *AiService) ScreenAi() ([]response.ScreenAi, error) {
|
|
var asTasks []response.ScreenAi
|
|
err := global.PCM_MYSQL.Raw("select name,status,strategy,replica,cluster_name from task_ai where deleted_at is NULL and strategy != ''").Scan(&asTasks).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return asTasks, nil
|
|
}
|
|
|
|
func (aiService *AiService) ScreenPieChart() ([]response.PieChart, error) {
|
|
var pieCharts []response.PieChart
|
|
err := global.PCM_MYSQL.Raw("SELECT model_name ,COUNT(*) AS model_count,(COUNT(*) / (SELECT COUNT(*) FROM task_ai WHERE deleted_at IS NULL AND model_name != '')) AS ratio FROM task_ai WHERE deleted_at IS NULL AND model_name != '' GROUP BY model_name;").Scan(&pieCharts).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return pieCharts, nil
|
|
}
|