mirror of https://github.com/etcd-io/dbtester.git
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
// Copyright 2016 CoreOS, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package upload
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
|
|
"github.com/coreos/dbtester/remotestorage"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
Command = &cobra.Command{
|
|
Use: "upload",
|
|
Short: "Uploads to cloud storage.",
|
|
RunE: CommandFunc,
|
|
}
|
|
)
|
|
|
|
var (
|
|
from string
|
|
to string
|
|
isDirectory bool
|
|
|
|
googleCloudProjectName string
|
|
keyPath string
|
|
bucket string
|
|
)
|
|
|
|
func init() {
|
|
cobra.EnablePrefixMatching = true
|
|
}
|
|
|
|
func init() {
|
|
Command.PersistentFlags().StringVar(&from, "from", "", "file to upload.")
|
|
Command.PersistentFlags().StringVar(&to, "to", "", "file path to upload.")
|
|
Command.PersistentFlags().BoolVar(&isDirectory, "directory", false, "'true' if uploading directory.")
|
|
Command.PersistentFlags().StringVar(&googleCloudProjectName, "google-cloud-project-name", "", "Google cloud project name.")
|
|
Command.PersistentFlags().StringVar(&keyPath, "key-path", "", "Path of key file.")
|
|
Command.PersistentFlags().StringVar(&bucket, "bucket", "", "Bucket name in cloud storage.")
|
|
}
|
|
|
|
func CommandFunc(cmd *cobra.Command, args []string) error {
|
|
log.Println("opening key", keyPath)
|
|
kbs, err := ioutil.ReadFile(keyPath)
|
|
if err != nil {
|
|
return fmt.Errorf("error when opening key %s(%v)", keyPath, err)
|
|
}
|
|
u, err := remotestorage.NewGoogleCloudStorage(kbs, googleCloudProjectName)
|
|
if err != nil {
|
|
return fmt.Errorf("error when NewGoogleCloudStorage %s(%v)", googleCloudProjectName, err)
|
|
}
|
|
if !isDirectory {
|
|
if err := u.UploadFile(bucket, from, to); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if err := u.UploadDir(bucket, from, to); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|