mirror of https://github.com/MedzikUser/go-cdn
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"gitlab.com/gaming0skar123/go/cdn/common"
|
|
"gitlab.com/gaming0skar123/go/cdn/config"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
type Channel struct {
|
|
ID string `bson:"_id"`
|
|
GuildID string `bson:"gid"`
|
|
}
|
|
|
|
type ChannelAPI struct {
|
|
ID string `bson:"_id"`
|
|
}
|
|
|
|
func InsertChannel(channel *Channel) {
|
|
collection := DB.Collection(config.Mongo_Collection_Channel)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
_, err := collection.InsertOne(ctx, channel)
|
|
common.CheckErr(err, "inserting to DB")
|
|
}
|
|
|
|
func InsertAPIChannel(channel *ChannelAPI) {
|
|
collection := DB.Collection(config.Mongo_Collection_Channel_API)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
_, err := collection.InsertOne(ctx, channel)
|
|
common.CheckErr(err, "inserting to DB")
|
|
}
|
|
|
|
func FindChannel(channel *Channel) bool {
|
|
collection := DB.Collection(config.Mongo_Collection_Channel)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
err := collection.FindOne(ctx, channel).Decode(&channel)
|
|
return err == nil
|
|
}
|
|
|
|
func FindAPIChannel() (*ChannelAPI, error) {
|
|
collection := DB.Collection(config.Mongo_Collection_Channel_API)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
var channel *ChannelAPI
|
|
|
|
err := collection.FindOne(ctx, bson.D{}).Decode(&channel)
|
|
return channel, err
|
|
}
|