tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2024-09-20 08:15:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: use range loops introduced in go 1.22

Browse source
This commit is contained in:
divyam234 2024-05-20 02:54:32 +05:30
parent 7bbcbf80a2
commit c8148a2207
3 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cmd/run.go
View file

@ -163,7 +163,7 @@ func bindFlagsRecursive(flags *pflag.FlagSet, prefix string, v reflect.Value) {
if t.Kind() == reflect.Ptr { if t.Kind() == reflect.Ptr {
t = t.Elem() t = t.Elem()
} }
for i := 0; i < t.NumField(); i++ { for i := range t.NumField() {
field := t.Field(i) field := t.Field(i)
switch field.Type.Kind() { switch field.Type.Kind() {
case reflect.Struct: case reflect.Struct:

2
internal/crypt/cipher.go
View file

@ -143,7 +143,7 @@ func (n *nonce) increment() {
func (n *nonce) add(x uint64) { func (n *nonce) add(x uint64) {
carry := uint16(0) carry := uint16(0)
for i := 0; i < 8; i++ { for i := range 8 {
digit := (*n)[i] digit := (*n)[i]
xDigit := byte(x) xDigit := byte(x)
x >>= 8 x >>= 8

16
pkg/services/common.go
View file

@ -124,7 +124,7 @@ func getBotInfo(ctx context.Context, KV kv.KV, config *config.TGConfig, token st
return &types.BotInfo{Id: user.ID, UserName: user.Username, Token: token}, nil return &types.BotInfo{Id: user.ID, UserName: user.Username, Token: token}, nil
} }
func getTGMessagesBatch(ctx context.Context, client *telegram.Client, channel *tg.InputChannel, parts []schemas.Part, userID string, index int, func getTGMessagesBatch(ctx context.Context, client *telegram.Client, channel *tg.InputChannel, parts []schemas.Part, index int,
results chan<- batchResult, errors chan<- error, wg *sync.WaitGroup) { results chan<- batchResult, errors chan<- error, wg *sync.WaitGroup) {
defer wg.Done() defer wg.Done()
@ -172,10 +172,10 @@ func getTGMessages(ctx context.Context, client *telegram.Client, parts []schemas
errors := make(chan error, batchCount) errors := make(chan error, batchCount)
for i := 0; i < batchCount; i++ { for i := range batchCount {
wg.Add(1) wg.Add(1)
splitParts := parts[i*batchSize : min((i+1)*batchSize, len(parts))] splitParts := parts[i*batchSize : min((i+1)*batchSize, len(parts))]
go getTGMessagesBatch(ctx, client, channel, splitParts, userID, i, results, errors, &wg) go getTGMessagesBatch(ctx, client, channel, splitParts, i, results, errors, &wg)
} }
wg.Wait() wg.Wait()
@ -188,19 +188,19 @@ func getTGMessages(ctx context.Context, client *telegram.Client, parts []schemas
} }
} }
channelResult := []batchResult{} batchResults := []batchResult{}
for result := range results { for result := range results {
channelResult = append(channelResult, result) batchResults = append(batchResults, result)
} }
sort.Slice(channelResult, func(i, j int) bool { sort.Slice(batchResults, func(i, j int) bool {
return channelResult[i].Index < channelResult[j].Index return batchResults[i].Index < batchResults[j].Index
}) })
allMessages := []tg.MessageClass{} allMessages := []tg.MessageClass{}
for _, result := range channelResult { for _, result := range batchResults {
allMessages = append(allMessages, result.Messages.GetMessages()...) allMessages = append(allMessages, result.Messages.GetMessages()...)
} }