fusenapi/server/info/internal/logic/infologic_test.go

package logic

import (
	"encoding/json"
	"fmt"
	"fusenapi/initalize"
	"fusenapi/utils/check"
	"strings"
	"testing"

	"github.com/zeromicro/go-zero/core/logx"
	"gorm.io/gorm"
)

func TestMain(t *testing.T) {

	conn := initalize.InitMysql("fsreaderwriter:XErSYmLELKMnf3Dh@tcp(fusen.cdmigcvz3rle.us-east-2.rds.amazonaws.com:3306)/fusen")
	condUser := "user_id = 105"

	var mquerys map[string]*ModuleQuery = make(map[string]*ModuleQuery)
	var metadict map[string]any = make(map[string]any)

	for _, module := range []string{"material.logo.logo_url", "userinfo.profile"} {
		if !check.CheckModuleQuery(module) {

		}

		mlist := strings.Split(module, ".")
		if len(mlist) < 2 {

		}

		mtable := mlist[0]
		tname, ok := ModuleTable[mtable]
		if !ok {

		}

		moduleName := mlist[1]
		cond := fmt.Sprintf("module = '%s' and %s", moduleName, condUser)

		if mquery, ok := mquerys[mtable]; ok {
			mquery.ModuleQuery[strings.Join(mlist[2:], ",")] = struct{}{}
		} else {
			mquery := &ModuleQuery{
				TableName:   tname,
				Cond:        cond,
				ModuleName:  mtable + "." + moduleName,
				ModuleQuery: map[string]struct{}{strings.Join(mlist[2:], ","): {}}}
			mquerys[mtable] = mquery
		}
	}

	for _, mquery := range mquerys {

		sqlstr := fmt.Sprintf("select id, module, %s as querydata from %s where %s order by ctime asc limit 1", mquery.EncodeQuery("metadata"), mquery.TableName, mquery.Cond)

		raw := conn.Raw(sqlstr)

		if raw.Error != nil {
			if raw.Error == gorm.ErrRecordNotFound {
				continue
			} else {
				logx.Error(raw.Error)
			}
		}

		var info map[string]any = make(map[string]any)
		err := raw.Scan(&info).Error
		if err == gorm.ErrRecordNotFound {
			continue
		}

		if err != nil {
			logx.Error(err, mquery.EncodeQuery("metadata"))

		}

		logx.Error(info)

		queryjson, ok := info["querydata"].(string)
		if !ok {
			return
		}

		var querydata map[string]any = make(map[string]any)
		err = json.Unmarshal([]byte(queryjson), &querydata)
		if err != nil {
			logx.Error(info)

		}

		for k, v := range querydata {
			metadict[k] = v
		}
	}

}