xk6-frostfs/cmd/xk6-registry/importer/import.go

package importer

import (
	"encoding/json"
	"os"

	"git.frostfs.info/TrueCloudLab/xk6-frostfs/internal/registry"
)

type PreGenObj struct {
	Bucket    string `json:"bucket"`
	Object    string `json:"object"`
	Container string `json:"container"`
}

type PreGenerateInfo struct {
	Buckets    []string    `json:"buckets"`
	Containers []string    `json:"containers"`
	Objects    []PreGenObj `json:"objects"`
	ObjSize    string      `json:"obj_size"`
}

func ImportJSONPreGen(o *registry.ObjRegistry, filename string) error {
	f, err := os.ReadFile(filename)
	if err != nil {
		return err
	}

	var pregenInfo PreGenerateInfo
	err = json.Unmarshal(f, &pregenInfo)
	if err != nil {
		return err
	}

	for _, obj := range pregenInfo.Objects {
		if obj.Bucket != "" {
			err = o.AddObject("", "", obj.Bucket, obj.Object, "")
		} else {
			err = o.AddObject(obj.Container, obj.Object, "", "", "")
		}
		if err != nil {
			return err
		}
	}

	return nil
}