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.
141 lines
6.3 KiB
141 lines
6.3 KiB
package gojobs
|
|
|
|
import (
|
|
"context"
|
|
"go.dtapp.net/gojobs/jobs_gorm_model"
|
|
"go.dtapp.net/gojobs/jobs_mongo_model"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"log"
|
|
)
|
|
|
|
// 创建模型
|
|
func (c *Client) autoMigrateTask() {
|
|
log.Println(c.db.gormClient.Db.AutoMigrate(&jobs_gorm_model.Task{}))
|
|
}
|
|
|
|
// 创建模型
|
|
func (c *Client) autoMigrateTaskIp() {
|
|
log.Println(c.db.gormClient.Db.AutoMigrate(&jobs_gorm_model.TaskIp{}))
|
|
}
|
|
|
|
// 创建模型
|
|
func (c *Client) autoMigrateTaskLog() {
|
|
log.Println(c.db.gormClient.Db.AutoMigrate(&jobs_gorm_model.TaskLog{}))
|
|
}
|
|
|
|
// 创建模型
|
|
func (c *Client) autoMigrateTaskLogRun() {
|
|
log.Println(c.db.gormClient.Db.AutoMigrate(&jobs_gorm_model.TaskLogRun{}))
|
|
}
|
|
|
|
// 创建时间序列集合
|
|
func (c *Client) mongoCreateCollectionTask(ctx context.Context) {
|
|
var commandResult bson.M
|
|
commandErr := c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).RunCommand(ctx, bson.D{{
|
|
"listCollections", 1,
|
|
}}).Decode(&commandResult)
|
|
if commandErr != nil {
|
|
log.Println("检查时间序列集合:", commandErr)
|
|
} else {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).CreateCollection(ctx, jobs_mongo_model.Task{}.TableName(), options.CreateCollection().SetTimeSeriesOptions(options.TimeSeries().SetTimeField("created_at"))))
|
|
}
|
|
}
|
|
|
|
// 创建索引
|
|
func (c *Client) mongoCreateIndexesTask(ctx context.Context) {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"status", 1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"frequency", 1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"custom_id", 1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"type", 1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"created_ip", 1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"specify_ip", 1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.Task{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"updated_ip", 1},
|
|
}}))
|
|
}
|
|
|
|
// 创建索引
|
|
func (c *Client) mongoCreateIndexesTaskIp(ctx context.Context) {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskIp{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"task_type", 1},
|
|
}}))
|
|
}
|
|
|
|
// 创建时间序列集合
|
|
func (c *Client) mongoCreateCollectionTaskLog(ctx context.Context) {
|
|
var commandResult bson.M
|
|
commandErr := c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).RunCommand(ctx, bson.D{{
|
|
"listCollections", 1,
|
|
}}).Decode(&commandResult)
|
|
if commandErr != nil {
|
|
log.Println("检查时间序列集合:", commandErr)
|
|
} else {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).CreateCollection(ctx, jobs_mongo_model.TaskLog{}.TableName(), options.CreateCollection().SetTimeSeriesOptions(options.TimeSeries().SetTimeField("created_at"))))
|
|
}
|
|
}
|
|
|
|
// 创建索引
|
|
func (c *Client) mongoCreateIndexesTaskLog(ctx context.Context) {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"task_id", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"status_code", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"version", -1},
|
|
}}))
|
|
}
|
|
|
|
// 创建时间序列集合
|
|
func (c *Client) mongoCreateCollectionTaskLogRun(ctx context.Context) {
|
|
var commandResult bson.M
|
|
commandErr := c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).RunCommand(ctx, bson.D{{
|
|
"listCollections", 1,
|
|
}}).Decode(&commandResult)
|
|
if commandErr != nil {
|
|
log.Println("检查时间序列集合:", commandErr)
|
|
} else {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).CreateCollection(ctx, jobs_mongo_model.TaskLogRun{}.TableName(), options.CreateCollection().SetTimeSeriesOptions(options.TimeSeries().SetTimeField("created_at"))))
|
|
}
|
|
}
|
|
|
|
// 创建索引
|
|
func (c *Client) mongoCreateIndexesTaskLogRun(ctx context.Context) {
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"task_id", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"run_id", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"outside_ip", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"inside_ip", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"os", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"arch", -1},
|
|
}}))
|
|
log.Println(c.db.mongoClient.Db.Database(c.db.mongoDatabaseName).Collection(jobs_mongo_model.TaskLog{}.TableName()).Indexes().CreateOne(ctx, mongo.IndexModel{Keys: bson.D{
|
|
{"go_version", -1},
|
|
}}))
|
|
}
|