storj/internal/fpath/temp_data.go

39 lines
972 B
Go
Raw Normal View History

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
// TODO maybe there is better place for this
package fpath
import "context"
// The key type is unexported to prevent collisions with context keys defined in
// other packages.
type key int
// temp is the context key for temp struct
const tempKey key = 0
type temp struct {
inmemory bool
directory string
}
// WithTempData creates context with information how store temporary data, in memory or on disk
func WithTempData(ctx context.Context, directory string, inmemory bool) context.Context {
2019-06-05 15:02:29 +01:00
temp := temp{
inmemory: inmemory,
directory: directory,
}
return context.WithValue(ctx, tempKey, temp)
}
// GetTempData returns if temporary data should be stored in memory or on disk
func GetTempData(ctx context.Context) (string, bool, bool) {
tempValue, ok := ctx.Value(tempKey).(temp)
if !ok {
return "", false, false
}
return tempValue.directory, tempValue.inmemory, ok
}