add query thread polaris tools
parent
9332ebfe05
commit
3187c887b8
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisEpollTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Epoll',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisFileTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - File',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisFutexTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Futex',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisIdleTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Idle',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisNetTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Net',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisOncpuTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - OnCPU',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisOtherTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Other',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
import json
|
||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from configs import dify_config
|
||||||
|
from core.tools.builtin_tool.tool import BuiltinTool
|
||||||
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class ThreadPolarisRunqueueTimeTool(BuiltinTool):
|
||||||
|
def _invoke(
|
||||||
|
self,
|
||||||
|
user_id: str,
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
conversation_id: Optional[str] = None,
|
||||||
|
app_id: Optional[str] = None,
|
||||||
|
message_id: Optional[str] = None,
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
pod = tool_parameters.get('pod', '.*')
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Runqueue',
|
||||||
|
'params': {
|
||||||
|
'pod': pod
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
Loading…
Reference in New Issue