侯体宗的博客
  • 首页
  • Hyperf版
  • beego仿版
  • 人生(杂谈)
  • 技术
  • 关于我
  • 更多分类
    • 文件下载
    • 文字修仙
    • 中国象棋ai
    • 群聊
    • 九宫格抽奖
    • 拼图
    • 消消乐
    • 相册

在go语言中实现Google Drive API 上传和重命名示例

Go  /  管理员 发布于 1年前   587

在go语言中如何使用 RESTful 方法将文件上传到 Google Drive,并使用 Drive API 重命名文件。

在开始之前,如果尚未开启 Google Drive API,请先开启,并按照 

https://developers.google.com/drive/v3/web/quickstart/go#prerequisites 

中的 "步骤 1:开启 Drive API "部分下载证书文件 - client_secret.json。

下载凭证文件后,在运行程序前将其移动到与下面源代码相同的目录。

注意:
您需要将要上传的文件 - img.png 改为其他文件。

第一次提示输入授权码时,将终端中的 URL 剪切并粘贴到浏览器中,

然后会得到一个字符串(令牌),将该字符串剪切并粘贴到执行程序的终端中。


如果在 API JSON 回复中出现授权错误,则需要在 

https://developers.google.com/oauthplayground/ 

授权 Drive API v3 范围。


接下来,执行 go get 命令

>go get google.golang.org/api/drive/v3


最后,访问 

https://developers.google.com/drive/v3/web/enable-sdk 

并按照下面列出的说明启用 Drive API 要启用 Drive API,请完成以下步骤:


运行下面的代码示例,并在 

https://drive.google.com/drive/my-drive 

上观察自己的 Google Drive 内容的变化。


示例代码:

package main
 import (
         "crypto/rand"
         "encoding/json"
         "fmt"
         "github.com/antonholmquist/jason"
         "golang.org/x/net/context"
         "golang.org/x/oauth2"
         "golang.org/x/oauth2/google"
         "google.golang.org/api/drive/v3"
         "io/ioutil"
         "log"
         "net/http"
         "net/url"
         "os"
         "os/user"
         "path/filepath"
         "strconv"
         "strings"
 )
 // 注意:我们不想访问 CSRF URL 来获取授权代码。
 // 并在每次发送电子邮件时将其粘贴到终端中。
 // 因此,我们将为客户端获取一个令牌,并将该令牌保存到一个文件中。
 // 只会提示您访问一次浏览器中的链接以获取授权码
 // 随后执行程序时将不会再提示您输入授权码
 // 直到令牌过期。
 // getClient 使用 Context 和 Config 获取令牌
 // 然后生成一个客户端。它会返回生成的客户端。
 func getClient(ctx context.Context, config *oauth2.Config) *http.Client {
         cacheFile, err := tokenCacheFile()
         if err != nil {
                 log.Fatalf("Unable to get path to cached credential file. %v", err)
         }
         tok, err := tokenFromFile(cacheFile)
         if err != nil {
                 tok = getTokenFromWeb(config)
                 saveToken(cacheFile, tok)
         }
         return config.Client(ctx, tok)
 }
 // getTokenFromWeb 使用配置请求令牌。
 // 返回检索到的令牌。
 func getTokenFromWeb(config *oauth2.Config) *oauth2.Token {
         authURL := config.AuthCodeURL("state-token", oauth2.AccessTypeOffline)
         fmt.Printf("在浏览器中转入以下链接 "+
                 "然后键入授权代码: \n%v\n", authURL)
         var code string
         if _, err := fmt.Scan(&code); err != nil {
                 log.Fatalf("Unable to read authorization code %v", err)
         }
         tok, err := config.Exchange(oauth2.NoContext, code)
         if err != nil {
                 log.Fatalf("Unable to retrieve token from web %v", err)
         }
         return tok
 }
 // tokenCacheFile 生成证书文件路径/文件名。
 // 返回生成的证书路径/文件名。
 func tokenCacheFile() (string, error) {
         usr, err := user.Current()
         if err != nil {
                 return "", err
         }
         tokenCacheDir := filepath.Join(usr.HomeDir, ".credentials")
         os.MkdirAll(tokenCacheDir, 0700)
         return filepath.Join(tokenCacheDir,
                 url.QueryEscape("google-drive-golang.json")), err
 }
 // tokenFromFile 从给定的文件路径中获取令牌。
 // 返回检索到的令牌和遇到的任何读取错误。
 func tokenFromFile(file string) (*oauth2.Token, error) {
         f, err := os.Open(file)
         if err != nil {
                 return nil, err
         }
         t := &oauth2.Token{}
         err = json.NewDecoder(f).Decode(t)
         defer f.Close()
         return t, err
 }
 // saveToken 使用文件路径创建文件,并在其中存储令牌。
 // 标记。
 func saveToken(file string, token *oauth2.Token) {
         fmt.Printf("保存证书文件到: %s\n", file)
         f, err := os.Create(file)
         if err != nil {
                 log.Fatalf("Unable to cache oauth token: %v", err)
         }
         defer f.Close()
         json.NewEncoder(f).Encode(token)
 }
 func randStr(strSize int, randType string) string {
         var dictionary string
         if randType == "alphanum" {
                 dictionary = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
         }
         if randType == "alpha" {
                 dictionary = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
         }
         if randType == "number" {
                 dictionary = "0123456789"
         }
         var bytes = make([]byte, strSize)
         rand.Read(bytes)
         for k, v := range bytes {
                 bytes[k] = dictionary[v%byte(len(dictionary))]
         }
         return string(bytes)
 }
 func main() {
         ctx := context.Background()
         // 处理证书文件
         credential, err := ioutil.ReadFile("client_secret.json")
         if err != nil {
                 log.Fatalf("Unable to read client secret file: %v", err)
         }
         // 为了使 POST 上传附件生效
         // 您需要授权 Gmail API v1 范围
         // 在 https://developers.google.com/oauthplayground/
         // 否则,您将在 API JSON 回复中收到授权错误信息。
         // 本示例使用 DriveScope。因为我们要在
         // Google Drive 中的文件。
         // 其他内容请参见 https://godoc.org/google.golang.org/api/drive/v3#pkg-constants
         config, err := google.ConfigFromJSON(credential, drive.DriveScope)
         if err != nil {
                 log.Fatalf("Unable to parse client secret file to config: %v", err)
         }
         client := getClient(ctx, config)
         // 启动新的 Google Drive 服务
         driveClientService, err := drive.New(client)
         if err != nil {
                 log.Fatalf("Unable to initiate new Drive client: %v", err)
         }
         // 获取我们的令牌
         cacheFile, err := tokenCacheFile()
         if err != nil {
                 log.Fatalf("Unable to get path to cached credential file. %v", err)
         }
         token, err := tokenFromFile(cacheFile)
         if err != nil {
                 log.Fatalf("Unable to get token from file. %v", err)
         }
         // 我们将使用多部分上传方法
         // 参见 https://developers.google.com/drive/v3/web/manage-uploads
         // 读取文件以便上传
         fileName := "img.png" // <------------ CHANGE HERE!
         fileBytes, err := ioutil.ReadFile(fileName)
         if err != nil {
                 log.Fatalf("Unable to read file for upload: %v", err)
         }
         fileMIMEType := http.DetectContentType(fileBytes)
         // 使用多部分上传方法,因为我们需要媒体及其元数据(文件名)
         // 简单上传方法将导致文件名被命名为 "无标题"。
         postURL := "https://www.googleapis.com/upload/drive/v3/files?uploadType=multipart"
         // 从令牌文件中提取认证或访问令牌
         // 参见 https://godoc.org/golang.org/x/oauth2#Token
         authToken := token.AccessToken
         boundary := randStr(32, "alphanum")
         uploadData := []byte("\n" +
                 "--" + boundary + "\n" +
                 "Content-Type: application/json; charset=" + string('"') + "UTF-8" + string('"') + "\n\n" +
                 "{ \n" +
                 string('"') + "name" + string('"') + ":" + string('"') + fileName + string('"') + "\n" +
                 "} \n\n" +
                 "--" + boundary + "\n" +
                 "Content-Type:" + fileMIMEType + "\n\n" +
                 string(fileBytes) + "\n" +
                 "--" + boundary + "--")
         // 使用 RESTful 方法发布到 Drive
         request, _ := http.NewRequest("POST", postURL, strings.NewReader(string(uploadData)))
         request.Header.Add("Host", "www.googleapis.com")
         request.Header.Add("Authorization", "Bearer "+authToken)
         request.Header.Add("Content-Type", "multipart/related; boundary="+string('"')+boundary+string('"'))
         request.Header.Add("Content-Length", strconv.FormatInt(request.ContentLength, 10))
         // debug
         //fmt.Println(request)
         response, err := client.Do(request)
         if err != nil {
                 log.Fatalf("Unable to be post to Google API: %v", err)
         }
         defer response.Body.Close()
         body, err := ioutil.ReadAll(response.Body)
         if err != nil {
                 log.Fatalf("Unable to read Google API response: %v", err)
         }
         // 输出驱动程序应用程序接口的响应
         fmt.Println(string(body))
         // 我们需要提取上传文件的 ID,以执行更新命令
         jsonAPIreply, _ := jason.NewObjectFromBytes(body)
         uploadedFileID, _ := jsonAPIreply.GetString("id")
         fmt.Println("上传文件 ID : ", uploadedFileID)
         // -------------------------------------------------------------------------------
         // 只是为了好玩,让我们使用 Google Drive API 重命名我们上传的文件
         // 参见 https://godoc.org/google.golang.org/api/drive/v3#File
         renamedFile := drive.File{Name: "renamed-" + fileName}
         _, err = driveClientService.Files.Update(uploadedFileID, &renamedFile).Do()
         if err != nil {
                 log.Fatalf("Unable to rename(update) uploaded file in Drive:  %v", err)
         } else {
                 fmt.Println("将 " + fileName + " 重命名为 " + renamedFile.Name)
         }
         // 列出上传的文件
         // see https://godoc.org/google.golang.org/api/drive/v3#FilesService.List
         filesListCall, err := driveClientService.Files.List().OrderBy("name").Do()
         if err != nil {
                 log.Fatalf("Unable to list files in Drive:  %v", err)
         }
         // 注:还将列出垃圾文件夹内的文件
         fmt.Println("Files:")
         if len(filesListCall.Files) > 0 {
                 for num, file := range filesListCall.Files {
                         fmt.Printf("[%d]: %s (%s) Trashed : %v\n", num, file.Name, file.Id, file.Trashed)
                 }
         } else {
                 fmt.Println("No files found.")
         }
 }

输出:

在浏览器中转入以下链接,然后键入授权代码: 
https://accounts.google.com/o/oauth2/auth?accesstype=offline&clientid=[redacted]&state=state-token4/sh-[redacted]-8Q
保存证书文件到 
/Users/admin/.credentials/google-drive-golang.json
{
"kind": "drive#file",
"id": "0B_wYQdperIb5Vk5vaDFjdmpibE0",
"name": "img.jpg",
"mimeType": "image/jpeg"
}
上传文件 ID : 0B_wYQdperIb5Vk5vaDFjdmpibE0
将 img.jpg 重命名为 renamed-img.jpg
文件:
[0]: renamed-img.jpg (0B_wYQdperIb5Vk5vaDFjdmpibE0) Trashed : false


以下是上传一些文件后 Google Drive 内容的屏幕截图示例

Google Drive API.png

相关链接

https://godoc.org/google.golang.org/api/drive/v3#File
https://golang.org/pkg/net/http/#Request

  • 上一条:
    在go语言中无需压缩即可创建 zip/ePub 文件(使用存储算法)
    下一条:
    使用 PHP2JS 在 PHP 和 JavaScript 之间共享数据
  • 昵称:

    邮箱:

    0条评论 (评论内容有缓存机制,请悉知!)
    最新最热
    • 分类目录
    • 人生(杂谈)
    • 技术
    • linux
    • Java
    • php
    • 框架(架构)
    • 前端
    • ThinkPHP
    • 数据库
    • 微信(小程序)
    • Laravel
    • Redis
    • Docker
    • Go
    • swoole
    • Windows
    • Python
    • 苹果(mac/ios)
    • 相关文章
    • 在go+gin中使用"github.com/skip2/go-qrcode"实现url转二维码功能(0个评论)
    • 在go语言中使用api.geonames.org接口实现根据国际邮政编码获取地址信息功能(1个评论)
    • 在go语言中使用github.com/signintech/gopdf实现生成pdf分页文件功能(0个评论)
    • 在go语言中使用github.com/signintech/gopdf实现生成pdf文件功能(0个评论)
    • 在go + gin中gorm实现指定搜索/区间搜索分页列表功能接口实例(0个评论)
    • 近期文章
    • 在go+gin中使用"github.com/skip2/go-qrcode"实现url转二维码功能(0个评论)
    • 在go语言中使用api.geonames.org接口实现根据国际邮政编码获取地址信息功能(1个评论)
    • 在go语言中使用github.com/signintech/gopdf实现生成pdf分页文件功能(0个评论)
    • gmail发邮件报错:534 5.7.9 Application-specific password required...解决方案(0个评论)
    • 欧盟关于强迫劳动的规定的官方举报渠道及官方举报网站(0个评论)
    • 在go语言中使用github.com/signintech/gopdf实现生成pdf文件功能(0个评论)
    • Laravel从Accel获得5700万美元A轮融资(0个评论)
    • 在go + gin中gorm实现指定搜索/区间搜索分页列表功能接口实例(0个评论)
    • 在go语言中实现IP/CIDR的ip和netmask互转及IP段形式互转及ip是否存在IP/CIDR(0个评论)
    • PHP 8.4 Alpha 1现已发布!(0个评论)
    • 近期评论
    • 122 在

      学历:一种延缓就业设计,生活需求下的权衡之选中评论 工作几年后,报名考研了,到现在还没认真学习备考,迷茫中。作为一名北漂互联网打工人..
    • 123 在

      Clash for Windows作者删库跑路了,github已404中评论 按理说只要你在国内,所有的流量进出都在监控范围内,不管你怎么隐藏也没用,想搞你分..
    • 原梓番博客 在

      在Laravel框架中使用模型Model分表最简单的方法中评论 好久好久都没看友情链接申请了,今天刚看,已经添加。..
    • 博主 在

      佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 @1111老铁这个不行了,可以看看近期评论的其他文章..
    • 1111 在

      佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 网站不能打开,博主百忙中能否发个APP下载链接,佛跳墙或极光..
    • 2016-10
    • 2017-09
    • 2020-03
    • 2020-05
    • 2020-06
    • 2020-07
    • 2020-12
    • 2021-01
    • 2021-05
    • 2021-06
    • 2021-07
    • 2021-08
    • 2021-10
    • 2021-11
    • 2021-12
    • 2022-01
    • 2022-02
    • 2022-03
    • 2022-04
    • 2022-05
    • 2022-06
    • 2022-07
    • 2022-08
    • 2022-09
    • 2022-10
    • 2022-11
    • 2022-12
    • 2023-01
    • 2023-02
    • 2023-03
    • 2023-04
    • 2023-05
    • 2023-06
    • 2023-07
    • 2023-08
    • 2023-09
    • 2023-10
    • 2023-11
    • 2023-12
    • 2024-01
    • 2024-02
    • 2024-03
    • 2024-04
    • 2024-05
    • 2024-06
    • 2024-07
    • 2024-08
    • 2024-11
    • 2025-02
    • 2025-04
    • 2025-05
    Top

    Copyright·© 2019 侯体宗版权所有· 粤ICP备20027696号 PHP交流群

    侯体宗的博客