在go语言中实现Google Drive API 上传和重命名示例
Go / 管理员 发布于 1年前 596
在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 内容的屏幕截图示例
相关链接
https://godoc.org/google.golang.org/api/drive/v3#File
https://golang.org/pkg/net/http/#Request- 近期文章
- 在go中实现一个常用的先进先出的缓存淘汰算法示例代码(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个评论)
- 近期评论
-
122 在
学历：一种延缓就业设计，生活需求下的权衡之选中评论 工作几年后，报名考研了，到现在还没认真学习备考，迷茫中。作为一名北漂互联网打工人.. -
123 在
Clash for Windows作者删库跑路了，github已404中评论 按理说只要你在国内，所有的流量进出都在监控范围内，不管你怎么隐藏也没用，想搞你分.. -
原梓番博客 在
在Laravel框架中使用模型Model分表最简单的方法中评论 好久好久都没看友情链接申请了，今天刚看，已经添加。.. -
博主 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 @1111老铁这个不行了，可以看看近期评论的其他文章.. -
1111 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 网站不能打开，博主百忙中能否发个APP下载链接，佛跳墙或极光..
Copyright·© 2019 侯体宗版权所有·
粤ICP备20027696号
