Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1157)

Unified Diff: properties.go

Issue 1243323002: Refactor a bit. (Closed) Base URL: https://github.com/luci/gae.git@master
Patch Set: fix golint Created 5 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « prod/taskqueue.go ('k') | properties_test.go » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: properties.go
diff --git a/properties.go b/properties.go
deleted file mode 100644
index c7a7fb8c9f77dbad8cb3a3cc934ac35c3359540a..0000000000000000000000000000000000000000
--- a/properties.go
+++ /dev/null
@@ -1,449 +0,0 @@
-// Copyright 2015 The Chromium Authors. All rights reserved.
-// Use of this source code is governed by a BSD-style license that can be
-// found in the LICENSE file.
-
-package gae
-
-import (
- "errors"
- "fmt"
- "math"
- "reflect"
- "time"
-)
-
-var (
- // ErrDSMetaFieldUnset is returned from DSPropertyLoadSaver.{Get,Set}Meta
- // implementations when the specified meta key isn't set on the struct at
- // all.
- ErrDSMetaFieldUnset = fmt.Errorf("gae: meta field unset")
-)
-
-var (
- typeOfBSKey = reflect.TypeOf(BSKey(""))
- typeOfBool = reflect.TypeOf(false)
- typeOfByteSlice = reflect.TypeOf([]byte(nil))
- typeOfDSByteString = reflect.TypeOf(DSByteString(nil))
- typeOfDSGeoPoint = reflect.TypeOf(DSGeoPoint{})
- typeOfDSKey = reflect.TypeOf((*DSKey)(nil)).Elem()
- typeOfFloat64 = reflect.TypeOf(float64(0))
- typeOfInt64 = reflect.TypeOf(int64(0))
- typeOfString = reflect.TypeOf("")
- typeOfTime = reflect.TypeOf(time.Time{})
-)
-
-var (
- minTime = time.Unix(int64(math.MinInt64)/1e6, (int64(math.MinInt64)%1e6)*1e3)
- maxTime = time.Unix(int64(math.MaxInt64)/1e6, (int64(math.MaxInt64)%1e6)*1e3)
-)
-
-type IndexSetting bool
-
-const (
- // ShouldIndex is the default, which is why it must assume the zero value,
- // even though it's werid :(.
- ShouldIndex IndexSetting = false
- NoIndex IndexSetting = true
-)
-
-func (i IndexSetting) String() string {
- if i {
- return "NoIndex"
- }
- return "ShouldIndex"
-}
-
-// DSProperty is a value plus an indicator of whether the value should be
-// indexed. Name and Multiple are stored in the DSPropertyMap object.
-type DSProperty struct {
- value interface{}
- indexSetting IndexSetting
- propType DSPropertyType
-}
-
-// MkDSProperty makes a new indexed* DSProperty and returns it. If val is an
-// invalid value, this panics (so don't do it). If you want to handle the error
-// normally, use SetValue(..., ShouldIndex) instead.
-//
-// *indexed if val is not an unindexable type like []byte.
-func MkDSProperty(val interface{}) DSProperty {
- ret := DSProperty{}
- if err := ret.SetValue(val, ShouldIndex); err != nil {
- panic(err)
- }
- return ret
-}
-
-// MkDSPropertyNI makes a new DSProperty (with noindex set to true), and returns
-// it. If val is an invalid value, this panics (so don't do it). If you want to
-// handle the error normally, use SetValue(..., NoIndex) instead.
-func MkDSPropertyNI(val interface{}) DSProperty {
- ret := DSProperty{}
- if err := ret.SetValue(val, NoIndex); err != nil {
- panic(err)
- }
- return ret
-}
-
-// DSPropertyConverter may be implemented by the pointer-to a struct field which
-// is serialized by RawDatastore. Its ToDSProperty will be called on save, and
-// it's FromDSProperty will be called on load (from datastore). The method may
-// do arbitrary computation, and if it encounters an error, may return it. This
-// error will be a fatal error (as defined by DSPropertyLoadSaver) for the
-// struct conversion.
-//
-// Example:
-// type Complex complex
-// func (c *Complex) ToDSProperty() (ret DSProperty, err error) {
-// // something like:
-// err = ret.SetValue(fmt.Sprint(*c), true)
-// return
-// }
-// func (c *Complex) FromDSProperty(p DSProperty) (err error) {
-// ... load *c from p ...
-// }
-//
-// type MyStruct struct {
-// Complexity []Complex // acts like []complex, but can be serialized to DS
-// }
-type DSPropertyConverter interface {
- // TODO(riannucci): Allow a convertable to return multiple values. This is
- // eminently doable (as long as the single-slice restriction is kept). It
- // could also cut down on the amount of reflection necessary when resolving
- // a path in a struct (in the struct loading routine in helper).
-
- ToDSProperty() (DSProperty, error)
- FromDSProperty(DSProperty) error
-}
-
-// DSPropertyType is a single-byte representation of the type of data contained
-// in a DSProperty. The specific values of this type information are chosen so
-// that the types sort according to the order of types as sorted by the
-// datastore.
-type DSPropertyType byte
-
-// These constants are in the order described by
-// https://cloud.google.com/appengine/docs/go/datastore/entities#Go_Value_type_ordering
-// with a slight divergence for the Int/Time split.
-// NOTE: this enum can only occupy 7 bits, because we use the high bit to encode
-// indexed/non-indexed. See typData.WriteBinary.
-const (
- DSPTNull DSPropertyType = iota
- DSPTInt
-
- // DSPTTime is a slight divergence from the way that datastore natively stores
- // time. In datastore, times and integers actually sort together
- // (apparently?). This is probably insane, and I don't want to add the
- // complexity of field 'meaning' as a sparate concept from the field's 'type'
- // (which is what datastore seems to do, judging from the protobufs). So if
- // you're here because you implemented an app which relies on time.Time and
- // int64 sorting together, then this is why your app acts differently in
- // production. My advice is to NOT DO THAT. If you really want this (and you
- // probably don't), you should take care of the time.Time <-> int64 conversion
- // in your app and just use a property type of int64 (consider using
- // DSPropertyConverter).
- DSPTTime
-
- // DSPTBoolFalse and True are also a slight divergence, but not a semantic
- // one. IIUC, in datastore 'bool' is actually the type and the value is either
- // 0 or 1 (taking another byte to store). Since we have plenty of space in
- // this type byte, I just merge the value into the type for booleans. If this
- // becomes problematic, consider changing this to just pvBool, and then
- // encoding a 0 or 1 as a byte in the relevant marshalling routines.
- DSPTBoolFalse
- DSPTBoolTrue
-
- DSPTBytes // []byte or datastore.ByteString
- DSPTString // string or string noindex
- DSPTFloat
- DSPTGeoPoint
- DSPTKey
- DSPTBlobKey
-
- DSPTUnknown
-)
-
-func (t DSPropertyType) String() string {
- switch t {
- case DSPTNull:
- return "DSPTNull"
- case DSPTInt:
- return "DSPTInt"
- case DSPTTime:
- return "DSPTTime"
- case DSPTBoolFalse:
- return "DSPTBoolFalse"
- case DSPTBoolTrue:
- return "DSPTBoolTrue"
- case DSPTBytes:
- return "DSPTBytes"
- case DSPTString:
- return "DSPTString"
- case DSPTFloat:
- return "DSPTFloat"
- case DSPTGeoPoint:
- return "DSPTGeoPoint"
- case DSPTKey:
- return "DSPTKey"
- case DSPTBlobKey:
- return "DSPTBlobKey"
- default:
- return fmt.Sprintf("DSPTUnknown(%02x)", byte(t))
- }
-}
-
-// DSPropertyTypeOf returns the DSPT* type of the given DSProperty-compatible
-// value v. If checkValid is true, this method will also ensure that time.Time
-// and DSGeoPoint have valid values.
-func DSPropertyTypeOf(v interface{}, checkValid bool) (DSPropertyType, error) {
- switch x := v.(type) {
- case nil:
- return DSPTNull, nil
- case int64:
- return DSPTInt, nil
- case float64:
- return DSPTFloat, nil
- case bool:
- if x {
- return DSPTBoolTrue, nil
- }
- return DSPTBoolFalse, nil
- case []byte, DSByteString:
- return DSPTBytes, nil
- case BSKey:
- return DSPTBlobKey, nil
- case string:
- return DSPTString, nil
- case DSKey:
- // TODO(riannucci): Check key for validity in its own namespace?
- return DSPTKey, nil
- case time.Time:
- err := error(nil)
- if checkValid && (x.Before(minTime) || x.After(maxTime)) {
- err = errors.New("time value out of range")
- }
- if checkValid && x.Location() != time.UTC {
- err = fmt.Errorf("time value has wrong Location: %s", x.Location())
- }
- return DSPTTime, err
- case DSGeoPoint:
- err := error(nil)
- if checkValid && !x.Valid() {
- err = errors.New("invalid GeoPoint value")
- }
- return DSPTGeoPoint, err
- default:
- return DSPTUnknown, fmt.Errorf("gae: DSProperty has bad type %T", v)
- }
-}
-
-// DSUpconvertUnderlyingType takes an object o, and attempts to convert it to
-// its native datastore-compatible type. e.g. int16 will convert to int64, and
-// `type Foo string` will convert to `string`.
-func DSUpconvertUnderlyingType(o interface{}, t reflect.Type) (interface{}, reflect.Type) {
- v := reflect.ValueOf(o)
- switch t.Kind() {
- case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
- o = v.Int()
- t = typeOfInt64
- case reflect.Bool:
- o = v.Bool()
- t = typeOfBool
- case reflect.String:
- if t != typeOfBSKey {
- o = v.String()
- t = typeOfString
- }
- case reflect.Float32, reflect.Float64:
- o = v.Float()
- t = typeOfFloat64
- case reflect.Slice:
- if t != typeOfDSByteString && t.Elem().Kind() == reflect.Uint8 {
- o = v.Bytes()
- t = typeOfByteSlice
- }
- case reflect.Struct:
- if t == typeOfTime {
- // time in a DSProperty can only hold microseconds
- o = v.Interface().(time.Time).Round(time.Microsecond)
- }
- }
- return o, t
-}
-
-// Value returns the current value held by this property. It's guaranteed to
-// be a valid value type (i.e. `p.SetValue(p.Value(), true)` will never return
-// an error).
-func (p DSProperty) Value() interface{} { return p.value }
-
-// IndexSetting says weather or not the datastore should create indicies for
-// this value.
-func (p DSProperty) IndexSetting() IndexSetting { return p.indexSetting }
-
-// Type is the DSPT* type of the data contained in Value().
-func (p DSProperty) Type() DSPropertyType { return p.propType }
-
-// SetValue sets the Value field of a DSProperty, and ensures that its value
-// conforms to the permissible types. That way, you're guaranteed that if you
-// have a DSProperty, its value is valid.
-//
-// value is the property value. The valid types are:
-// - int64
-// - bool
-// - string
-// - float64
-// - DSByteString
-// - DSKey
-// - time.Time
-// - BSKey
-// - DSGeoPoint
-// - []byte (up to 1 megabyte in length)
-// This set is smaller than the set of valid struct field types that the
-// datastore can load and save. A Property Value cannot be a slice (apart
-// from []byte); use multiple Properties instead. Also, a Value's type
-// must be explicitly on the list above; it is not sufficient for the
-// underlying type to be on that list. For example, a Value of "type
-// myInt64 int64" is invalid. Smaller-width integers and floats are also
-// invalid. Again, this is more restrictive than the set of valid struct
-// field types.
-//
-// A value may also be the nil interface value; this is equivalent to
-// Python's None but not directly representable by a Go struct. Loading
-// a nil-valued property into a struct will set that field to the zero
-// value.
-func (p *DSProperty) SetValue(value interface{}, is IndexSetting) (err error) {
- t := reflect.Type(nil)
- pt := DSPTNull
- if value != nil {
- t = reflect.TypeOf(value)
- value, t = DSUpconvertUnderlyingType(value, t)
- if pt, err = DSPropertyTypeOf(value, true); err != nil {
- return
- }
- }
- p.propType = pt
- p.value = value
- p.indexSetting = is
- if t == typeOfByteSlice {
- p.indexSetting = NoIndex
- }
- return
-}
-
-// DSPropertyLoadSaver may be implemented by a user type, and RawDatastore will
-// use this interface to serialize the type instead of trying to automatically
-// create a serialization codec for it with helper.GetPLS.
-type DSPropertyLoadSaver interface {
- // Load takes the values from the given map and attempts to save them into
- // the underlying object (usually a struct or a DSPropertyMap). If a fatal
- // error occurs, it's returned via error. If non-fatal conversion errors
- // occur, error will be a MultiError containing one or more ErrDSFieldMismatch
- // objects.
- Load(DSPropertyMap) error
-
- // Save returns the current property as a DSPropertyMap. if withMeta is true,
- // then the DSPropertyMap contains all the metadata (e.g. '$meta' fields)
- // which was held by this DSPropertyLoadSaver.
- Save(withMeta bool) (DSPropertyMap, error)
-
- // GetMeta will get information about the field which has the struct tag in
- // the form of `gae:"$<key>[,<value>]?"`.
- //
- // string and int64 fields will return the <value> in the struct tag,
- // converted to the appropriate type, if the field has the zero value.
- //
- // Example:
- // type MyStruct struct {
- // CoolField int64 `gae:"$id,1"`
- // }
- // val, err := helper.GetPLS(&MyStruct{}).GetMeta("id")
- // // val == 1
- // // err == nil
- //
- // val, err := helper.GetPLS(&MyStruct{10}).GetMeta("id")
- // // val == 10
- // // err == nil
- GetMeta(key string) (interface{}, error)
-
- // SetMeta allows you to set the current value of the meta-keyed field.
- SetMeta(key string, val interface{}) error
-
- // Problem indicates that this PLS has a fatal problem. Usually this is
- // set when the underlying struct has recursion, invalid field types, nested
- // slices, etc.
- Problem() error
-}
-
-// DSPropertyMap represents the contents of a datastore entity in a generic way.
-// It maps from property name to a list of property values which correspond to
-// that property name. It is the spiritual successor to PropertyList from the
-// original SDK.
-//
-// DSPropertyMap may contain "meta" values, which are keyed with a '$' prefix.
-// Technically the datastore allows arbitrary property names, but all of the
-// SDKs go out of their way to try to make all property names valid programming
-// language tokens. Special values must correspond to a single DSProperty...
-// corresponding to 0 is equivalent to unset, and corresponding to >1 is an
-// error. So:
-//
-// {
-// "$id": {MkDSProperty(1)}, // GetProperty("id") -> 1, nil
-// "$foo": {}, // GetProperty("foo") -> nil, ErrDSMetaFieldUnset
-// // GetProperty("bar") -> nil, ErrDSMetaFieldUnset
-// "$meep": {
-// MkDSProperty("hi"),
-// MkDSProperty("there")}, // GetProperty("meep") -> nil, error!
-// }
-//
-// Additionally, Save returns a copy of the map with the meta keys omitted (e.g.
-// these keys are not going to be serialized to the datastore).
-type DSPropertyMap map[string][]DSProperty
-
-var _ DSPropertyLoadSaver = DSPropertyMap(nil)
-
-// Load implements DSPropertyLoadSaver.Load
-func (pm DSPropertyMap) Load(props DSPropertyMap) error {
- for k, v := range props {
- pm[k] = append(pm[k], v...)
- }
- return nil
-}
-
-// Save implements DSPropertyLoadSaver.Save by returning a copy of the
-// current map data.
-func (pm DSPropertyMap) Save(withMeta bool) (DSPropertyMap, error) {
- if len(pm) == 0 {
- return DSPropertyMap{}, nil
- }
- ret := make(DSPropertyMap, len(pm))
- for k, v := range pm {
- if withMeta || len(k) == 0 || k[0] != '$' {
- ret[k] = append(ret[k], v...)
- }
- }
- return ret, nil
-}
-
-func (pm DSPropertyMap) GetMeta(key string) (interface{}, error) {
- v, ok := pm["$"+key]
- if !ok || len(v) == 0 {
- return nil, ErrDSMetaFieldUnset
- }
- if len(v) > 1 {
- return nil, errors.New("gae: too many values for Meta key")
- }
- return v[0].Value(), nil
-}
-
-func (pm DSPropertyMap) SetMeta(key string, val interface{}) error {
- prop := DSProperty{}
- if err := prop.SetValue(val, NoIndex); err != nil {
- return err
- }
- pm["$"+key] = []DSProperty{prop}
- return nil
-}
-
-func (pm DSPropertyMap) Problem() error {
- return nil
-}
« no previous file with comments | « prod/taskqueue.go ('k') | properties_test.go » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698