muffin/goMuffin
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: Getting data length logic?

Browse source
This commit is contained in:
Siwoo Jeon 2025-04-05 18:38:46 +09:00
parent a35ba84efc
commit 92f875f3c5
Signed by: migan
GPG key ID: 036E9A8C5E8E48DA
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
commands/dataLength.go
View file

@ -4,6 +4,7 @@ import (
"context" "context"
"fmt" "fmt"
"strconv" "strconv"
"sync"
"git.wh64.net/muffin/goMuffin/configs" "git.wh64.net/muffin/goMuffin/configs"
"git.wh64.net/muffin/goMuffin/databases" "git.wh64.net/muffin/goMuffin/databases"
@ -28,7 +29,8 @@ const (
userLearn userLearn
) )
var ch chan chStruct = make(chan chStruct) var dataLengthCh chan chStruct = make(chan chStruct)
var dataLengthWg sync.WaitGroup
var DataLengthCommand *Command = &Command{ var DataLengthCommand *Command = &Command{
ApplicationCommand: &discordgo.ApplicationCommand{ ApplicationCommand: &discordgo.ApplicationCommand{
@ -50,6 +52,7 @@ var DataLengthCommand *Command = &Command{
} }
func getLength(data dataType, coll *mongo.Collection, filter bson.D) { func getLength(data dataType, coll *mongo.Collection, filter bson.D) {
defer dataLengthWg.Done()
var err error var err error
var cur *mongo.Cursor var cur *mongo.Cursor
var datas []bson.M var datas []bson.M
@ -62,7 +65,7 @@ func getLength(data dataType, coll *mongo.Collection, filter bson.D) {
defer cur.Close(context.TODO()) defer cur.Close(context.TODO())
cur.All(context.TODO(), &datas) cur.All(context.TODO(), &datas)
ch <- chStruct{name: data, length: len(datas)} dataLengthCh <- chStruct{name: data, length: len(datas)}
} }
func dataLengthRun(s *discordgo.Session, m any) { func dataLengthRun(s *discordgo.Session, m any) {
@ -85,6 +88,7 @@ func dataLengthRun(s *discordgo.Session, m any) {
channelId = m.ChannelID channelId = m.ChannelID
} }
dataLengthWg.Add(5)
go getLength(text, databases.Texts, bson.D{{}}) go getLength(text, databases.Texts, bson.D{{}})
go getLength(muffin, databases.Texts, bson.D{{Key: "persona", Value: "muffin"}}) go getLength(muffin, databases.Texts, bson.D{{Key: "persona", Value: "muffin"}})
go getLength(nsfw, databases.Texts, bson.D{ go getLength(nsfw, databases.Texts, bson.D{
@ -98,8 +102,12 @@ func dataLengthRun(s *discordgo.Session, m any) {
go getLength(learn, databases.Learns, bson.D{{}}) go getLength(learn, databases.Learns, bson.D{{}})
go getLength(userLearn, databases.Learns, bson.D{{Key: "user_id", Value: userId}}) go getLength(userLearn, databases.Learns, bson.D{{Key: "user_id", Value: userId}})
for range 5 { go func() {
resp := <-ch dataLengthWg.Wait()
close(dataLengthCh)
}()
for resp := range dataLengthCh {
switch dataType(resp.name) { switch dataType(resp.name) {
case text: case text:
textLength = resp.length textLength = resp.length
@ -113,7 +121,6 @@ func dataLengthRun(s *discordgo.Session, m any) {
userLearnLength = resp.length userLearnLength = resp.length
} }
} }
close(ch)
sum := textLength + learnLength sum := textLength + learnLength