Muat Turun Fail Selari dalam Go Menggunakan Goroutines
Dalam Go, anda boleh memuat turun dan menyimpan fail secara serentak menggunakan goroutine. Goroutines membenarkan pelaksanaan benang yang ringan, memudahkan pemprosesan selari dan meningkatkan prestasi.
Berikut ialah kod yang anda berikan, diubah suai untuk menggunakan 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>
Dalam kod yang diubah suai ini, kami menambah penyegerakan. WaitGroup memanggil wg untuk menjejaki bilangan goroutine yang dilaksanakan. Kami menambah wg sebelum memulakan setiap goroutine dan mengurangkannya apabila setiap goroutine selesai menggunakan wg.Done(). Goroutine utama menunggu semua goroutine selesai dengan memanggil wg.Wait(). Ini memastikan program tidak keluar sebelum semua fail dimuat turun.
Atas ialah kandungan terperinci Bagaimanakah saya boleh menggunakan Goroutines dalam Go untuk memuat turun berbilang fail secara serentak daripada Dropbox?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!