You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package model
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
type ServerItem struct {
|
|
ID int64 `json:"id" bson:"_id"`
|
|
ServerId int64 `json:"server_id" bson:"server_id"`
|
|
Status int64 `json:"status" bson:"status"`
|
|
ActivityId int64 `json:"activity_id" bson:"activity_id"`
|
|
ActivityEndTime string `json:"activity_end_time" bson:"activity_end_time"`
|
|
CreatedAt string `json:"created_at" bson:"created_at"`
|
|
}
|
|
|
|
type RespServerItem struct {
|
|
ID int64 `json:"id"`
|
|
ServerId int64 `json:"server_name"`
|
|
Status int64 `json:"server_id"`
|
|
UserCount int64 `json:"user_count"`
|
|
OnlineCount int64 `json:"online_count"`
|
|
ActivityId int64 `json:"activity_id"`
|
|
ActivityEndTime string `json:"activity_end_time"`
|
|
CreatedAt string `json:"created_at"`
|
|
}
|
|
|
|
func GetServerItemList(collection *mongo.Collection, server_id int64, offset int64, limit int64) ([]ServerItem, int64, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
filter := bson.D{{Key: "server_id", Value: server_id}}
|
|
opts := options.Find().SetLimit(limit).SetSkip(offset).SetSort(bson.D{{Key: "_id", Value: 1}})
|
|
cur, err := collection.Find(ctx, filter, opts)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
defer cur.Close(ctx)
|
|
list := []ServerItem{}
|
|
err = cur.All(context.Background(), &list)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
count, err := collection.CountDocuments(ctx, filter)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return list, count, nil
|
|
}
|
|
|
|
func GetServerItemCount(collection *mongo.Collection, server_id int64) (int64, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
filter := bson.D{{Key: "server_id", Value: server_id}}
|
|
count, err := collection.CountDocuments(ctx, filter)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func UpdateServer(collection *mongo.Collection) error {
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
filter := bson.D{{Key: "status", Value: 1}}
|
|
update := bson.D{{Key: "$set", Value: bson.D{{Key: "status", Value: 2}}}}
|
|
_, err := collection.UpdateMany(ctx, filter, update)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func AddServer(collection *mongo.Collection, docs []interface{}) error {
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
_, err := collection.InsertMany(ctx, docs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|