milvus/internal/metastore/model/resource.go

package model

import (
	"github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
	pb "github.com/milvus-io/milvus/pkg/v2/proto/datapb"
)

type FileResource struct {
	ID   int64
	Name string
	Path string
	Type commonpb.FileResourceType
}

func (resource *FileResource) Marshal() *pb.FileResourceInfo {
	if resource == nil {
		return nil
	}

	return &pb.FileResourceInfo{
		ResourceId: resource.ID,
		Name:       resource.Name,
		Path:       resource.Path,
		Type:       resource.Type,
	}
}

func UnmarshalFileResourceInfo(resource *pb.FileResourceInfo) *FileResource {
	if resource == nil {
		return nil
	}

	return &FileResource{
		ID:   resource.ResourceId,
		Name: resource.Name,
		Path: resource.Path,
		Type: resource.Type,
	}
}