You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
4.3 KiB
100 lines
4.3 KiB
package gojobs
|
|
|
|
import (
|
|
"context"
|
|
"go.dtapp.net/dorm"
|
|
"go.dtapp.net/gojobs/jobs_gorm_model"
|
|
"go.dtapp.net/golog"
|
|
"go.dtapp.net/gotime"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
// TaskLog 任务日志模型
|
|
type TaskLog struct {
|
|
LogId primitive.ObjectID `json:"log_id,omitempty" bson:"_id,omitempty"` //【记录】编号
|
|
LogTime primitive.DateTime `json:"log_time,omitempty" bson:"log_time"` //【记录】时间
|
|
Task struct {
|
|
Id uint `json:"id,omitempty" bson:"id,omitempty"` //【任务】编号
|
|
RunId string `json:"run_id,omitempty" bson:"run_id,omitempty"` //【任务】执行编号
|
|
ResultCode int `json:"result_code,omitempty" bson:"result_code,omitempty"` //【任务】执行状态码
|
|
ResultDesc string `json:"result_desc,omitempty" bson:"result_desc,omitempty"` //【任务】执行结果
|
|
ResultTime dorm.BsonTime `json:"result_time,omitempty" bson:"result_time,omitempty"` //【任务】执行时间
|
|
} `json:"task,omitempty" bson:"task,omitempty"` //【任务】信息
|
|
System struct {
|
|
HostName string `json:"host_name,omitempty" bson:"host_name,omitempty"` //【系统】主机名
|
|
InsideIp string `json:"inside_ip,omitempty" bson:"inside_ip,omitempty"` //【系统】内网ip
|
|
OutsideIp string `json:"outside_ip,omitempty" bson:"outside_ip,omitempty"` //【系统】外网ip
|
|
Os string `json:"os,omitempty" bson:"os,omitempty"` //【系统】系统类型
|
|
Arch string `json:"arch,omitempty" bson:"arch,omitempty"` //【系统】系统架构
|
|
} `json:"system,omitempty" bson:"system,omitempty"` //【系统】信息
|
|
Version struct {
|
|
Go string `json:"go,omitempty" bson:"go,omitempty"` //【程序】Go版本
|
|
Sdk string `json:"sdk,omitempty" bson:"sdk,omitempty"` //【程序】Sdk版本
|
|
} `json:"version,omitempty" bson:"version,omitempty"` //【程序】版本信息
|
|
}
|
|
|
|
func (TaskLog) CollectionName() string {
|
|
return "task_log"
|
|
}
|
|
|
|
// 创建时间序列集合
|
|
func (TaskLog) createCollection(ctx context.Context, zapLog *golog.ZapLog, db *dorm.MongoClient, databaseName string) {
|
|
err := db.Database(databaseName).CreateCollection(ctx, TaskLog{}.CollectionName(), options.CreateCollection().SetTimeSeriesOptions(options.TimeSeries().SetTimeField("log_time")))
|
|
if err != nil {
|
|
zapLog.WithTraceId(ctx).Sugar().Errorf("创建时间序列集合：%s", err)
|
|
}
|
|
}
|
|
|
|
// 创建索引
|
|
func (TaskLog) createIndexes(ctx context.Context, zapLog *golog.ZapLog, db *dorm.MongoClient, databaseName string) {
|
|
_, err := db.Database(databaseName).Collection(TaskLog{}.CollectionName()).CreateManyIndexes(ctx, []mongo.IndexModel{{
|
|
Keys: bson.D{{
|
|
Key: "log_time",
|
|
Value: -1,
|
|
}},
|
|
}})
|
|
if err != nil {
|
|
zapLog.WithTraceId(ctx).Sugar().Errorf("创建索引：%s", err)
|
|
}
|
|
}
|
|
|
|
// MongoTaskLogRecord 记录
|
|
func (c *Client) MongoTaskLogRecord(ctx context.Context, task jobs_gorm_model.Task, runId string, taskResultCode int, taskResultDesc string) {
|
|
|
|
taskLog := TaskLog{
|
|
LogId: primitive.NewObjectID(),
|
|
LogTime: primitive.NewDateTimeFromTime(gotime.Current().Time),
|
|
}
|
|
|
|
taskLog.Task.Id = task.Id
|
|
taskLog.Task.RunId = runId
|
|
taskLog.Task.ResultCode = taskResultCode
|
|
taskLog.Task.ResultDesc = taskResultDesc
|
|
taskLog.Task.ResultTime = dorm.NewBsonTimeCurrent()
|
|
|
|
taskLog.System.HostName = c.config.systemHostName
|
|
taskLog.System.InsideIp = c.config.systemInsideIp
|
|
taskLog.System.OutsideIp = c.config.systemOutsideIp
|
|
taskLog.System.Os = c.config.systemOs
|
|
taskLog.System.Arch = c.config.systemArch
|
|
|
|
taskLog.Version.Go = c.config.goVersion
|
|
taskLog.Version.Sdk = c.config.sdkVersion
|
|
|
|
_, err := c.mongoClient.Database(c.mongoConfig.databaseName).Collection(TaskLog{}.CollectionName()).InsertOne(ctx, taskLog)
|
|
if err != nil {
|
|
c.zapLog.WithTraceId(ctx).Sugar().Errorf("记录失败：%s", err)
|
|
c.zapLog.WithTraceId(ctx).Sugar().Errorf("记录数据：%+v", taskLog)
|
|
}
|
|
|
|
}
|
|
|
|
// MongoTaskLogDelete 删除
|
|
func (c *Client) MongoTaskLogDelete(ctx context.Context, hour int64) (*mongo.DeleteResult, error) {
|
|
filter := bson.D{{"log_time", bson.D{{"$lt", primitive.NewDateTimeFromTime(gotime.Current().BeforeHour(hour).Time)}}}}
|
|
return c.mongoClient.Database(c.mongoConfig.databaseName).Collection(TaskLog{}.CollectionName()).DeleteMany(ctx, filter)
|
|
}
|