func getBucket(ctx context.Context) (b gcs.Bucket, err error) {
if *fBucket == "" {
err = errors.New("You must set --bucket.")
return
}
// Set up the token source.
const scope = gcs.Scope_ReadOnly
tokenSrc, err := google.DefaultTokenSource(context.Background(), scope)
if err != nil {
err = fmt.Errorf("DefaultTokenSource: %v", err)
return
}
// Use that to create a GCS connection.
cfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
}
conn, err := gcs.NewConn(cfg)
if err != nil {
err = fmt.Errorf("NewConn: %v", err)
return
}
// Open the bucket.
b, err = conn.OpenBucket(ctx, *fBucket)
if err != nil {
err = fmt.Errorf("OpenBucket: %v", err)
return
}
return
}
func getBucket() (b gcs.Bucket, err error) {
if *fBucket == "" {
err = errors.New("You must set --bucket.")
return
}
// Create the HTTP client.
const scope = gcs.Scope_FullControl
tokenSrc, err := google.DefaultTokenSource(context.Background(), scope)
if err != nil {
err = fmt.Errorf("DefaultTokenSource: %v", err)
return
}
// Set up a GCS connection.
cfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
MaxBackoffSleep: 60 * time.Minute,
}
conn, err := gcs.NewConn(cfg)
if err != nil {
err = fmt.Errorf("gcs.NewConn: %v", err)
return
}
// Open the bucket.
b, err = conn.OpenBucket(context.Background(), *fBucket)
if err != nil {
err = fmt.Errorf("OpenBucket: %v", err)
return
}
return
}
func init() {
makeDeps := func(ctx context.Context) (deps gcstesting.BucketTestDeps) {
var err error
// Set up the token source.
const scope = gcs.Scope_FullControl
tokenSrc, err := google.DefaultTokenSource(context.Background(), scope)
AssertEq(nil, err)
// Use that to create a GCS connection, enabling retry if requested.
cfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
}
if *fUseRetry {
cfg.MaxBackoffSleep = 5 * time.Minute
deps.BuffersEntireContentsForCreate = true
}
if *fDebugGCS {
cfg.GCSDebugLogger = log.New(os.Stderr, "gcs: ", 0)
}
if *fDebugHTTP {
cfg.HTTPDebugLogger = log.New(os.Stderr, "http: ", 0)
}
conn, err := gcs.NewConn(cfg)
AssertEq(nil, err)
// Open the bucket.
deps.Bucket, err = conn.OpenBucket(ctx, *fBucket)
AssertEq(nil, err)
// Clear the bucket.
err = gcsutil.DeleteAllObjects(ctx, deps.Bucket)
if err != nil {
panic("DeleteAllObjects: " + err.Error())
}
// Set up other information.
deps.Clock = timeutil.RealClock()
deps.SupportsCancellation = true
return
}
gcstesting.RegisterBucketTests(makeDeps)
}
func getConn() (c gcs.Conn, err error) {
// Create the oauth2 token source.
const scope = gcs.Scope_FullControl
tokenSrc, err := google.DefaultTokenSource(context.Background(), scope)
if err != nil {
return nil, err
}
// Create the connection.
const userAgent = "gcsfuse/0.0"
cfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
UserAgent: userAgent,
}
return gcs.NewConn(cfg)
}
func (t *ConnTest) SetUp(ti *TestInfo) {
var err error
t.ctx = ti.Ctx
// Set up a token source.
const scope = gcs.Scope_FullControl
tokenSrc, err := google.DefaultTokenSource(context.Background(), scope)
AssertEq(nil, err)
// Use that to create a GCS connection, enabling retry if requested.
cfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
}
t.conn, err = gcs.NewConn(cfg)
AssertEq(nil, err)
}
func getConn(flags *flagStorage) (c gcs.Conn, err error) {
// Create the oauth2 token source.
const scope = gcs.Scope_FullControl
var tokenSrc oauth2.TokenSource
if flags.KeyFile != "" {
tokenSrc, err = newTokenSourceFromPath(flags.KeyFile, scope)
if err != nil {
err = fmt.Errorf("newTokenSourceFromPath: %v", err)
return
}
} else {
tokenSrc, err = google.DefaultTokenSource(context.Background(), scope)
if err != nil {
err = fmt.Errorf("DefaultTokenSource: %v", err)
return
}
}
// Create the connection.
const userAgent = "gcsfuse/0.0"
cfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
UserAgent: userAgent,
}
if flags.DebugHTTP {
cfg.HTTPDebugLogger = log.New(os.Stderr, "http: ", 0)
}
if flags.DebugGCS {
cfg.GCSDebugLogger = log.New(os.Stderr, "gcs: ", 0)
}
return gcs.NewConn(cfg)
}
func createBucket() (bucket gcs.Bucket, err error) {
// Create an authenticated HTTP client.
tokenSrc, err := google.DefaultTokenSource(
context.Background(),
gcs.Scope_FullControl)
if err != nil {
err = fmt.Errorf("DefaultTokenSource: %v", err)
return
}
// Use that to create a connection.
connCfg := &gcs.ConnConfig{
TokenSource: tokenSrc,
}
conn, err := gcs.NewConn(connCfg)
if err != nil {
err = fmt.Errorf("NewConn: %v", err)
return
}
// Extract the bucket.
if *fBucket == "" {
err = errors.New("You must set --bucket.")
return
}
bucket, err = conn.OpenBucket(context.Background(), *fBucket)
if err != nil {
err = fmt.Errorf("OpenBucket: %v", err)
return
}
return
}
