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&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", &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, &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!