refactor: move the PullByDragonfly as private method (#204)

Signed-off-by: chlins <[email protected]>

Author: chlins

cmd/pull.go

@@ -73,14 +73,8 @@ func runPull(ctx context.Context, target string) error {
 		return fmt.Errorf("target is required")
 	}
 
-	if pullConfig.DragonflyEndpoint != "" {
-		if err := b.PullByDragonfly(ctx, target, pullConfig); err != nil {
-			return err
-		}
-	} else {
-		if err := b.Pull(ctx, target, pullConfig); err != nil {
-			return err
-		}
+	if err := b.Pull(ctx, target, pullConfig); err != nil {
+		return err
 	}
 
 	fmt.Printf("Successfully pulled model artifact: %s\n", target)

pkg/backend/backend.go

@@ -40,9 +40,6 @@ type Backend interface {
 	// Pull pulls an artifact from a registry.
 	Pull(ctx context.Context, target string, cfg *config.Pull) error
 
-	// PullByDragonfly pulls an artifact from a registry by Dragonfly.
-	PullByDragonfly(ctx context.Context, target string, cfg *config.Pull) error
-
 	// Fetch fetches partial files to the output.
 	Fetch(ctx context.Context, target string, cfg *config.Fetch) error
 

pkg/backend/pull.go

@@ -35,6 +35,11 @@ import (
 
 // Pull pulls an artifact from a registry.
 func (b *backend) Pull(ctx context.Context, target string, cfg *config.Pull) error {
+	// pullByDragonfly is called if a Dragonfly endpoint is specified in the configuration.
+	if cfg.DragonflyEndpoint != "" {
+		return b.pullByDragonfly(ctx, target, cfg)
+	}
+
 	// parse the repository and tag from the target.
 	ref, err := ParseReference(target)
 	if err != nil {

pkg/backend/pull_by_d7y.go

@@ -45,8 +45,8 @@ const (
 	mediaTypeTarSuffix = ".tar"
 )
 
-// PullByDragonfly pulls and hardlinks blobs from Dragonfly gRPC service for remote extraction.
-func (b *backend) PullByDragonfly(ctx context.Context, target string, cfg *config.Pull) error {
+// pullByDragonfly pulls and hardlinks blobs from Dragonfly gRPC service for remote extraction.
+func (b *backend) pullByDragonfly(ctx context.Context, target string, cfg *config.Pull) error {
 	// Parse reference and initialize remote client.
 	ref, err := ParseReference(target)
 	if err != nil {