Compare commits
11 Commits
bd8729d473
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| e66ee67cf7 | |||
| 40eb57ee47 | |||
| 6a8e8f1f7d | |||
| 5c83c19bce | |||
| 86c9073da8 | |||
| 43c1839345 | |||
| f62cc1c4a9 | |||
| f6d2069e1a | |||
| f33e14f60f | |||
| d6f275b2d1 | |||
| d8de5a68eb |
@@ -48,8 +48,10 @@ chirp_stack:
|
||||
api_host: "http://localhost:8080" # ChirpStack API 主机地址
|
||||
api_token: "your_chirpstack_api_token" # ChirpStack API Token
|
||||
fport: 10 # ChirpStack FPort
|
||||
api_timeout: 5 # API 请求超时时间 (秒)
|
||||
collection_request_timeout: 10 # 采集请求超时时间 (秒)
|
||||
api_timeout: 10 # ChirpStack API请求超时时间(秒)
|
||||
# 等待设备上行响应的超时时间(秒)。
|
||||
# 对于LoRaWAN这种延迟较高的网络,建议设置为5分钟 (300秒) 或更长。
|
||||
collection_request_timeout: 300
|
||||
|
||||
# 任务调度配置
|
||||
task:
|
||||
@@ -62,12 +64,28 @@ lora:
|
||||
|
||||
# Lora Mesh 配置
|
||||
lora_mesh:
|
||||
uart_port: "/dev/ttyUSB0" # UART 串口路径
|
||||
baud_rate: 115200 # 波特率
|
||||
timeout: 5 # 超时时间 (秒)
|
||||
lora_mesh_mode: "transparent" # Lora Mesh 模式: transparent, command
|
||||
max_chunk_size: 200 # 最大数据块大小
|
||||
reassembly_timeout: 10 # 重组超时时间 (秒)
|
||||
# 主节点串口
|
||||
uart_port: "COM7"
|
||||
# LoRa模块的通信波特率
|
||||
baud_rate: 9600
|
||||
# 等待LoRa模块AT指令响应的超时时间(ms)
|
||||
timeout: 50
|
||||
# LoRa Mesh 模块发送模式(EC: 透传; ED: 完整数据包)
|
||||
# e.g.
|
||||
# EC: 接收端只会接收到消息, 不会接收到请求头
|
||||
# e.g. 发送: EC 05 02 01 48 65 6c 6c 6f
|
||||
# (EC + 05(消息长度) + 0201(地址) + "Hello"(消息本体))
|
||||
# 接收: 48 65 6c 6c 6f ("Hello")
|
||||
# ED: 接收端会接收完整数据包,包含自定义协议头和地址信息。
|
||||
# e.g. 发送: ED 05 12 34 01 00 01 02 03
|
||||
# (ED(帧头) + 05(Length, 即 1(总包数)+1(当前包序号)+3(数据块)) + 12 34(目标地址) + 01(总包数) + 00(当前包序号) + 01 02 03(数据块))
|
||||
# 接收: ED 05 12 34 01 00 01 02 03 56 78(56 78 是发送方地址,会自动拼接到消息末尾)
|
||||
lora_mesh_mode: "ED"
|
||||
# 单包最大用户数据数据长度, 模块限制240, 去掉两位自定义包头, 还剩238
|
||||
max_chunk_size: 238
|
||||
#分片重组超时时间(秒)。如果在一个分片到达后,超过这个时间
|
||||
# 还没收到完整的包,则认为接收失败。
|
||||
reassembly_timeout: 30
|
||||
|
||||
# 通知服务配置
|
||||
notify:
|
||||
@@ -91,3 +109,7 @@ notify:
|
||||
enabled: false # 是否启用飞书通知
|
||||
appID: "cli_xxxxxxxxxx" # 应用 ID
|
||||
appSecret: "your_lark_app_secret" # 应用密钥
|
||||
|
||||
# 定时采集配置
|
||||
collection:
|
||||
interval: 300 # 采集间隔 (秒)
|
||||
|
||||
@@ -86,4 +86,8 @@ lora_mesh:
|
||||
max_chunk_size: 238
|
||||
#分片重组超时时间(秒)。如果在一个分片到达后,超过这个时间
|
||||
# 还没收到完整的包,则认为接收失败。
|
||||
reassembly_timeout: 30
|
||||
reassembly_timeout: 30
|
||||
|
||||
# 定时采集配置
|
||||
collection:
|
||||
interval: 300 # 采集间隔 (秒)
|
||||
259
docs/docs.go
259
docs/docs.go
@@ -975,6 +975,149 @@ const docTemplate = `{
|
||||
}
|
||||
}
|
||||
},
|
||||
"/api/v1/monitor/notifications": {
|
||||
"get": {
|
||||
"security": [
|
||||
{
|
||||
"BearerAuth": []
|
||||
}
|
||||
],
|
||||
"description": "根据提供的过滤条件,分页获取通知列表",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"数据监控"
|
||||
],
|
||||
"summary": "批量查询通知",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "string",
|
||||
"name": "end_time",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"enum": [
|
||||
7,
|
||||
-1,
|
||||
0,
|
||||
1,
|
||||
2,
|
||||
3,
|
||||
4,
|
||||
5,
|
||||
-1,
|
||||
5,
|
||||
6
|
||||
],
|
||||
"type": "integer",
|
||||
"format": "int32",
|
||||
"x-enum-varnames": [
|
||||
"_numLevels",
|
||||
"DebugLevel",
|
||||
"InfoLevel",
|
||||
"WarnLevel",
|
||||
"ErrorLevel",
|
||||
"DPanicLevel",
|
||||
"PanicLevel",
|
||||
"FatalLevel",
|
||||
"_minLevel",
|
||||
"_maxLevel",
|
||||
"InvalidLevel"
|
||||
],
|
||||
"name": "level",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"enum": [
|
||||
"邮件",
|
||||
"企业微信",
|
||||
"飞书",
|
||||
"日志"
|
||||
],
|
||||
"type": "string",
|
||||
"x-enum-varnames": [
|
||||
"NotifierTypeSMTP",
|
||||
"NotifierTypeWeChat",
|
||||
"NotifierTypeLark",
|
||||
"NotifierTypeLog"
|
||||
],
|
||||
"name": "notifier_type",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "string",
|
||||
"name": "order_by",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "integer",
|
||||
"name": "page",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "integer",
|
||||
"name": "pageSize",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "string",
|
||||
"name": "start_time",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"enum": [
|
||||
"发送成功",
|
||||
"发送失败",
|
||||
"已跳过"
|
||||
],
|
||||
"type": "string",
|
||||
"x-enum-comments": {
|
||||
"NotificationStatusFailed": "通知发送失败",
|
||||
"NotificationStatusSkipped": "通知因某些原因被跳过(例如:用户未配置联系方式)",
|
||||
"NotificationStatusSuccess": "通知已成功发送"
|
||||
},
|
||||
"x-enum-descriptions": [
|
||||
"通知已成功发送",
|
||||
"通知发送失败",
|
||||
"通知因某些原因被跳过(例如:用户未配置联系方式)"
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"NotificationStatusSuccess",
|
||||
"NotificationStatusFailed",
|
||||
"NotificationStatusSkipped"
|
||||
],
|
||||
"name": "status",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "integer",
|
||||
"name": "user_id",
|
||||
"in": "query"
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "OK",
|
||||
"schema": {
|
||||
"allOf": [
|
||||
{
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
},
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"data": {
|
||||
"$ref": "#/definitions/dto.ListNotificationResponse"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/api/v1/monitor/pending-collections": {
|
||||
"get": {
|
||||
"security": [
|
||||
@@ -4489,6 +4632,20 @@ const docTemplate = `{
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.ListNotificationResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"list": {
|
||||
"type": "array",
|
||||
"items": {
|
||||
"$ref": "#/definitions/dto.NotificationDTO"
|
||||
}
|
||||
},
|
||||
"pagination": {
|
||||
"$ref": "#/definitions/dto.PaginationDTO"
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.ListPendingCollectionResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
@@ -4797,6 +4954,47 @@ const docTemplate = `{
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.NotificationDTO": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"alarm_timestamp": {
|
||||
"type": "string"
|
||||
},
|
||||
"created_at": {
|
||||
"type": "string"
|
||||
},
|
||||
"error_message": {
|
||||
"type": "string"
|
||||
},
|
||||
"id": {
|
||||
"type": "integer"
|
||||
},
|
||||
"level": {
|
||||
"$ref": "#/definitions/zapcore.Level"
|
||||
},
|
||||
"message": {
|
||||
"type": "string"
|
||||
},
|
||||
"notifier_type": {
|
||||
"$ref": "#/definitions/notify.NotifierType"
|
||||
},
|
||||
"status": {
|
||||
"$ref": "#/definitions/models.NotificationStatus"
|
||||
},
|
||||
"title": {
|
||||
"type": "string"
|
||||
},
|
||||
"to_address": {
|
||||
"type": "string"
|
||||
},
|
||||
"updated_at": {
|
||||
"type": "string"
|
||||
},
|
||||
"user_id": {
|
||||
"type": "integer"
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.PaginationDTO": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
@@ -6275,6 +6473,29 @@ const docTemplate = `{
|
||||
"ReasonTypeHealthCare"
|
||||
]
|
||||
},
|
||||
"models.NotificationStatus": {
|
||||
"type": "string",
|
||||
"enum": [
|
||||
"发送成功",
|
||||
"发送失败",
|
||||
"已跳过"
|
||||
],
|
||||
"x-enum-comments": {
|
||||
"NotificationStatusFailed": "通知发送失败",
|
||||
"NotificationStatusSkipped": "通知因某些原因被跳过(例如:用户未配置联系方式)",
|
||||
"NotificationStatusSuccess": "通知已成功发送"
|
||||
},
|
||||
"x-enum-descriptions": [
|
||||
"通知已成功发送",
|
||||
"通知发送失败",
|
||||
"通知因某些原因被跳过(例如:用户未配置联系方式)"
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"NotificationStatusSuccess",
|
||||
"NotificationStatusFailed",
|
||||
"NotificationStatusSkipped"
|
||||
]
|
||||
},
|
||||
"models.PenStatus": {
|
||||
"type": "string",
|
||||
"enum": [
|
||||
@@ -6608,10 +6829,10 @@ const docTemplate = `{
|
||||
"notify.NotifierType": {
|
||||
"type": "string",
|
||||
"enum": [
|
||||
"smtp",
|
||||
"wechat",
|
||||
"lark",
|
||||
"log"
|
||||
"邮件",
|
||||
"企业微信",
|
||||
"飞书",
|
||||
"日志"
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"NotifierTypeSMTP",
|
||||
@@ -6619,6 +6840,36 @@ const docTemplate = `{
|
||||
"NotifierTypeLark",
|
||||
"NotifierTypeLog"
|
||||
]
|
||||
},
|
||||
"zapcore.Level": {
|
||||
"type": "integer",
|
||||
"format": "int32",
|
||||
"enum": [
|
||||
7,
|
||||
-1,
|
||||
0,
|
||||
1,
|
||||
2,
|
||||
3,
|
||||
4,
|
||||
5,
|
||||
-1,
|
||||
5,
|
||||
6
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"_numLevels",
|
||||
"DebugLevel",
|
||||
"InfoLevel",
|
||||
"WarnLevel",
|
||||
"ErrorLevel",
|
||||
"DPanicLevel",
|
||||
"PanicLevel",
|
||||
"FatalLevel",
|
||||
"_minLevel",
|
||||
"_maxLevel",
|
||||
"InvalidLevel"
|
||||
]
|
||||
}
|
||||
},
|
||||
"securityDefinitions": {
|
||||
|
||||
@@ -967,6 +967,149 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
"/api/v1/monitor/notifications": {
|
||||
"get": {
|
||||
"security": [
|
||||
{
|
||||
"BearerAuth": []
|
||||
}
|
||||
],
|
||||
"description": "根据提供的过滤条件,分页获取通知列表",
|
||||
"produces": [
|
||||
"application/json"
|
||||
],
|
||||
"tags": [
|
||||
"数据监控"
|
||||
],
|
||||
"summary": "批量查询通知",
|
||||
"parameters": [
|
||||
{
|
||||
"type": "string",
|
||||
"name": "end_time",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"enum": [
|
||||
7,
|
||||
-1,
|
||||
0,
|
||||
1,
|
||||
2,
|
||||
3,
|
||||
4,
|
||||
5,
|
||||
-1,
|
||||
5,
|
||||
6
|
||||
],
|
||||
"type": "integer",
|
||||
"format": "int32",
|
||||
"x-enum-varnames": [
|
||||
"_numLevels",
|
||||
"DebugLevel",
|
||||
"InfoLevel",
|
||||
"WarnLevel",
|
||||
"ErrorLevel",
|
||||
"DPanicLevel",
|
||||
"PanicLevel",
|
||||
"FatalLevel",
|
||||
"_minLevel",
|
||||
"_maxLevel",
|
||||
"InvalidLevel"
|
||||
],
|
||||
"name": "level",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"enum": [
|
||||
"邮件",
|
||||
"企业微信",
|
||||
"飞书",
|
||||
"日志"
|
||||
],
|
||||
"type": "string",
|
||||
"x-enum-varnames": [
|
||||
"NotifierTypeSMTP",
|
||||
"NotifierTypeWeChat",
|
||||
"NotifierTypeLark",
|
||||
"NotifierTypeLog"
|
||||
],
|
||||
"name": "notifier_type",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "string",
|
||||
"name": "order_by",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "integer",
|
||||
"name": "page",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "integer",
|
||||
"name": "pageSize",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "string",
|
||||
"name": "start_time",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"enum": [
|
||||
"发送成功",
|
||||
"发送失败",
|
||||
"已跳过"
|
||||
],
|
||||
"type": "string",
|
||||
"x-enum-comments": {
|
||||
"NotificationStatusFailed": "通知发送失败",
|
||||
"NotificationStatusSkipped": "通知因某些原因被跳过(例如:用户未配置联系方式)",
|
||||
"NotificationStatusSuccess": "通知已成功发送"
|
||||
},
|
||||
"x-enum-descriptions": [
|
||||
"通知已成功发送",
|
||||
"通知发送失败",
|
||||
"通知因某些原因被跳过(例如:用户未配置联系方式)"
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"NotificationStatusSuccess",
|
||||
"NotificationStatusFailed",
|
||||
"NotificationStatusSkipped"
|
||||
],
|
||||
"name": "status",
|
||||
"in": "query"
|
||||
},
|
||||
{
|
||||
"type": "integer",
|
||||
"name": "user_id",
|
||||
"in": "query"
|
||||
}
|
||||
],
|
||||
"responses": {
|
||||
"200": {
|
||||
"description": "OK",
|
||||
"schema": {
|
||||
"allOf": [
|
||||
{
|
||||
"$ref": "#/definitions/controller.Response"
|
||||
},
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"data": {
|
||||
"$ref": "#/definitions/dto.ListNotificationResponse"
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"/api/v1/monitor/pending-collections": {
|
||||
"get": {
|
||||
"security": [
|
||||
@@ -4481,6 +4624,20 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.ListNotificationResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"list": {
|
||||
"type": "array",
|
||||
"items": {
|
||||
"$ref": "#/definitions/dto.NotificationDTO"
|
||||
}
|
||||
},
|
||||
"pagination": {
|
||||
"$ref": "#/definitions/dto.PaginationDTO"
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.ListPendingCollectionResponse": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
@@ -4789,6 +4946,47 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.NotificationDTO": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"alarm_timestamp": {
|
||||
"type": "string"
|
||||
},
|
||||
"created_at": {
|
||||
"type": "string"
|
||||
},
|
||||
"error_message": {
|
||||
"type": "string"
|
||||
},
|
||||
"id": {
|
||||
"type": "integer"
|
||||
},
|
||||
"level": {
|
||||
"$ref": "#/definitions/zapcore.Level"
|
||||
},
|
||||
"message": {
|
||||
"type": "string"
|
||||
},
|
||||
"notifier_type": {
|
||||
"$ref": "#/definitions/notify.NotifierType"
|
||||
},
|
||||
"status": {
|
||||
"$ref": "#/definitions/models.NotificationStatus"
|
||||
},
|
||||
"title": {
|
||||
"type": "string"
|
||||
},
|
||||
"to_address": {
|
||||
"type": "string"
|
||||
},
|
||||
"updated_at": {
|
||||
"type": "string"
|
||||
},
|
||||
"user_id": {
|
||||
"type": "integer"
|
||||
}
|
||||
}
|
||||
},
|
||||
"dto.PaginationDTO": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
@@ -6267,6 +6465,29 @@
|
||||
"ReasonTypeHealthCare"
|
||||
]
|
||||
},
|
||||
"models.NotificationStatus": {
|
||||
"type": "string",
|
||||
"enum": [
|
||||
"发送成功",
|
||||
"发送失败",
|
||||
"已跳过"
|
||||
],
|
||||
"x-enum-comments": {
|
||||
"NotificationStatusFailed": "通知发送失败",
|
||||
"NotificationStatusSkipped": "通知因某些原因被跳过(例如:用户未配置联系方式)",
|
||||
"NotificationStatusSuccess": "通知已成功发送"
|
||||
},
|
||||
"x-enum-descriptions": [
|
||||
"通知已成功发送",
|
||||
"通知发送失败",
|
||||
"通知因某些原因被跳过(例如:用户未配置联系方式)"
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"NotificationStatusSuccess",
|
||||
"NotificationStatusFailed",
|
||||
"NotificationStatusSkipped"
|
||||
]
|
||||
},
|
||||
"models.PenStatus": {
|
||||
"type": "string",
|
||||
"enum": [
|
||||
@@ -6600,10 +6821,10 @@
|
||||
"notify.NotifierType": {
|
||||
"type": "string",
|
||||
"enum": [
|
||||
"smtp",
|
||||
"wechat",
|
||||
"lark",
|
||||
"log"
|
||||
"邮件",
|
||||
"企业微信",
|
||||
"飞书",
|
||||
"日志"
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"NotifierTypeSMTP",
|
||||
@@ -6611,6 +6832,36 @@
|
||||
"NotifierTypeLark",
|
||||
"NotifierTypeLog"
|
||||
]
|
||||
},
|
||||
"zapcore.Level": {
|
||||
"type": "integer",
|
||||
"format": "int32",
|
||||
"enum": [
|
||||
7,
|
||||
-1,
|
||||
0,
|
||||
1,
|
||||
2,
|
||||
3,
|
||||
4,
|
||||
5,
|
||||
-1,
|
||||
5,
|
||||
6
|
||||
],
|
||||
"x-enum-varnames": [
|
||||
"_numLevels",
|
||||
"DebugLevel",
|
||||
"InfoLevel",
|
||||
"WarnLevel",
|
||||
"ErrorLevel",
|
||||
"DPanicLevel",
|
||||
"PanicLevel",
|
||||
"FatalLevel",
|
||||
"_minLevel",
|
||||
"_maxLevel",
|
||||
"InvalidLevel"
|
||||
]
|
||||
}
|
||||
},
|
||||
"securityDefinitions": {
|
||||
|
||||
@@ -349,6 +349,15 @@ definitions:
|
||||
pagination:
|
||||
$ref: '#/definitions/dto.PaginationDTO'
|
||||
type: object
|
||||
dto.ListNotificationResponse:
|
||||
properties:
|
||||
list:
|
||||
items:
|
||||
$ref: '#/definitions/dto.NotificationDTO'
|
||||
type: array
|
||||
pagination:
|
||||
$ref: '#/definitions/dto.PaginationDTO'
|
||||
type: object
|
||||
dto.ListPendingCollectionResponse:
|
||||
properties:
|
||||
list:
|
||||
@@ -552,6 +561,33 @@ definitions:
|
||||
- quantity
|
||||
- toPenID
|
||||
type: object
|
||||
dto.NotificationDTO:
|
||||
properties:
|
||||
alarm_timestamp:
|
||||
type: string
|
||||
created_at:
|
||||
type: string
|
||||
error_message:
|
||||
type: string
|
||||
id:
|
||||
type: integer
|
||||
level:
|
||||
$ref: '#/definitions/zapcore.Level'
|
||||
message:
|
||||
type: string
|
||||
notifier_type:
|
||||
$ref: '#/definitions/notify.NotifierType'
|
||||
status:
|
||||
$ref: '#/definitions/models.NotificationStatus'
|
||||
title:
|
||||
type: string
|
||||
to_address:
|
||||
type: string
|
||||
updated_at:
|
||||
type: string
|
||||
user_id:
|
||||
type: integer
|
||||
type: object
|
||||
dto.PaginationDTO:
|
||||
properties:
|
||||
page:
|
||||
@@ -1557,6 +1593,24 @@ definitions:
|
||||
- ReasonTypePreventive
|
||||
- ReasonTypeTreatment
|
||||
- ReasonTypeHealthCare
|
||||
models.NotificationStatus:
|
||||
enum:
|
||||
- 发送成功
|
||||
- 发送失败
|
||||
- 已跳过
|
||||
type: string
|
||||
x-enum-comments:
|
||||
NotificationStatusFailed: 通知发送失败
|
||||
NotificationStatusSkipped: 通知因某些原因被跳过(例如:用户未配置联系方式)
|
||||
NotificationStatusSuccess: 通知已成功发送
|
||||
x-enum-descriptions:
|
||||
- 通知已成功发送
|
||||
- 通知发送失败
|
||||
- 通知因某些原因被跳过(例如:用户未配置联系方式)
|
||||
x-enum-varnames:
|
||||
- NotificationStatusSuccess
|
||||
- NotificationStatusFailed
|
||||
- NotificationStatusSkipped
|
||||
models.PenStatus:
|
||||
enum:
|
||||
- 空闲
|
||||
@@ -1827,16 +1881,43 @@ definitions:
|
||||
type: object
|
||||
notify.NotifierType:
|
||||
enum:
|
||||
- smtp
|
||||
- wechat
|
||||
- lark
|
||||
- log
|
||||
- 邮件
|
||||
- 企业微信
|
||||
- 飞书
|
||||
- 日志
|
||||
type: string
|
||||
x-enum-varnames:
|
||||
- NotifierTypeSMTP
|
||||
- NotifierTypeWeChat
|
||||
- NotifierTypeLark
|
||||
- NotifierTypeLog
|
||||
zapcore.Level:
|
||||
enum:
|
||||
- 7
|
||||
- -1
|
||||
- 0
|
||||
- 1
|
||||
- 2
|
||||
- 3
|
||||
- 4
|
||||
- 5
|
||||
- -1
|
||||
- 5
|
||||
- 6
|
||||
format: int32
|
||||
type: integer
|
||||
x-enum-varnames:
|
||||
- _numLevels
|
||||
- DebugLevel
|
||||
- InfoLevel
|
||||
- WarnLevel
|
||||
- ErrorLevel
|
||||
- DPanicLevel
|
||||
- PanicLevel
|
||||
- FatalLevel
|
||||
- _minLevel
|
||||
- _maxLevel
|
||||
- InvalidLevel
|
||||
info:
|
||||
contact:
|
||||
email: divano@example.com
|
||||
@@ -2400,6 +2481,105 @@ paths:
|
||||
summary: 获取用药记录列表
|
||||
tags:
|
||||
- 数据监控
|
||||
/api/v1/monitor/notifications:
|
||||
get:
|
||||
description: 根据提供的过滤条件,分页获取通知列表
|
||||
parameters:
|
||||
- in: query
|
||||
name: end_time
|
||||
type: string
|
||||
- enum:
|
||||
- 7
|
||||
- -1
|
||||
- 0
|
||||
- 1
|
||||
- 2
|
||||
- 3
|
||||
- 4
|
||||
- 5
|
||||
- -1
|
||||
- 5
|
||||
- 6
|
||||
format: int32
|
||||
in: query
|
||||
name: level
|
||||
type: integer
|
||||
x-enum-varnames:
|
||||
- _numLevels
|
||||
- DebugLevel
|
||||
- InfoLevel
|
||||
- WarnLevel
|
||||
- ErrorLevel
|
||||
- DPanicLevel
|
||||
- PanicLevel
|
||||
- FatalLevel
|
||||
- _minLevel
|
||||
- _maxLevel
|
||||
- InvalidLevel
|
||||
- enum:
|
||||
- 邮件
|
||||
- 企业微信
|
||||
- 飞书
|
||||
- 日志
|
||||
in: query
|
||||
name: notifier_type
|
||||
type: string
|
||||
x-enum-varnames:
|
||||
- NotifierTypeSMTP
|
||||
- NotifierTypeWeChat
|
||||
- NotifierTypeLark
|
||||
- NotifierTypeLog
|
||||
- in: query
|
||||
name: order_by
|
||||
type: string
|
||||
- in: query
|
||||
name: page
|
||||
type: integer
|
||||
- in: query
|
||||
name: pageSize
|
||||
type: integer
|
||||
- in: query
|
||||
name: start_time
|
||||
type: string
|
||||
- enum:
|
||||
- 发送成功
|
||||
- 发送失败
|
||||
- 已跳过
|
||||
in: query
|
||||
name: status
|
||||
type: string
|
||||
x-enum-comments:
|
||||
NotificationStatusFailed: 通知发送失败
|
||||
NotificationStatusSkipped: 通知因某些原因被跳过(例如:用户未配置联系方式)
|
||||
NotificationStatusSuccess: 通知已成功发送
|
||||
x-enum-descriptions:
|
||||
- 通知已成功发送
|
||||
- 通知发送失败
|
||||
- 通知因某些原因被跳过(例如:用户未配置联系方式)
|
||||
x-enum-varnames:
|
||||
- NotificationStatusSuccess
|
||||
- NotificationStatusFailed
|
||||
- NotificationStatusSkipped
|
||||
- in: query
|
||||
name: user_id
|
||||
type: integer
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: OK
|
||||
schema:
|
||||
allOf:
|
||||
- $ref: '#/definitions/controller.Response'
|
||||
- properties:
|
||||
data:
|
||||
$ref: '#/definitions/dto.ListNotificationResponse'
|
||||
type: object
|
||||
security:
|
||||
- BearerAuth: []
|
||||
summary: 批量查询通知
|
||||
tags:
|
||||
- 数据监控
|
||||
/api/v1/monitor/pending-collections:
|
||||
get:
|
||||
description: 根据提供的过滤条件,分页获取待采集请求
|
||||
|
||||
@@ -43,7 +43,7 @@ type API struct {
|
||||
engine *gin.Engine // Gin 引擎实例,用于处理 HTTP 请求
|
||||
logger *logs.Logger // 日志记录器,用于输出日志信息
|
||||
userRepo repository.UserRepository // 用户数据仓库接口,用于用户数据操作
|
||||
tokenService token.TokenService // Token 服务接口,用于 JWT token 的生成和解析
|
||||
tokenService token.Service // Token 服务接口,用于 JWT token 的生成和解析
|
||||
auditService audit.Service // 审计服务,用于记录用户操作
|
||||
httpServer *http.Server // 标准库的 HTTP 服务器实例,用于启动和停止服务
|
||||
config config.ServerConfig // API 服务器的配置,使用 infra/config 包中的 ServerConfig
|
||||
@@ -69,7 +69,7 @@ func NewAPI(cfg config.ServerConfig,
|
||||
pigFarmService service.PigFarmService,
|
||||
pigBatchService service.PigBatchService,
|
||||
monitorService service.MonitorService,
|
||||
tokenService token.TokenService,
|
||||
tokenService token.Service,
|
||||
auditService audit.Service,
|
||||
notifyService domain_notify.Service,
|
||||
deviceService domain_device.Service,
|
||||
|
||||
@@ -176,6 +176,7 @@ func (a *API) setupRoutes() {
|
||||
monitorGroup.GET("/pig-sick-logs", a.monitorController.ListPigSickLogs)
|
||||
monitorGroup.GET("/pig-purchases", a.monitorController.ListPigPurchases)
|
||||
monitorGroup.GET("/pig-sales", a.monitorController.ListPigSales)
|
||||
monitorGroup.GET("/notifications", a.monitorController.ListNotifications)
|
||||
}
|
||||
a.logger.Info("数据监控相关接口注册成功 (需要认证和审计)")
|
||||
}
|
||||
|
||||
@@ -839,3 +839,50 @@ func (c *Controller) ListPigSales(ctx *gin.Context) {
|
||||
c.logger.Infof("%s: 成功, 获取到 %d 条记录, 总计 %d 条", actionType, len(data), total)
|
||||
controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取猪只售卖记录成功", resp, actionType, "获取猪只售卖记录成功", req)
|
||||
}
|
||||
|
||||
// ListNotifications godoc
|
||||
// @Summary 批量查询通知
|
||||
// @Description 根据提供的过滤条件,分页获取通知列表
|
||||
// @Tags 数据监控
|
||||
// @Security BearerAuth
|
||||
// @Produce json
|
||||
// @Param query query dto.ListNotificationRequest true "查询参数"
|
||||
// @Success 200 {object} controller.Response{data=dto.ListNotificationResponse}
|
||||
// @Router /api/v1/monitor/notifications [get]
|
||||
func (c *Controller) ListNotifications(ctx *gin.Context) {
|
||||
const actionType = "批量查询通知"
|
||||
|
||||
var req dto.ListNotificationRequest
|
||||
if err := ctx.ShouldBindQuery(&req); err != nil {
|
||||
c.logger.Errorf("%s: 参数绑定失败: %v", actionType, err)
|
||||
controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的查询参数: "+err.Error(), actionType, "参数绑定失败", req)
|
||||
return
|
||||
}
|
||||
|
||||
opts := repository.NotificationListOptions{
|
||||
UserID: req.UserID,
|
||||
NotifierType: req.NotifierType,
|
||||
Level: req.Level,
|
||||
StartTime: req.StartTime,
|
||||
EndTime: req.EndTime,
|
||||
OrderBy: req.OrderBy,
|
||||
Status: req.Status,
|
||||
}
|
||||
|
||||
data, total, err := c.monitorService.ListNotifications(opts, req.Page, req.PageSize)
|
||||
if err != nil {
|
||||
if errors.Is(err, repository.ErrInvalidPagination) {
|
||||
c.logger.Warnf("%s: 无效的分页参数: %v", actionType, err)
|
||||
controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的分页参数: "+err.Error(), actionType, "无效分页参数", req)
|
||||
return
|
||||
}
|
||||
|
||||
c.logger.Errorf("%s: 服务层查询失败: %v", actionType, err)
|
||||
controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "批量查询通知失败: "+err.Error(), actionType, "服务层查询失败", req)
|
||||
return
|
||||
}
|
||||
|
||||
resp := dto.NewListNotificationResponse(data, total, req.Page, req.PageSize)
|
||||
c.logger.Infof("%s: 成功, 获取到 %d 条记录, 总计 %d 条", actionType, len(data), total)
|
||||
controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "批量查询通知成功", resp, actionType, "批量查询通知成功", req)
|
||||
}
|
||||
|
||||
@@ -20,7 +20,7 @@ import (
|
||||
type Controller struct {
|
||||
userRepo repository.UserRepository
|
||||
monitorService service.MonitorService
|
||||
tokenService token.TokenService
|
||||
tokenService token.Service
|
||||
notifyService domain_notify.Service
|
||||
logger *logs.Logger
|
||||
}
|
||||
@@ -30,7 +30,7 @@ func NewController(
|
||||
userRepo repository.UserRepository,
|
||||
monitorService service.MonitorService,
|
||||
logger *logs.Logger,
|
||||
tokenService token.TokenService,
|
||||
tokenService token.Service,
|
||||
notifyService domain_notify.Service,
|
||||
) *Controller {
|
||||
return &Controller{
|
||||
|
||||
36
internal/app/dto/notification_converter.go
Normal file
36
internal/app/dto/notification_converter.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package dto
|
||||
|
||||
import (
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"go.uber.org/zap/zapcore"
|
||||
)
|
||||
|
||||
// NewListNotificationResponse 从模型数据创建通知列表响应 DTO
|
||||
func NewListNotificationResponse(data []models.Notification, total int64, page, pageSize int) *ListNotificationResponse {
|
||||
dtos := make([]NotificationDTO, len(data))
|
||||
for i, item := range data {
|
||||
dtos[i] = NotificationDTO{
|
||||
ID: item.ID,
|
||||
CreatedAt: item.CreatedAt,
|
||||
UpdatedAt: item.UpdatedAt,
|
||||
NotifierType: item.NotifierType,
|
||||
UserID: item.UserID,
|
||||
Title: item.Title,
|
||||
Message: item.Message,
|
||||
Level: zapcore.Level(item.Level),
|
||||
AlarmTimestamp: item.AlarmTimestamp,
|
||||
ToAddress: item.ToAddress,
|
||||
Status: item.Status,
|
||||
ErrorMessage: item.ErrorMessage,
|
||||
}
|
||||
}
|
||||
|
||||
return &ListNotificationResponse{
|
||||
List: dtos,
|
||||
Pagination: PaginationDTO{
|
||||
Total: total,
|
||||
Page: page,
|
||||
PageSize: pageSize,
|
||||
},
|
||||
}
|
||||
}
|
||||
@@ -1,9 +1,50 @@
|
||||
package dto
|
||||
|
||||
import "git.huangwc.com/pig/pig-farm-controller/internal/infra/notify"
|
||||
import (
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/notify"
|
||||
"go.uber.org/zap/zapcore"
|
||||
)
|
||||
|
||||
// SendTestNotificationRequest 定义了发送测试通知请求的 JSON 结构
|
||||
type SendTestNotificationRequest struct {
|
||||
// Type 指定要测试的通知渠道
|
||||
Type notify.NotifierType `json:"type" binding:"required"`
|
||||
}
|
||||
|
||||
// ListNotificationRequest 定义了获取通知列表的请求参数
|
||||
type ListNotificationRequest struct {
|
||||
Page int `form:"page,default=1"`
|
||||
PageSize int `form:"pageSize,default=10"`
|
||||
UserID *uint `form:"user_id"`
|
||||
NotifierType *notify.NotifierType `form:"notifier_type"`
|
||||
Status *models.NotificationStatus `form:"status"`
|
||||
Level *zapcore.Level `form:"level"`
|
||||
StartTime *time.Time `form:"start_time"`
|
||||
EndTime *time.Time `form:"end_time"`
|
||||
OrderBy string `form:"order_by"`
|
||||
}
|
||||
|
||||
// NotificationDTO 是用于API响应的通知结构
|
||||
type NotificationDTO struct {
|
||||
ID uint `json:"id"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
NotifierType notify.NotifierType `json:"notifier_type"`
|
||||
UserID uint `json:"user_id"`
|
||||
Title string `json:"title"`
|
||||
Message string `json:"message"`
|
||||
Level zapcore.Level `json:"level"`
|
||||
AlarmTimestamp time.Time `json:"alarm_timestamp"`
|
||||
ToAddress string `json:"to_address"`
|
||||
Status models.NotificationStatus `json:"status"`
|
||||
ErrorMessage string `json:"error_message"`
|
||||
}
|
||||
|
||||
// ListNotificationResponse 是获取通知列表的响应结构
|
||||
type ListNotificationResponse struct {
|
||||
List []NotificationDTO `json:"list"`
|
||||
Pagination PaginationDTO `json:"pagination"`
|
||||
}
|
||||
|
||||
@@ -14,7 +14,7 @@ import (
|
||||
|
||||
// AuthMiddleware 创建一个Gin中间件,用于JWT身份验证
|
||||
// 它依赖于 TokenService 来解析和验证 token,并使用 UserRepository 来获取完整的用户信息
|
||||
func AuthMiddleware(tokenService token.TokenService, userRepo repository.UserRepository) gin.HandlerFunc {
|
||||
func AuthMiddleware(tokenService token.Service, userRepo repository.UserRepository) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
// 从 Authorization header 获取 token
|
||||
authHeader := c.GetHeader("Authorization")
|
||||
|
||||
@@ -24,6 +24,7 @@ type MonitorService interface {
|
||||
ListPigSickLogs(opts repository.PigSickLogListOptions, page, pageSize int) ([]models.PigSickLog, int64, error)
|
||||
ListPigPurchases(opts repository.PigPurchaseListOptions, page, pageSize int) ([]models.PigPurchase, int64, error)
|
||||
ListPigSales(opts repository.PigSaleListOptions, page, pageSize int) ([]models.PigSale, int64, error)
|
||||
ListNotifications(opts repository.NotificationListOptions, page, pageSize int) ([]models.Notification, int64, error)
|
||||
}
|
||||
|
||||
// monitorService 是 MonitorService 接口的具体实现
|
||||
@@ -40,6 +41,7 @@ type monitorService struct {
|
||||
pigTransferLogRepo repository.PigTransferLogRepository
|
||||
pigSickLogRepo repository.PigSickLogRepository
|
||||
pigTradeRepo repository.PigTradeRepository
|
||||
notificationRepo repository.NotificationRepository
|
||||
}
|
||||
|
||||
// NewMonitorService 创建一个新的 MonitorService 实例
|
||||
@@ -56,6 +58,7 @@ func NewMonitorService(
|
||||
pigTransferLogRepo repository.PigTransferLogRepository,
|
||||
pigSickLogRepo repository.PigSickLogRepository,
|
||||
pigTradeRepo repository.PigTradeRepository,
|
||||
notificationRepo repository.NotificationRepository,
|
||||
) MonitorService {
|
||||
return &monitorService{
|
||||
sensorDataRepo: sensorDataRepo,
|
||||
@@ -70,6 +73,7 @@ func NewMonitorService(
|
||||
pigTransferLogRepo: pigTransferLogRepo,
|
||||
pigSickLogRepo: pigSickLogRepo,
|
||||
pigTradeRepo: pigTradeRepo,
|
||||
notificationRepo: notificationRepo,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -157,3 +161,8 @@ func (s *monitorService) ListPigPurchases(opts repository.PigPurchaseListOptions
|
||||
func (s *monitorService) ListPigSales(opts repository.PigSaleListOptions, page, pageSize int) ([]models.PigSale, int64, error) {
|
||||
return s.pigTradeRepo.ListPigSales(opts, page, pageSize)
|
||||
}
|
||||
|
||||
// ListNotifications 负责处理查询通知列表的业务逻辑
|
||||
func (s *monitorService) ListNotifications(opts repository.NotificationListOptions, page, pageSize int) ([]models.Notification, int64, error) {
|
||||
return s.notificationRepo.List(opts, page, pageSize)
|
||||
}
|
||||
|
||||
@@ -5,328 +5,97 @@ import (
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/api"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/service"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/webhook"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/audit"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/device"
|
||||
domain_notify "git.huangwc.com/pig/pig-farm-controller/internal/domain/notify"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/pig"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/task"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/token"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/config"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/database"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/notify"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport/lora"
|
||||
)
|
||||
|
||||
// Application 是整个应用的核心,封装了所有组件和生命周期。
|
||||
type Application struct {
|
||||
Config *config.Config
|
||||
Logger *logs.Logger
|
||||
Storage database.Storage
|
||||
Executor *task.Scheduler
|
||||
API *api.API // 添加 API 对象
|
||||
Config *config.Config
|
||||
Logger *logs.Logger
|
||||
API *api.API
|
||||
|
||||
// 新增的仓库和管理器字段,以便在 initializePendingTasks 中访问
|
||||
planRepo repository.PlanRepository
|
||||
pendingTaskRepo repository.PendingTaskRepository
|
||||
executionLogRepo repository.ExecutionLogRepository
|
||||
pendingCollectionRepo repository.PendingCollectionRepository
|
||||
analysisPlanTaskManager *task.AnalysisPlanTaskManager
|
||||
|
||||
// Lora Mesh 监听器
|
||||
loraMeshCommunicator transport.Listener
|
||||
|
||||
// 通知服务
|
||||
NotifyService domain_notify.Service
|
||||
Infra *Infrastructure
|
||||
Domain *DomainServices
|
||||
App *AppServices
|
||||
}
|
||||
|
||||
// NewApplication 创建并初始化一个新的 Application 实例。
|
||||
// 这是应用的“组合根”,所有依赖都在这里被创建和注入。
|
||||
func NewApplication(configPath string) (*Application, error) {
|
||||
// 加载配置
|
||||
// 1. 初始化基本组件: 配置和日志
|
||||
cfg := config.NewConfig()
|
||||
if err := cfg.Load(configPath); err != nil {
|
||||
return nil, fmt.Errorf("无法加载配置: %w", err)
|
||||
}
|
||||
|
||||
// 初始化日志记录器
|
||||
logger := logs.NewLogger(cfg.Log)
|
||||
|
||||
// 初始化数据库存储
|
||||
storage, err := initStorage(cfg.Database, logger)
|
||||
// 2. 初始化所有分层服务
|
||||
infra, err := initInfrastructure(cfg, logger)
|
||||
if err != nil {
|
||||
return nil, err // 错误已在 initStorage 中被包装
|
||||
return nil, fmt.Errorf("初始化基础设施失败: %w", err)
|
||||
}
|
||||
domain := initDomainServices(cfg, infra, logger)
|
||||
appServices := initAppServices(infra, domain, logger)
|
||||
|
||||
// 初始化 Token 服务
|
||||
tokenService := token.NewTokenService([]byte(cfg.App.JWTSecret))
|
||||
|
||||
// --- 仓库对象初始化 ---
|
||||
userRepo := repository.NewGormUserRepository(storage.GetDB())
|
||||
deviceRepo := repository.NewGormDeviceRepository(storage.GetDB())
|
||||
areaControllerRepo := repository.NewGormAreaControllerRepository(storage.GetDB())
|
||||
deviceTemplateRepo := repository.NewGormDeviceTemplateRepository(storage.GetDB())
|
||||
planRepo := repository.NewGormPlanRepository(storage.GetDB())
|
||||
pendingTaskRepo := repository.NewGormPendingTaskRepository(storage.GetDB())
|
||||
executionLogRepo := repository.NewGormExecutionLogRepository(storage.GetDB())
|
||||
sensorDataRepo := repository.NewGormSensorDataRepository(storage.GetDB())
|
||||
deviceCommandLogRepo := repository.NewGormDeviceCommandLogRepository(storage.GetDB())
|
||||
pendingCollectionRepo := repository.NewGormPendingCollectionRepository(storage.GetDB())
|
||||
userActionLogRepo := repository.NewGormUserActionLogRepository(storage.GetDB())
|
||||
pigBatchRepo := repository.NewGormPigBatchRepository(storage.GetDB())
|
||||
pigBatchLogRepo := repository.NewGormPigBatchLogRepository(storage.GetDB())
|
||||
pigFarmRepo := repository.NewGormPigFarmRepository(storage.GetDB())
|
||||
pigPenRepo := repository.NewGormPigPenRepository(storage.GetDB())
|
||||
pigTransferLogRepo := repository.NewGormPigTransferLogRepository(storage.GetDB())
|
||||
pigTradeRepo := repository.NewGormPigTradeRepository(storage.GetDB())
|
||||
pigSickPigLogRepo := repository.NewGormPigSickLogRepository(storage.GetDB())
|
||||
medicationLogRepo := repository.NewGormMedicationLogRepository(storage.GetDB())
|
||||
rawMaterialRepo := repository.NewGormRawMaterialRepository(storage.GetDB())
|
||||
|
||||
// 初始化事务管理器
|
||||
unitOfWork := repository.NewGormUnitOfWork(storage.GetDB(), logger)
|
||||
|
||||
// 初始化猪群管理领域
|
||||
pigPenTransferManager := pig.NewPigPenTransferManager(pigPenRepo, pigTransferLogRepo, pigBatchRepo)
|
||||
pigTradeManager := pig.NewPigTradeManager(pigTradeRepo)
|
||||
pigSickManager := pig.NewSickPigManager(pigSickPigLogRepo, medicationLogRepo)
|
||||
pigBatchDomain := pig.NewPigBatchService(pigBatchRepo, pigBatchLogRepo, unitOfWork,
|
||||
pigPenTransferManager, pigTradeManager, pigSickManager)
|
||||
|
||||
// --- 业务逻辑处理器初始化 ---
|
||||
pigFarmService := service.NewPigFarmService(pigFarmRepo, pigPenRepo, pigBatchRepo, pigBatchDomain, unitOfWork, logger)
|
||||
pigBatchService := service.NewPigBatchService(pigBatchDomain, logger)
|
||||
monitorService := service.NewMonitorService(
|
||||
sensorDataRepo,
|
||||
deviceCommandLogRepo,
|
||||
executionLogRepo,
|
||||
pendingCollectionRepo,
|
||||
userActionLogRepo,
|
||||
rawMaterialRepo,
|
||||
medicationLogRepo,
|
||||
pigBatchRepo,
|
||||
pigBatchLogRepo,
|
||||
pigTransferLogRepo,
|
||||
pigSickPigLogRepo,
|
||||
pigTradeRepo,
|
||||
)
|
||||
|
||||
// 初始化审计服务
|
||||
auditService := audit.NewService(userActionLogRepo, logger)
|
||||
|
||||
// 初始化通知服务
|
||||
notifyService, err := initNotifyService(cfg.Notify, logger, userRepo)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("初始化通知服务失败: %w", err)
|
||||
}
|
||||
|
||||
// --- 初始化 LoRa 相关组件 ---
|
||||
var listenHandler webhook.ListenHandler
|
||||
var comm transport.Communicator
|
||||
var loraListener transport.Listener
|
||||
|
||||
if cfg.Lora.Mode == config.LoraMode_LoRaWAN {
|
||||
logger.Info("当前运行模式: lora_wan。初始化 ChirpStack 监听器和传输层。")
|
||||
listenHandler = webhook.NewChirpStackListener(logger, sensorDataRepo, deviceRepo, areaControllerRepo, deviceCommandLogRepo, pendingCollectionRepo)
|
||||
comm = lora.NewChirpStackTransport(cfg.ChirpStack, logger)
|
||||
loraListener = lora.NewPlaceholderTransport(logger)
|
||||
} else {
|
||||
logger.Info("当前运行模式: lora_mesh。初始化 LoRa Mesh 传输层和占位符监听器。")
|
||||
listenHandler = webhook.NewPlaceholderListener(logger)
|
||||
tp, err := lora.NewLoRaMeshUartPassthroughTransport(cfg.LoraMesh, logger, areaControllerRepo, pendingCollectionRepo, deviceRepo, sensorDataRepo)
|
||||
loraListener = tp
|
||||
comm = tp
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("无法初始化 LoRa Mesh 模块: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 初始化计划触发器管理器
|
||||
analysisPlanTaskManager := task.NewAnalysisPlanTaskManager(planRepo, pendingTaskRepo, executionLogRepo, logger)
|
||||
|
||||
// 初始化通用设备服务
|
||||
generalDeviceService := device.NewGeneralDeviceService(
|
||||
deviceRepo,
|
||||
deviceCommandLogRepo,
|
||||
pendingCollectionRepo,
|
||||
logger,
|
||||
comm,
|
||||
)
|
||||
|
||||
// 初始化任务执行器
|
||||
executor := task.NewScheduler(
|
||||
pendingTaskRepo,
|
||||
executionLogRepo,
|
||||
deviceRepo,
|
||||
sensorDataRepo,
|
||||
planRepo,
|
||||
analysisPlanTaskManager,
|
||||
logger,
|
||||
generalDeviceService,
|
||||
time.Duration(cfg.Task.Interval)*time.Second,
|
||||
cfg.Task.NumWorkers,
|
||||
)
|
||||
|
||||
// 初始化 API 服务器
|
||||
// 3. 初始化 API 入口点
|
||||
apiServer := api.NewAPI(
|
||||
cfg.Server,
|
||||
logger,
|
||||
userRepo,
|
||||
deviceRepo,
|
||||
areaControllerRepo,
|
||||
deviceTemplateRepo,
|
||||
planRepo,
|
||||
pigFarmService,
|
||||
pigBatchService,
|
||||
monitorService,
|
||||
tokenService,
|
||||
auditService,
|
||||
notifyService,
|
||||
generalDeviceService,
|
||||
listenHandler,
|
||||
analysisPlanTaskManager,
|
||||
infra.Repos.UserRepo,
|
||||
infra.Repos.DeviceRepo,
|
||||
infra.Repos.AreaControllerRepo,
|
||||
infra.Repos.DeviceTemplateRepo,
|
||||
infra.Repos.PlanRepo,
|
||||
appServices.PigFarmService,
|
||||
appServices.PigBatchService,
|
||||
appServices.MonitorService,
|
||||
infra.TokenService,
|
||||
appServices.AuditService,
|
||||
infra.NotifyService,
|
||||
domain.GeneralDeviceService,
|
||||
infra.Lora.ListenHandler,
|
||||
domain.AnalysisPlanTaskManager,
|
||||
)
|
||||
|
||||
// 组装 Application 对象
|
||||
// 4. 组装 Application 对象
|
||||
app := &Application{
|
||||
Config: cfg,
|
||||
Logger: logger,
|
||||
Storage: storage,
|
||||
Executor: executor,
|
||||
API: apiServer,
|
||||
planRepo: planRepo,
|
||||
pendingTaskRepo: pendingTaskRepo,
|
||||
executionLogRepo: executionLogRepo,
|
||||
pendingCollectionRepo: pendingCollectionRepo,
|
||||
analysisPlanTaskManager: analysisPlanTaskManager,
|
||||
loraMeshCommunicator: loraListener,
|
||||
NotifyService: notifyService,
|
||||
Config: cfg,
|
||||
Logger: logger,
|
||||
API: apiServer,
|
||||
Infra: infra,
|
||||
Domain: domain,
|
||||
App: appServices,
|
||||
}
|
||||
|
||||
return app, nil
|
||||
}
|
||||
|
||||
// initNotifyService 根据配置初始化并返回一个通知领域服务。
|
||||
// 它确保至少有一个 LogNotifier 总是可用,并根据配置启用其他通知器。
|
||||
func initNotifyService(
|
||||
cfg config.NotifyConfig,
|
||||
log *logs.Logger,
|
||||
userRepo repository.UserRepository,
|
||||
) (domain_notify.Service, error) {
|
||||
var availableNotifiers []notify.Notifier
|
||||
|
||||
// 1. 总是创建 LogNotifier 作为所有告警的最终记录渠道
|
||||
logNotifier := notify.NewLogNotifier(log)
|
||||
availableNotifiers = append(availableNotifiers, logNotifier)
|
||||
log.Info("Log通知器已启用 (作为所有告警的最终记录渠道)")
|
||||
|
||||
// 2. 根据配置,按需创建并收集所有启用的其他 Notifier 实例
|
||||
if cfg.SMTP.Enabled {
|
||||
smtpNotifier := notify.NewSMTPNotifier(
|
||||
cfg.SMTP.Host,
|
||||
cfg.SMTP.Port,
|
||||
cfg.SMTP.Username,
|
||||
cfg.SMTP.Password,
|
||||
cfg.SMTP.Sender,
|
||||
)
|
||||
availableNotifiers = append(availableNotifiers, smtpNotifier)
|
||||
log.Info("SMTP通知器已启用")
|
||||
}
|
||||
|
||||
if cfg.WeChat.Enabled {
|
||||
wechatNotifier := notify.NewWechatNotifier(
|
||||
cfg.WeChat.CorpID,
|
||||
cfg.WeChat.AgentID,
|
||||
cfg.WeChat.Secret,
|
||||
)
|
||||
availableNotifiers = append(availableNotifiers, wechatNotifier)
|
||||
log.Info("企业微信通知器已启用")
|
||||
}
|
||||
|
||||
if cfg.Lark.Enabled {
|
||||
larkNotifier := notify.NewLarkNotifier(
|
||||
cfg.Lark.AppID,
|
||||
cfg.Lark.AppSecret,
|
||||
)
|
||||
availableNotifiers = append(availableNotifiers, larkNotifier)
|
||||
log.Info("飞书通知器已启用")
|
||||
}
|
||||
|
||||
// 3. 动态确定首选通知器
|
||||
var primaryNotifier notify.Notifier
|
||||
primaryNotifierType := notify.NotifierType(cfg.Primary)
|
||||
|
||||
// 检查用户指定的主渠道是否已启用
|
||||
for _, n := range availableNotifiers {
|
||||
if n.Type() == primaryNotifierType {
|
||||
primaryNotifier = n
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// 如果用户指定的主渠道未启用或未指定,则自动选择第一个可用的 (这将是 LogNotifier,如果其他都未启用)
|
||||
if primaryNotifier == nil {
|
||||
primaryNotifier = availableNotifiers[0] // 确保总能找到一个,因为 LogNotifier 总是存在的
|
||||
log.Warnf("配置的首选渠道 '%s' 未启用或未指定,已自动降级使用 '%s' 作为首选渠道。", cfg.Primary, primaryNotifier.Type())
|
||||
}
|
||||
|
||||
// 4. 使用创建的 Notifier 列表来组装领域服务
|
||||
notifyService, err := domain_notify.NewFailoverService(
|
||||
log,
|
||||
userRepo,
|
||||
availableNotifiers,
|
||||
primaryNotifier.Type(),
|
||||
cfg.FailureThreshold,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建故障转移通知服务失败: %w", err)
|
||||
}
|
||||
|
||||
log.Infof("通知服务初始化成功,首选渠道: %s, 故障阈值: %d", primaryNotifier.Type(), cfg.FailureThreshold)
|
||||
return notifyService, nil
|
||||
}
|
||||
|
||||
// Start 启动应用的所有组件并阻塞,直到接收到关闭信号。
|
||||
func (app *Application) Start() error {
|
||||
app.Logger.Info("应用启动中...")
|
||||
|
||||
// -- 启动 LoRa Mesh 监听器
|
||||
if err := app.loraMeshCommunicator.Listen(); err != nil {
|
||||
// 1. 启动底层监听器
|
||||
if err := app.Infra.Lora.LoraListener.Listen(); err != nil {
|
||||
return fmt.Errorf("启动 LoRa Mesh 监听器失败: %w", err)
|
||||
}
|
||||
|
||||
// --- 清理待采集任务 ---
|
||||
if err := app.initializePendingCollections(); err != nil {
|
||||
// 这是一个非致命错误,记录它,但应用应继续启动
|
||||
app.Logger.Error(err)
|
||||
// 2. 初始化应用状态 (清理、刷新任务等)
|
||||
if err := app.initializeState(); err != nil {
|
||||
return fmt.Errorf("初始化应用状态失败: %w", err)
|
||||
}
|
||||
|
||||
// --- 初始化待执行任务列表 ---
|
||||
if err := app.initializePendingTasks(
|
||||
app.planRepo, // 传入 planRepo
|
||||
app.pendingTaskRepo, // 传入 pendingTaskRepo
|
||||
app.executionLogRepo, // 传入 executionLogRepo
|
||||
app.analysisPlanTaskManager, // 传入 analysisPlanTaskManager
|
||||
app.Logger, // 传入 logger
|
||||
); err != nil {
|
||||
return fmt.Errorf("初始化待执行任务列表失败: %w", err)
|
||||
}
|
||||
// 3. 启动后台工作协程
|
||||
app.Domain.Scheduler.Start()
|
||||
app.Domain.TimedCollector.Start()
|
||||
|
||||
// 启动任务执行器
|
||||
app.Executor.Start()
|
||||
|
||||
// 启动 API 服务器
|
||||
// 4. 启动 API 服务器
|
||||
app.API.Start()
|
||||
|
||||
// 等待关闭信号
|
||||
// 5. 等待关闭信号
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
@@ -343,15 +112,18 @@ func (app *Application) Stop() error {
|
||||
app.API.Stop()
|
||||
|
||||
// 关闭任务执行器
|
||||
app.Executor.Stop()
|
||||
app.Domain.Scheduler.Stop()
|
||||
|
||||
// 关闭定时采集器
|
||||
app.Domain.TimedCollector.Stop()
|
||||
|
||||
// 断开数据库连接
|
||||
if err := app.Storage.Disconnect(); err != nil {
|
||||
if err := app.Infra.Storage.Disconnect(); err != nil {
|
||||
app.Logger.Errorw("数据库连接断开失败", "error", err)
|
||||
}
|
||||
|
||||
// 关闭 LoRa Mesh 监听器
|
||||
if err := app.loraMeshCommunicator.Stop(); err != nil {
|
||||
if err := app.Infra.Lora.LoraListener.Stop(); err != nil {
|
||||
app.Logger.Errorw("LoRa Mesh 监听器关闭失败", "error", err)
|
||||
}
|
||||
|
||||
@@ -362,6 +134,22 @@ func (app *Application) Stop() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// initializeState 在应用启动时准备其初始数据状态。
|
||||
// 这包括清理任何因上次异常关闭而留下的悬空任务或请求。
|
||||
func (app *Application) initializeState() error {
|
||||
// 清理待采集任务 (非致命错误)
|
||||
if err := app.initializePendingCollections(); err != nil {
|
||||
app.Logger.Errorw("清理待采集任务时发生非致命错误", "error", err)
|
||||
}
|
||||
|
||||
// 初始化待执行任务列表 (致命错误)
|
||||
if err := app.initializePendingTasks(); err != nil {
|
||||
return fmt.Errorf("初始化待执行任务列表失败: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// initializePendingCollections 在应用启动时处理所有未完成的采集请求。
|
||||
// 我们的策略是:任何在程序重启前仍处于“待处理”状态的请求,都应被视为已失败。
|
||||
// 这保证了系统在每次启动时都处于一个干净、确定的状态。
|
||||
@@ -369,7 +157,7 @@ func (app *Application) initializePendingCollections() error {
|
||||
app.Logger.Info("开始清理所有未完成的采集请求...")
|
||||
|
||||
// 直接将所有 'pending' 状态的请求更新为 'timed_out'。
|
||||
count, err := app.pendingCollectionRepo.MarkAllPendingAsTimedOut()
|
||||
count, err := app.Infra.Repos.PendingCollectionRepo.MarkAllPendingAsTimedOut()
|
||||
if err != nil {
|
||||
return fmt.Errorf("清理未完成的采集请求失败: %v", err)
|
||||
} else if count > 0 {
|
||||
@@ -382,13 +170,13 @@ func (app *Application) initializePendingCollections() error {
|
||||
}
|
||||
|
||||
// initializePendingTasks 在应用启动时清理并刷新待执行任务列表。
|
||||
func (app *Application) initializePendingTasks(
|
||||
planRepo repository.PlanRepository,
|
||||
pendingTaskRepo repository.PendingTaskRepository,
|
||||
executionLogRepo repository.ExecutionLogRepository,
|
||||
analysisPlanTaskManager *task.AnalysisPlanTaskManager,
|
||||
logger *logs.Logger,
|
||||
) error {
|
||||
func (app *Application) initializePendingTasks() error {
|
||||
logger := app.Logger
|
||||
planRepo := app.Infra.Repos.PlanRepo
|
||||
pendingTaskRepo := app.Infra.Repos.PendingTaskRepo
|
||||
executionLogRepo := app.Infra.Repos.ExecutionLogRepo
|
||||
analysisPlanTaskManager := app.Domain.AnalysisPlanTaskManager
|
||||
|
||||
logger.Info("开始初始化待执行任务列表...")
|
||||
|
||||
// 阶段一:修正因崩溃导致状态不一致的固定次数计划
|
||||
@@ -475,21 +263,3 @@ func (app *Application) initializePendingTasks(
|
||||
logger.Info("待执行任务列表初始化完成。")
|
||||
return nil
|
||||
}
|
||||
|
||||
// initStorage 封装了数据库的初始化、连接和迁移逻辑。
|
||||
func initStorage(cfg config.DatabaseConfig, logger *logs.Logger) (database.Storage, error) {
|
||||
// 创建存储实例
|
||||
storage := database.NewStorage(cfg, logger)
|
||||
if err := storage.Connect(); err != nil {
|
||||
// 错误已在 Connect 内部被记录,这里只需包装并返回
|
||||
return nil, fmt.Errorf("数据库连接失败: %w", err)
|
||||
}
|
||||
|
||||
// 执行数据库迁移
|
||||
if err := storage.Migrate(models.GetAllModels()...); err != nil {
|
||||
return nil, fmt.Errorf("数据库迁移失败: %w", err)
|
||||
}
|
||||
|
||||
logger.Info("数据库初始化完成。")
|
||||
return storage, nil
|
||||
}
|
||||
|
||||
362
internal/core/initializers.go
Normal file
362
internal/core/initializers.go
Normal file
@@ -0,0 +1,362 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/service"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/webhook"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/audit"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/collection"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/device"
|
||||
domain_notify "git.huangwc.com/pig/pig-farm-controller/internal/domain/notify"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/pig"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/task"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/token"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/config"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/database"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/notify"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport/lora"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// Infrastructure 聚合了所有基础设施层的组件。
|
||||
type Infrastructure struct {
|
||||
Storage database.Storage
|
||||
Repos *Repositories
|
||||
Lora *LoraComponents
|
||||
NotifyService domain_notify.Service
|
||||
TokenService token.Service
|
||||
}
|
||||
|
||||
// initInfrastructure 初始化所有基础设施层组件。
|
||||
func initInfrastructure(cfg *config.Config, logger *logs.Logger) (*Infrastructure, error) {
|
||||
storage, err := initStorage(cfg.Database, logger)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
repos := initRepositories(storage.GetDB(), logger)
|
||||
|
||||
lora, err := initLora(cfg, logger, repos)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
notifyService, err := initNotifyService(cfg.Notify, logger, repos.UserRepo, repos.NotificationRepo)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("初始化通知服务失败: %w", err)
|
||||
}
|
||||
|
||||
tokenService := token.NewTokenService([]byte(cfg.App.JWTSecret))
|
||||
|
||||
return &Infrastructure{
|
||||
Storage: storage,
|
||||
Repos: repos,
|
||||
Lora: lora,
|
||||
NotifyService: notifyService,
|
||||
TokenService: tokenService,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Repositories 聚合了所有的仓库实例。
|
||||
type Repositories struct {
|
||||
UserRepo repository.UserRepository
|
||||
DeviceRepo repository.DeviceRepository
|
||||
AreaControllerRepo repository.AreaControllerRepository
|
||||
DeviceTemplateRepo repository.DeviceTemplateRepository
|
||||
PlanRepo repository.PlanRepository
|
||||
PendingTaskRepo repository.PendingTaskRepository
|
||||
ExecutionLogRepo repository.ExecutionLogRepository
|
||||
SensorDataRepo repository.SensorDataRepository
|
||||
DeviceCommandLogRepo repository.DeviceCommandLogRepository
|
||||
PendingCollectionRepo repository.PendingCollectionRepository
|
||||
UserActionLogRepo repository.UserActionLogRepository
|
||||
PigBatchRepo repository.PigBatchRepository
|
||||
PigBatchLogRepo repository.PigBatchLogRepository
|
||||
PigFarmRepo repository.PigFarmRepository
|
||||
PigPenRepo repository.PigPenRepository
|
||||
PigTransferLogRepo repository.PigTransferLogRepository
|
||||
PigTradeRepo repository.PigTradeRepository
|
||||
PigSickPigLogRepo repository.PigSickLogRepository
|
||||
MedicationLogRepo repository.MedicationLogRepository
|
||||
RawMaterialRepo repository.RawMaterialRepository
|
||||
NotificationRepo repository.NotificationRepository
|
||||
UnitOfWork repository.UnitOfWork
|
||||
}
|
||||
|
||||
// initRepositories 初始化所有的仓库。
|
||||
func initRepositories(db *gorm.DB, logger *logs.Logger) *Repositories {
|
||||
return &Repositories{
|
||||
UserRepo: repository.NewGormUserRepository(db),
|
||||
DeviceRepo: repository.NewGormDeviceRepository(db),
|
||||
AreaControllerRepo: repository.NewGormAreaControllerRepository(db),
|
||||
DeviceTemplateRepo: repository.NewGormDeviceTemplateRepository(db),
|
||||
PlanRepo: repository.NewGormPlanRepository(db),
|
||||
PendingTaskRepo: repository.NewGormPendingTaskRepository(db),
|
||||
ExecutionLogRepo: repository.NewGormExecutionLogRepository(db),
|
||||
SensorDataRepo: repository.NewGormSensorDataRepository(db),
|
||||
DeviceCommandLogRepo: repository.NewGormDeviceCommandLogRepository(db),
|
||||
PendingCollectionRepo: repository.NewGormPendingCollectionRepository(db),
|
||||
UserActionLogRepo: repository.NewGormUserActionLogRepository(db),
|
||||
PigBatchRepo: repository.NewGormPigBatchRepository(db),
|
||||
PigBatchLogRepo: repository.NewGormPigBatchLogRepository(db),
|
||||
PigFarmRepo: repository.NewGormPigFarmRepository(db),
|
||||
PigPenRepo: repository.NewGormPigPenRepository(db),
|
||||
PigTransferLogRepo: repository.NewGormPigTransferLogRepository(db),
|
||||
PigTradeRepo: repository.NewGormPigTradeRepository(db),
|
||||
PigSickPigLogRepo: repository.NewGormPigSickLogRepository(db),
|
||||
MedicationLogRepo: repository.NewGormMedicationLogRepository(db),
|
||||
RawMaterialRepo: repository.NewGormRawMaterialRepository(db),
|
||||
NotificationRepo: repository.NewGormNotificationRepository(db),
|
||||
UnitOfWork: repository.NewGormUnitOfWork(db, logger),
|
||||
}
|
||||
}
|
||||
|
||||
// DomainServices 聚合了所有的领域服务实例。
|
||||
type DomainServices struct {
|
||||
PigPenTransferManager pig.PigPenTransferManager
|
||||
PigTradeManager pig.PigTradeManager
|
||||
PigSickManager pig.SickPigManager
|
||||
PigBatchDomain pig.PigBatchService
|
||||
TimedCollector collection.Collector
|
||||
GeneralDeviceService device.Service
|
||||
AnalysisPlanTaskManager *task.AnalysisPlanTaskManager
|
||||
Scheduler *task.Scheduler
|
||||
}
|
||||
|
||||
// initDomainServices 初始化所有的领域服务。
|
||||
func initDomainServices(cfg *config.Config, infra *Infrastructure, logger *logs.Logger) *DomainServices {
|
||||
// 猪群管理相关
|
||||
pigPenTransferManager := pig.NewPigPenTransferManager(infra.Repos.PigPenRepo, infra.Repos.PigTransferLogRepo, infra.Repos.PigBatchRepo)
|
||||
pigTradeManager := pig.NewPigTradeManager(infra.Repos.PigTradeRepo)
|
||||
pigSickManager := pig.NewSickPigManager(infra.Repos.PigSickPigLogRepo, infra.Repos.MedicationLogRepo)
|
||||
pigBatchDomain := pig.NewPigBatchService(infra.Repos.PigBatchRepo, infra.Repos.PigBatchLogRepo, infra.Repos.UnitOfWork,
|
||||
pigPenTransferManager, pigTradeManager, pigSickManager)
|
||||
|
||||
// 通用设备服务
|
||||
generalDeviceService := device.NewGeneralDeviceService(
|
||||
infra.Repos.DeviceRepo,
|
||||
infra.Repos.DeviceCommandLogRepo,
|
||||
infra.Repos.PendingCollectionRepo,
|
||||
logger,
|
||||
infra.Lora.Comm,
|
||||
)
|
||||
|
||||
// 计划任务管理器
|
||||
analysisPlanTaskManager := task.NewAnalysisPlanTaskManager(infra.Repos.PlanRepo, infra.Repos.PendingTaskRepo, infra.Repos.ExecutionLogRepo, logger)
|
||||
|
||||
// 任务执行器
|
||||
scheduler := task.NewScheduler(
|
||||
infra.Repos.PendingTaskRepo,
|
||||
infra.Repos.ExecutionLogRepo,
|
||||
infra.Repos.DeviceRepo,
|
||||
infra.Repos.SensorDataRepo,
|
||||
infra.Repos.PlanRepo,
|
||||
analysisPlanTaskManager,
|
||||
logger,
|
||||
generalDeviceService,
|
||||
time.Duration(cfg.Task.Interval)*time.Second,
|
||||
cfg.Task.NumWorkers,
|
||||
)
|
||||
|
||||
// 定时采集器
|
||||
timedCollector := collection.NewTimedCollector(
|
||||
infra.Repos.DeviceRepo,
|
||||
generalDeviceService,
|
||||
logger,
|
||||
time.Duration(cfg.Collection.Interval)*time.Second,
|
||||
)
|
||||
|
||||
return &DomainServices{
|
||||
PigPenTransferManager: pigPenTransferManager,
|
||||
PigTradeManager: pigTradeManager,
|
||||
PigSickManager: pigSickManager,
|
||||
PigBatchDomain: pigBatchDomain,
|
||||
GeneralDeviceService: generalDeviceService,
|
||||
AnalysisPlanTaskManager: analysisPlanTaskManager,
|
||||
Scheduler: scheduler,
|
||||
TimedCollector: timedCollector,
|
||||
}
|
||||
}
|
||||
|
||||
// AppServices 聚合了所有的应用服务实例。
|
||||
type AppServices struct {
|
||||
PigFarmService service.PigFarmService
|
||||
PigBatchService service.PigBatchService
|
||||
MonitorService service.MonitorService
|
||||
AuditService audit.Service
|
||||
}
|
||||
|
||||
// initAppServices 初始化所有的应用服务。
|
||||
func initAppServices(infra *Infrastructure, domainServices *DomainServices, logger *logs.Logger) *AppServices {
|
||||
pigFarmService := service.NewPigFarmService(infra.Repos.PigFarmRepo, infra.Repos.PigPenRepo, infra.Repos.PigBatchRepo, domainServices.PigBatchDomain, infra.Repos.UnitOfWork, logger)
|
||||
pigBatchService := service.NewPigBatchService(domainServices.PigBatchDomain, logger)
|
||||
monitorService := service.NewMonitorService(
|
||||
infra.Repos.SensorDataRepo,
|
||||
infra.Repos.DeviceCommandLogRepo,
|
||||
infra.Repos.ExecutionLogRepo,
|
||||
infra.Repos.PendingCollectionRepo,
|
||||
infra.Repos.UserActionLogRepo,
|
||||
infra.Repos.RawMaterialRepo,
|
||||
infra.Repos.MedicationLogRepo,
|
||||
infra.Repos.PigBatchRepo,
|
||||
infra.Repos.PigBatchLogRepo,
|
||||
infra.Repos.PigTransferLogRepo,
|
||||
infra.Repos.PigSickPigLogRepo,
|
||||
infra.Repos.PigTradeRepo,
|
||||
infra.Repos.NotificationRepo,
|
||||
)
|
||||
auditService := audit.NewService(infra.Repos.UserActionLogRepo, logger)
|
||||
|
||||
return &AppServices{
|
||||
PigFarmService: pigFarmService,
|
||||
PigBatchService: pigBatchService,
|
||||
MonitorService: monitorService,
|
||||
AuditService: auditService,
|
||||
}
|
||||
}
|
||||
|
||||
// LoraComponents 聚合了所有 LoRa 相关组件。
|
||||
type LoraComponents struct {
|
||||
ListenHandler webhook.ListenHandler
|
||||
Comm transport.Communicator
|
||||
LoraListener transport.Listener
|
||||
}
|
||||
|
||||
// initLora 根据配置初始化 LoRa 相关组件。
|
||||
func initLora(
|
||||
cfg *config.Config,
|
||||
logger *logs.Logger,
|
||||
repos *Repositories,
|
||||
) (*LoraComponents, error) {
|
||||
var listenHandler webhook.ListenHandler
|
||||
var comm transport.Communicator
|
||||
var loraListener transport.Listener
|
||||
|
||||
if cfg.Lora.Mode == config.LoraMode_LoRaWAN {
|
||||
logger.Info("当前运行模式: lora_wan。初始化 ChirpStack 监听器和传输层。")
|
||||
listenHandler = webhook.NewChirpStackListener(logger, repos.SensorDataRepo, repos.DeviceRepo, repos.AreaControllerRepo, repos.DeviceCommandLogRepo, repos.PendingCollectionRepo)
|
||||
comm = lora.NewChirpStackTransport(cfg.ChirpStack, logger)
|
||||
loraListener = lora.NewPlaceholderTransport(logger)
|
||||
} else {
|
||||
logger.Info("当前运行模式: lora_mesh。初始化 LoRa Mesh 传输层和占位符监听器。")
|
||||
listenHandler = webhook.NewPlaceholderListener(logger)
|
||||
tp, err := lora.NewLoRaMeshUartPassthroughTransport(cfg.LoraMesh, logger, repos.AreaControllerRepo, repos.PendingCollectionRepo, repos.DeviceRepo, repos.SensorDataRepo)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("无法初始化 LoRa Mesh 模块: %w", err)
|
||||
}
|
||||
loraListener = tp
|
||||
comm = tp
|
||||
}
|
||||
|
||||
return &LoraComponents{
|
||||
ListenHandler: listenHandler,
|
||||
Comm: comm,
|
||||
LoraListener: loraListener,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// initNotifyService 根据配置初始化并返回一个通知领域服务。
|
||||
// 它确保至少有一个 LogNotifier 总是可用,并根据配置启用其他通知器。
|
||||
func initNotifyService(
|
||||
cfg config.NotifyConfig,
|
||||
log *logs.Logger,
|
||||
userRepo repository.UserRepository,
|
||||
notificationRepo repository.NotificationRepository,
|
||||
) (domain_notify.Service, error) {
|
||||
var availableNotifiers []notify.Notifier
|
||||
|
||||
// 1. 总是创建 LogNotifier 作为所有告警的最终记录渠道
|
||||
logNotifier := notify.NewLogNotifier(log)
|
||||
availableNotifiers = append(availableNotifiers, logNotifier)
|
||||
log.Info("Log通知器已启用 (作为所有告警的最终记录渠道)")
|
||||
|
||||
// 2. 根据配置,按需创建并收集所有启用的其他 Notifier 实例
|
||||
if cfg.SMTP.Enabled {
|
||||
smtpNotifier := notify.NewSMTPNotifier(
|
||||
cfg.SMTP.Host,
|
||||
cfg.SMTP.Port,
|
||||
cfg.SMTP.Username,
|
||||
cfg.SMTP.Password,
|
||||
cfg.SMTP.Sender,
|
||||
)
|
||||
availableNotifiers = append(availableNotifiers, smtpNotifier)
|
||||
log.Info("SMTP通知器已启用")
|
||||
}
|
||||
|
||||
if cfg.WeChat.Enabled {
|
||||
wechatNotifier := notify.NewWechatNotifier(
|
||||
cfg.WeChat.CorpID,
|
||||
cfg.WeChat.AgentID,
|
||||
cfg.WeChat.Secret,
|
||||
)
|
||||
availableNotifiers = append(availableNotifiers, wechatNotifier)
|
||||
log.Info("企业微信通知器已启用")
|
||||
}
|
||||
|
||||
if cfg.Lark.Enabled {
|
||||
larkNotifier := notify.NewLarkNotifier(
|
||||
cfg.Lark.AppID,
|
||||
cfg.Lark.AppSecret,
|
||||
)
|
||||
availableNotifiers = append(availableNotifiers, larkNotifier)
|
||||
log.Info("飞书通知器已启用")
|
||||
}
|
||||
|
||||
// 3. 动态确定首选通知器
|
||||
var primaryNotifier notify.Notifier
|
||||
primaryNotifierType := notify.NotifierType(cfg.Primary)
|
||||
|
||||
// 检查用户指定的主渠道是否已启用
|
||||
for _, n := range availableNotifiers {
|
||||
if n.Type() == primaryNotifierType {
|
||||
primaryNotifier = n
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// 如果用户指定的主渠道未启用或未指定,则自动选择第一个可用的 (这将是 LogNotifier,如果其他都未启用)
|
||||
if primaryNotifier == nil {
|
||||
primaryNotifier = availableNotifiers[0] // 确保总能找到一个,因为 LogNotifier 总是存在的
|
||||
log.Warnf("配置的首选渠道 '%s' 未启用或未指定,已自动降级使用 '%s' 作为首选渠道。", cfg.Primary, primaryNotifier.Type())
|
||||
}
|
||||
|
||||
// 4. 使用创建的 Notifier 列表和 notificationRepo 来组装领域服务
|
||||
notifyService, err := domain_notify.NewFailoverService(
|
||||
log,
|
||||
userRepo,
|
||||
availableNotifiers,
|
||||
primaryNotifier.Type(),
|
||||
cfg.FailureThreshold,
|
||||
notificationRepo,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建故障转移通知服务失败: %w", err)
|
||||
}
|
||||
|
||||
log.Infof("通知服务初始化成功,首选渠道: %s, 故障阈值: %d", primaryNotifier.Type(), cfg.FailureThreshold)
|
||||
return notifyService, nil
|
||||
}
|
||||
|
||||
// initStorage 封装了数据库的初始化、连接和迁移逻辑。
|
||||
func initStorage(cfg config.DatabaseConfig, logger *logs.Logger) (database.Storage, error) {
|
||||
// 创建存储实例
|
||||
storage := database.NewStorage(cfg, logger)
|
||||
if err := storage.Connect(); err != nil {
|
||||
// 错误已在 Connect 内部被记录,这里只需包装并返回
|
||||
return nil, fmt.Errorf("数据库连接失败: %w", err)
|
||||
}
|
||||
|
||||
// 执行数据库迁移
|
||||
if err := storage.Migrate(models.GetAllModels()...); err != nil {
|
||||
return nil, fmt.Errorf("数据库迁移失败: %w", err)
|
||||
}
|
||||
|
||||
logger.Info("数据库初始化完成。")
|
||||
return storage, nil
|
||||
}
|
||||
6
internal/domain/collection/collector.go
Normal file
6
internal/domain/collection/collector.go
Normal file
@@ -0,0 +1,6 @@
|
||||
package collection
|
||||
|
||||
type Collector interface {
|
||||
Start()
|
||||
Stop()
|
||||
}
|
||||
89
internal/domain/collection/timed_collector.go
Normal file
89
internal/domain/collection/timed_collector.go
Normal file
@@ -0,0 +1,89 @@
|
||||
package collection
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/domain/device"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
|
||||
)
|
||||
|
||||
// TimedCollector 实现了 Collector 接口,用于定时从数据库获取设备信息并下发采集指令
|
||||
type TimedCollector struct {
|
||||
deviceRepo repository.DeviceRepository
|
||||
deviceService device.Service
|
||||
logger *logs.Logger
|
||||
interval time.Duration
|
||||
ticker *time.Ticker
|
||||
done chan bool
|
||||
}
|
||||
|
||||
// NewTimedCollector 创建一个定时采集器实例
|
||||
func NewTimedCollector(
|
||||
deviceRepo repository.DeviceRepository,
|
||||
deviceService device.Service,
|
||||
logger *logs.Logger,
|
||||
interval time.Duration,
|
||||
) Collector {
|
||||
return &TimedCollector{
|
||||
deviceRepo: deviceRepo,
|
||||
deviceService: deviceService,
|
||||
logger: logger,
|
||||
interval: interval,
|
||||
done: make(chan bool),
|
||||
}
|
||||
}
|
||||
|
||||
// Start 开始定时采集
|
||||
func (c *TimedCollector) Start() {
|
||||
c.logger.Infof("定时采集器启动,采集间隔: %s", c.interval)
|
||||
c.ticker = time.NewTicker(c.interval)
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case <-c.done:
|
||||
return
|
||||
case <-c.ticker.C:
|
||||
c.collect()
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// Stop 停止定时采集
|
||||
func (c *TimedCollector) Stop() {
|
||||
c.logger.Info("定时采集器停止")
|
||||
c.ticker.Stop()
|
||||
c.done <- true
|
||||
}
|
||||
|
||||
// collect 是核心的采集逻辑
|
||||
func (c *TimedCollector) collect() {
|
||||
c.logger.Info("开始新一轮的设备数据采集")
|
||||
|
||||
sensors, err := c.deviceRepo.ListAllSensors()
|
||||
if err != nil {
|
||||
c.logger.Errorf("采集周期: 从数据库获取所有传感器失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if len(sensors) == 0 {
|
||||
c.logger.Info("采集周期: 未发现任何传感器设备,跳过本次采集")
|
||||
return
|
||||
}
|
||||
|
||||
sensorsByController := make(map[uint][]*models.Device)
|
||||
for _, sensor := range sensors {
|
||||
sensorsByController[sensor.AreaControllerID] = append(sensorsByController[sensor.AreaControllerID], sensor)
|
||||
}
|
||||
|
||||
for controllerID, controllerSensors := range sensorsByController {
|
||||
c.logger.Infof("采集周期: 准备为区域主控 %d 下的 %d 个传感器下发采集指令", controllerID, len(controllerSensors))
|
||||
if err := c.deviceService.Collect(controllerID, controllerSensors); err != nil {
|
||||
c.logger.Errorf("采集周期: 为区域主控 %d 下发采集指令失败: %v", controllerID, err)
|
||||
}
|
||||
}
|
||||
|
||||
c.logger.Info("本轮设备数据采集完成")
|
||||
}
|
||||
@@ -33,6 +33,7 @@ type failoverService struct {
|
||||
primaryNotifier notify.Notifier
|
||||
failureThreshold int
|
||||
failureCounters *sync.Map // 使用 sync.Map 来安全地并发读写失败计数, key: userID (uint), value: counter (int)
|
||||
notificationRepo repository.NotificationRepository
|
||||
}
|
||||
|
||||
// NewFailoverService 创建一个新的故障转移通知服务
|
||||
@@ -42,6 +43,7 @@ func NewFailoverService(
|
||||
notifiers []notify.Notifier,
|
||||
primaryNotifierType notify.NotifierType,
|
||||
failureThreshold int,
|
||||
notificationRepo repository.NotificationRepository,
|
||||
) (Service, error) {
|
||||
notifierMap := make(map[notify.NotifierType]notify.Notifier)
|
||||
for _, n := range notifiers {
|
||||
@@ -60,6 +62,7 @@ func NewFailoverService(
|
||||
primaryNotifier: primaryNotifier,
|
||||
failureThreshold: failureThreshold,
|
||||
failureCounters: &sync.Map{},
|
||||
notificationRepo: notificationRepo,
|
||||
}, nil
|
||||
}
|
||||
|
||||
@@ -128,11 +131,15 @@ func (s *failoverService) sendAlarmToUser(userID uint, content notify.AlarmConte
|
||||
primaryType := s.primaryNotifier.Type()
|
||||
addr := getAddressForNotifier(primaryType, user.Contact)
|
||||
if addr == "" {
|
||||
// 记录跳过通知
|
||||
s.recordNotificationAttempt(userID, primaryType, content, "", models.NotificationStatusSkipped, fmt.Errorf("用户未配置首选通知方式 '%s' 的地址", primaryType))
|
||||
return fmt.Errorf("用户未配置首选通知方式 '%s' 的地址", primaryType)
|
||||
}
|
||||
|
||||
err = s.primaryNotifier.Send(content, addr)
|
||||
if err == nil {
|
||||
// 记录成功通知
|
||||
s.recordNotificationAttempt(userID, primaryType, content, addr, models.NotificationStatusSuccess, nil)
|
||||
if failureCount > 0 {
|
||||
s.log.Infow("首选渠道发送恢复正常", "userID", userID, "notifierType", primaryType)
|
||||
s.failureCounters.Store(userID, 0)
|
||||
@@ -140,6 +147,8 @@ func (s *failoverService) sendAlarmToUser(userID uint, content notify.AlarmConte
|
||||
return nil
|
||||
}
|
||||
|
||||
// 记录失败通知
|
||||
s.recordNotificationAttempt(userID, primaryType, content, addr, models.NotificationStatusFailed, err)
|
||||
newFailureCount := failureCount + 1
|
||||
s.failureCounters.Store(userID, newFailureCount)
|
||||
s.log.Warnw("首选渠道发送失败", "userID", userID, "notifierType", primaryType, "error", err, "failureCount", newFailureCount)
|
||||
@@ -152,13 +161,19 @@ func (s *failoverService) sendAlarmToUser(userID uint, content notify.AlarmConte
|
||||
for _, notifier := range s.notifiers {
|
||||
addr := getAddressForNotifier(notifier.Type(), user.Contact)
|
||||
if addr == "" {
|
||||
// 记录跳过通知
|
||||
s.recordNotificationAttempt(userID, notifier.Type(), content, "", models.NotificationStatusSkipped, fmt.Errorf("用户未配置通知方式 '%s' 的地址", notifier.Type()))
|
||||
continue
|
||||
}
|
||||
if err := notifier.Send(content, addr); err == nil {
|
||||
// 记录成功通知
|
||||
s.recordNotificationAttempt(userID, notifier.Type(), content, addr, models.NotificationStatusSuccess, nil)
|
||||
s.log.Infow("广播通知成功", "userID", userID, "notifierType", notifier.Type())
|
||||
s.failureCounters.Store(userID, 0)
|
||||
return nil
|
||||
}
|
||||
// 记录失败通知
|
||||
s.recordNotificationAttempt(userID, notifier.Type(), content, addr, models.NotificationStatusFailed, err)
|
||||
lastErr = err
|
||||
s.log.Warnw("广播通知:渠道发送失败", "userID", userID, "notifierType", notifier.Type(), "error", err)
|
||||
}
|
||||
@@ -185,6 +200,13 @@ func (s *failoverService) SendTestMessage(userID uint, notifierType notify.Notif
|
||||
addr := getAddressForNotifier(notifierType, user.Contact)
|
||||
if addr == "" {
|
||||
s.log.Warnw("发送测试消息失败:缺少地址", "userID", userID, "notifierType", notifierType)
|
||||
// 记录跳过通知
|
||||
s.recordNotificationAttempt(userID, notifierType, notify.AlarmContent{
|
||||
Title: "通知服务测试",
|
||||
Message: fmt.Sprintf("这是一条来自【%s】渠道的测试消息。如果您收到此消息,说明您的配置正确。", notifierType),
|
||||
Level: zap.InfoLevel,
|
||||
Timestamp: time.Now(),
|
||||
}, "", models.NotificationStatusFailed, fmt.Errorf("用户未配置通知方式 '%s' 的地址", notifierType))
|
||||
return fmt.Errorf("用户未配置通知方式 '%s' 的地址", notifierType)
|
||||
}
|
||||
|
||||
@@ -199,10 +221,14 @@ func (s *failoverService) SendTestMessage(userID uint, notifierType notify.Notif
|
||||
err = notifier.Send(testContent, addr)
|
||||
if err != nil {
|
||||
s.log.Errorw("发送测试消息失败", "userID", userID, "notifierType", notifierType, "error", err)
|
||||
// 记录失败通知
|
||||
s.recordNotificationAttempt(userID, notifierType, testContent, addr, models.NotificationStatusFailed, err)
|
||||
return err
|
||||
}
|
||||
|
||||
s.log.Infow("发送测试消息成功", "userID", userID, "notifierType", notifierType)
|
||||
// 记录成功通知
|
||||
s.recordNotificationAttempt(userID, notifierType, testContent, addr, models.NotificationStatusSuccess, nil)
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -221,3 +247,46 @@ func getAddressForNotifier(notifierType notify.NotifierType, contact models.Cont
|
||||
return ""
|
||||
}
|
||||
}
|
||||
|
||||
// recordNotificationAttempt 记录一次通知发送尝试的结果
|
||||
// userID: 接收通知的用户ID
|
||||
// notifierType: 使用的通知器类型
|
||||
// content: 通知内容
|
||||
// toAddress: 实际发送到的地址
|
||||
// status: 发送尝试的状态 (成功、失败、跳过)
|
||||
// err: 如果发送失败,记录的错误信息
|
||||
func (s *failoverService) recordNotificationAttempt(
|
||||
userID uint,
|
||||
notifierType notify.NotifierType,
|
||||
content notify.AlarmContent,
|
||||
toAddress string,
|
||||
status models.NotificationStatus,
|
||||
err error,
|
||||
) {
|
||||
errorMessage := ""
|
||||
if err != nil {
|
||||
errorMessage = err.Error()
|
||||
}
|
||||
|
||||
notification := &models.Notification{
|
||||
NotifierType: notifierType,
|
||||
UserID: userID,
|
||||
Title: content.Title,
|
||||
Message: content.Message,
|
||||
Level: models.LogLevel(content.Level),
|
||||
AlarmTimestamp: content.Timestamp,
|
||||
ToAddress: toAddress,
|
||||
Status: status,
|
||||
ErrorMessage: errorMessage,
|
||||
}
|
||||
|
||||
if saveErr := s.notificationRepo.Create(notification); saveErr != nil {
|
||||
s.log.Errorw("无法保存通知发送记录到数据库",
|
||||
"userID", userID,
|
||||
"notifierType", notifierType,
|
||||
"status", status,
|
||||
"originalError", errorMessage,
|
||||
"saveError", saveErr,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -13,19 +13,19 @@ type Claims struct {
|
||||
jwt.RegisteredClaims
|
||||
}
|
||||
|
||||
// TokenService 定义了 token 操作的接口
|
||||
type TokenService interface {
|
||||
// Service 定义了 token 操作的接口
|
||||
type Service interface {
|
||||
GenerateToken(userID uint) (string, error)
|
||||
ParseToken(tokenString string) (*Claims, error)
|
||||
}
|
||||
|
||||
// tokenService 是 TokenService 接口的实现
|
||||
// tokenService 是 Service 接口的实现
|
||||
type tokenService struct {
|
||||
secret []byte
|
||||
}
|
||||
|
||||
// NewTokenService 创建并返回一个新的 TokenService 实例
|
||||
func NewTokenService(secret []byte) TokenService {
|
||||
// NewTokenService 创建并返回一个新的 Service 实例
|
||||
func NewTokenService(secret []byte) Service {
|
||||
return &tokenService{secret: secret}
|
||||
}
|
||||
|
||||
|
||||
@@ -44,6 +44,9 @@ type Config struct {
|
||||
|
||||
// Notify 通知服务配置
|
||||
Notify NotifyConfig `yaml:"notify"`
|
||||
|
||||
// Collection 定时采集配置
|
||||
Collection CollectionConfig `yaml:"collection"`
|
||||
}
|
||||
|
||||
// AppConfig 代表应用基础配置
|
||||
@@ -195,6 +198,11 @@ type LarkConfig struct {
|
||||
AppSecret string `yaml:"appSecret"`
|
||||
}
|
||||
|
||||
// CollectionConfig 代表定时采集配置
|
||||
type CollectionConfig struct {
|
||||
Interval int `yaml:"interval"`
|
||||
}
|
||||
|
||||
// NewConfig 创建并返回一个新的配置实例
|
||||
func NewConfig() *Config {
|
||||
// 默认值可以在这里设置,但我们优先使用配置文件中的值
|
||||
|
||||
@@ -171,6 +171,7 @@ func (ps *PostgresStorage) creatingHyperTable() error {
|
||||
{models.PigSickLog{}, "happened_at"},
|
||||
{models.PigPurchase{}, "purchase_date"},
|
||||
{models.PigSale{}, "sale_date"},
|
||||
{models.Notification{}, "alarm_timestamp"},
|
||||
}
|
||||
|
||||
for _, table := range tablesToConvert {
|
||||
@@ -211,6 +212,7 @@ func (ps *PostgresStorage) applyCompressionPolicies() error {
|
||||
{models.PigSickLog{}, "pig_batch_id"},
|
||||
{models.PigPurchase{}, "pig_batch_id"},
|
||||
{models.PigSale{}, "pig_batch_id"},
|
||||
{models.Notification{}, "user_id"},
|
||||
}
|
||||
|
||||
for _, policy := range policies {
|
||||
|
||||
@@ -59,6 +59,9 @@ func GetAllModels() []interface{} {
|
||||
// Medication Models
|
||||
&Medication{},
|
||||
&MedicationLog{},
|
||||
|
||||
// Notification Models
|
||||
&Notification{},
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
77
internal/infra/models/notify.go
Normal file
77
internal/infra/models/notify.go
Normal file
@@ -0,0 +1,77 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"database/sql/driver"
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/notify"
|
||||
"go.uber.org/zap/zapcore"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// NotificationStatus 定义了通知发送尝试的状态枚举。
|
||||
type NotificationStatus string
|
||||
|
||||
const (
|
||||
NotificationStatusSuccess NotificationStatus = "发送成功" // 通知已成功发送
|
||||
NotificationStatusFailed NotificationStatus = "发送失败" // 通知发送失败
|
||||
NotificationStatusSkipped NotificationStatus = "已跳过" // 通知因某些原因被跳过(例如:用户未配置联系方式)
|
||||
)
|
||||
|
||||
// LogLevel is a custom type for zapcore.Level to handle database scanning and valuing.
|
||||
type LogLevel zapcore.Level
|
||||
|
||||
// Scan implements the sql.Scanner interface.
|
||||
func (l *LogLevel) Scan(value interface{}) error {
|
||||
var s string
|
||||
switch v := value.(type) {
|
||||
case []byte:
|
||||
s = string(v)
|
||||
case string:
|
||||
s = v
|
||||
default:
|
||||
return errors.New("LogLevel的类型无效")
|
||||
}
|
||||
|
||||
var zl zapcore.Level
|
||||
if err := zl.UnmarshalText([]byte(s)); err != nil {
|
||||
return err
|
||||
}
|
||||
*l = LogLevel(zl)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Value implements the driver.Valuer interface.
|
||||
func (l LogLevel) Value() (driver.Value, error) {
|
||||
return (zapcore.Level)(l).String(), nil
|
||||
}
|
||||
|
||||
// Notification 表示已发送或尝试发送的通知记录。
|
||||
type Notification struct {
|
||||
gorm.Model
|
||||
|
||||
// NotifierType 通知器类型 (例如:"邮件", "企业微信", "飞书", "日志")
|
||||
NotifierType notify.NotifierType `gorm:"type:varchar(20);not null;index" json:"notifier_type"`
|
||||
// UserID 接收通知的用户ID,用于追溯通知记录到特定用户
|
||||
UserID uint `gorm:"index" json:"user_id"` // 增加 UserID 字段,并添加索引
|
||||
// Title 通知标题
|
||||
Title string `gorm:"type:varchar(255);not null" json:"title"`
|
||||
// Message 通知内容
|
||||
Message string `gorm:"type:text;not null" json:"message"`
|
||||
// Level 通知级别 (例如:INFO, WARN, ERROR)
|
||||
Level LogLevel `gorm:"type:varchar(10);not null" json:"level"`
|
||||
// AlarmTimestamp 通知内容生成时的时间戳,与 ID 构成复合主键
|
||||
AlarmTimestamp time.Time `gorm:"primaryKey;not null" json:"alarm_timestamp"`
|
||||
// ToAddress 接收地址 (例如:邮箱地址, 企业微信ID, 日志标识符)
|
||||
ToAddress string `gorm:"type:varchar(255);not null" json:"to_address"`
|
||||
// Status 通知发送尝试的状态 (例如:"待发送", "发送成功", "发送失败", "已跳过")
|
||||
Status NotificationStatus `gorm:"type:varchar(20);not null;default:'待发送'" json:"status"`
|
||||
// ErrorMessage 如果通知发送失败,此字段存储错误信息
|
||||
ErrorMessage string `gorm:"type:text" json:"error_message"`
|
||||
}
|
||||
|
||||
// TableName 指定 Notification 模型的表名。
|
||||
func (Notification) TableName() string {
|
||||
return "notifications"
|
||||
}
|
||||
@@ -23,6 +23,9 @@ type DeviceRepository interface {
|
||||
// ListAll 获取所有设备的列表
|
||||
ListAll() ([]*models.Device, error)
|
||||
|
||||
// ListAllSensors 获取所有传感器类型的设备列表
|
||||
ListAllSensors() ([]*models.Device, error)
|
||||
|
||||
// ListByAreaControllerID 根据区域主控 ID 列出所有子设备。
|
||||
ListByAreaControllerID(areaControllerID uint) ([]*models.Device, error)
|
||||
|
||||
@@ -84,6 +87,19 @@ func (r *gormDeviceRepository) ListAll() ([]*models.Device, error) {
|
||||
return devices, nil
|
||||
}
|
||||
|
||||
// ListAllSensors 检索归类为传感器的所有设备
|
||||
func (r *gormDeviceRepository) ListAllSensors() ([]*models.Device, error) {
|
||||
var sensors []*models.Device
|
||||
err := r.db.Preload("AreaController").Preload("DeviceTemplate").
|
||||
Joins("JOIN device_templates ON device_templates.id = devices.device_template_id").
|
||||
Where("device_templates.category = ?", models.CategorySensor).
|
||||
Find(&sensors).Error
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询所有传感器失败: %w", err)
|
||||
}
|
||||
return sensors, nil
|
||||
}
|
||||
|
||||
// ListByAreaControllerID 根据区域主控 ID 列出所有子设备
|
||||
func (r *gormDeviceRepository) ListByAreaControllerID(areaControllerID uint) ([]*models.Device, error) {
|
||||
var devices []*models.Device
|
||||
|
||||
111
internal/infra/repository/notification_repository.go
Normal file
111
internal/infra/repository/notification_repository.go
Normal file
@@ -0,0 +1,111 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/notify"
|
||||
"go.uber.org/zap/zapcore"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// NotificationListOptions 定义了查询通知列表时的可选参数
|
||||
type NotificationListOptions struct {
|
||||
UserID *uint // 按用户ID过滤
|
||||
NotifierType *notify.NotifierType // 按通知器类型过滤
|
||||
Status *models.NotificationStatus // 按通知状态过滤 (例如:"success", "failed")
|
||||
Level *zapcore.Level // 按通知等级过滤 (例如:"info", "warning", "error")
|
||||
StartTime *time.Time // 通知内容生成时间范围 - 开始时间 (对应 AlarmTimestamp)
|
||||
EndTime *time.Time // 通知内容生成时间范围 - 结束时间 (对应 AlarmTimestamp)
|
||||
OrderBy string // 排序字段,例如 "alarm_timestamp DESC"
|
||||
}
|
||||
|
||||
// NotificationRepository 定义了与通知记录相关的数据库操作接口。
|
||||
type NotificationRepository interface {
|
||||
// Create 将一条新的通知记录插入数据库。
|
||||
Create(notification *models.Notification) error
|
||||
// CreateInTx 在给定的事务中插入一条新的通知记录。
|
||||
CreateInTx(tx *gorm.DB, notification *models.Notification) error
|
||||
// BatchCreate 批量插入多条通知记录。
|
||||
BatchCreate(notifications []*models.Notification) error
|
||||
// List 支持分页和过滤的通知列表查询。
|
||||
// 返回通知列表、总记录数和错误。
|
||||
List(opts NotificationListOptions, page, pageSize int) ([]models.Notification, int64, error)
|
||||
}
|
||||
|
||||
// gormNotificationRepository 是 NotificationRepository 的 GORM 实现。
|
||||
type gormNotificationRepository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
// NewGormNotificationRepository 创建一个新的 NotificationRepository GORM 实现实例。
|
||||
func NewGormNotificationRepository(db *gorm.DB) NotificationRepository {
|
||||
return &gormNotificationRepository{db: db}
|
||||
}
|
||||
|
||||
// Create 将一条新的通知记录插入数据库。
|
||||
func (r *gormNotificationRepository) Create(notification *models.Notification) error {
|
||||
return r.db.Create(notification).Error
|
||||
}
|
||||
|
||||
// CreateInTx 在给定的事务中插入一条新的通知记录。
|
||||
func (r *gormNotificationRepository) CreateInTx(tx *gorm.DB, notification *models.Notification) error {
|
||||
return tx.Create(notification).Error
|
||||
}
|
||||
|
||||
// BatchCreate 批量插入多条通知记录。
|
||||
func (r *gormNotificationRepository) BatchCreate(notifications []*models.Notification) error {
|
||||
// GORM 的 Create 方法在传入切片时会自动进行批量插入
|
||||
return r.db.Create(¬ifications).Error
|
||||
}
|
||||
|
||||
// List 实现了分页和过滤查询通知记录的功能
|
||||
func (r *gormNotificationRepository) List(opts NotificationListOptions, page, pageSize int) ([]models.Notification, int64, error) {
|
||||
// --- 校验分页参数 ---
|
||||
if page <= 0 || pageSize <= 0 {
|
||||
return nil, 0, ErrInvalidPagination // 复用已定义的错误
|
||||
}
|
||||
|
||||
var results []models.Notification
|
||||
var total int64
|
||||
|
||||
query := r.db.Model(&models.Notification{})
|
||||
|
||||
// --- 应用过滤条件 ---
|
||||
if opts.UserID != nil {
|
||||
query = query.Where("user_id = ?", *opts.UserID)
|
||||
}
|
||||
if opts.NotifierType != nil {
|
||||
query = query.Where("notifier_type = ?", *opts.NotifierType)
|
||||
}
|
||||
if opts.Status != nil {
|
||||
query = query.Where("status = ?", *opts.Status)
|
||||
}
|
||||
if opts.Level != nil {
|
||||
query = query.Where("level = ?", opts.Level.String())
|
||||
}
|
||||
if opts.StartTime != nil {
|
||||
query = query.Where("alarm_timestamp >= ?", *opts.StartTime)
|
||||
}
|
||||
if opts.EndTime != nil {
|
||||
query = query.Where("alarm_timestamp <= ?", *opts.EndTime)
|
||||
}
|
||||
|
||||
// --- 计算总数 ---
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
// --- 应用排序条件 ---
|
||||
orderBy := "alarm_timestamp DESC" // 默认按时间倒序
|
||||
if opts.OrderBy != "" {
|
||||
orderBy = opts.OrderBy
|
||||
}
|
||||
query = query.Order(orderBy)
|
||||
|
||||
// --- 分页 ---
|
||||
offset := (page - 1) * pageSize
|
||||
err := query.Limit(pageSize).Offset(offset).Find(&results).Error
|
||||
|
||||
return results, total, err
|
||||
}
|
||||
6
internal/infra/repository/repository.go
Normal file
6
internal/infra/repository/repository.go
Normal file
@@ -0,0 +1,6 @@
|
||||
package repository
|
||||
|
||||
import "errors"
|
||||
|
||||
// ErrInvalidPagination 表示分页参数无效
|
||||
var ErrInvalidPagination = errors.New("无效的分页参数:page和pageSize必须为大于0")
|
||||
@@ -1,16 +1,12 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// ErrInvalidPagination 表示分页参数无效
|
||||
var ErrInvalidPagination = errors.New("无效的分页参数:page和pageSize必须为大于0")
|
||||
|
||||
// SensorDataListOptions 定义了查询传感器数据列表时的可选参数
|
||||
type SensorDataListOptions struct {
|
||||
DeviceID *uint
|
||||
|
||||
Reference in New Issue
Block a user