Maison  >  Article  >  développement back-end  >  Comment puis-je utiliser Goroutines dans Go pour télécharger plusieurs fichiers simultanément depuis Dropbox ?

Comment puis-je utiliser Goroutines dans Go pour télécharger plusieurs fichiers simultanément depuis Dropbox ?

Susan Sarandon
Susan Sarandonoriginal
2024-10-27 19:46:31637parcourir

How can I utilize Goroutines in Go to download multiple files concurrently from Dropbox?

Téléchargements de fichiers parallèles dans Go à l'aide de Goroutines

Dans Go, il est possible de télécharger et d'enregistrer des fichiers simultanément à l'aide de goroutines. Les goroutines permettent une exécution légère des threads, facilitant le traitement parallèle et améliorant les performances.

Voici le code que vous avez fourni, modifié pour utiliser les goroutines :

<code class="go">package main

import (
    "encoding/json"
    "fmt"
    "io"
    "io/ioutil"
    "net/http"
    "net/url"
    "os"
    "path/filepath"
    "sync"
)

// Credentials
const app_key string = "<app_key>"
const app_secret string = "<app_secret>"

// Authorization code
var code string

// Token response
type TokenResponse struct {
    AccessToken string `json:"access_token"`
}

// File structure
type File struct {
    Path string
}

// File list response
type FileListResponse struct {
    FileList []File `json:"contents"`
}

// Download a file using goroutines
func download_file(file File, token TokenResponse, wg *sync.WaitGroup) {
    download_file := fmt.Sprintf("https://api-content.dropbox.com/1/files/dropbox/%s?access_token=%s", file.Path, token.AccessToken)

    resp, _ := http.Get(download_file)
    defer resp.Body.Close()

    filename := filepath.Base(file.Path)
    out, err := os.Create(filename)
    if err != nil {
        panic(err)
    }
    defer out.Close()

    io.Copy(out, resp.Body)
    wg.Done()
}

func main() {
    // Authorization and token retrieval
    authorize_url := fmt.Sprintf("https://www.dropbox.com/1/oauth2/authorize?response_type=code&amp;client_id=%s", app_key)
    fmt.Printf("1. Go to: %s\n", authorize_url)
    fmt.Println("2. Click 'Allow' (you might have to log in first)")
    fmt.Println("3. Copy the authorization code.")
    fmt.Printf("Enter the authorization code here: ")
    fmt.Scanf("%s", &amp;code)

    // Get file list
    file_list_url := fmt.Sprintf("https://api.dropbox.com/1/metadata/dropbox/Camera Uploads?access_token=%s", tr.AccessToken)

    resp2, _ := http.Get(file_list_url)
    defer resp2.Body.Close()

    contents2, _ := ioutil.ReadAll(resp2.Body)

    var flr FileListResponse
    json.Unmarshal(contents2, &amp;flr)

    // WaitGroup to wait for all goroutines to finish
    var wg sync.WaitGroup

    // Download files concurrently
    for i, file := range flr.FileList {
        wg.Add(1)

        go download_file(file, tr, &wg)

        if i >= 2 {
            break
        }
    }
    wg.Wait()
}</code>

Dans ce code modifié, nous ajoutons une synchronisation. WaitGroup a appelé wg pour suivre le nombre de goroutines en cours d'exécution. Nous incrémentons wg avant de démarrer chaque goroutine et le décrémentons lorsque chaque goroutine termine d'utiliser wg.Done(). La goroutine principale attend que toutes les goroutines aient fini en appelant wg.Wait(). Cela garantit que le programme ne se ferme pas avant que tous les fichiers ne soient téléchargés.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration:
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn