zbpdata/main.go

//go:build ignore
// +build ignore

package main

import (
	"crypto/md5"
	"encoding/hex"
	"fmt"
	"io/fs"
	"os"
	"strings"
	"sync"
	"unicode"
	"unsafe"

	"github.com/FloatTech/zbputils/process"
	"github.com/fumiama/go-registry"
	"github.com/wdvxdr1123/ZeroBot/utils/helper"
)

func main() {
	var files []string
	fs.WalkDir(os.DirFS("./"), ".", func(path string, d fs.DirEntry, err error) error {
		if err != nil {
			return err
		}
		if !d.IsDir() && strings.Contains(path, "/") && unicode.IsUpper([]rune(path)[0]) {
			files = append(files, path)
		}
		return nil
	})
	fmt.Println(files)
	md5s := make([]string, len(files))
	for i, fn := range files {
		data, err := os.ReadFile(fn)
		if err != nil {
			panic(err)
		}
		buf := md5.Sum(data)
		*(*unsafe.Pointer)(unsafe.Pointer(&md5s[i])) = unsafe.Pointer(&buf)
		*(*uintptr)(unsafe.Add(unsafe.Pointer(&md5s[i]), unsafe.Sizeof(uintptr(0)))) = uintptr(16)
	}
	r := registry.NewRegedit("reilia.westeurope.cloudapp.azure.com:32664", "fumiama", os.Getenv("REILIA_SPS"))
	err := r.Connect()
	if err != nil {
		panic(err)
	}
	defer r.Close()
	for i := 0; i < 1024; i++ {
		err = r.Set("__setlock__", "fill")
		if err == nil {
			break
		}
		fmt.Println("accqiring set lock, retry times:", i)
	}
	var wg sync.WaitGroup
	wg.Add(len(files))
	for i, fn := range files {
		go func(i int, fn string) {
			defer wg.Done()
			for c := 0; c < 5; c++ {
				err = r.Set("data/"+fn, md5s[i])
				fmt.Println("set", "data/"+fn, "=", hex.EncodeToString(helper.StringToBytes(md5s[i])))
				if err == nil {
					break
				}
				if c >= 4 {
					panic("ERROR:" + err.Error() + "max retry times exceeded")
				} else {
					fmt.Println("ERROR:", err, ", retry times:", c)
				}
				process.SleepAbout1sTo2s()
			}
		}(i, fn)
	}
	wg.Wait()
}