key.go

package aedatastore

import (
	"context"

	w "go.mercari.io/datastore"
	"google.golang.org/appengine/datastore"
)

var _ w.Key = (*keyImpl)(nil)
var _ w.PendingKey = (*pendingKeyImpl)(nil)

type keyImpl struct {
	ctx       context.Context
	kind      string
	id        int64
	name      string
	parent    *keyImpl
	namespace string
}

type pendingKeyImpl struct {
	ctx context.Context
	key *datastore.Key
}

type contextPendingKey struct{}

func (k *keyImpl) Kind() string {
	if k == nil {
		panic("k is nil")
	}
	return k.kind
}

func (k *keyImpl) ID() int64 {
	return k.id
}

func (k *keyImpl) Name() string {
	return k.name
}

func (k *keyImpl) ParentKey() w.Key {
	if k.parent == nil {
		return nil
	}
	return k.parent
}

func (k *keyImpl) Namespace() string {
	return k.namespace
}

func (k *keyImpl) SetNamespace(namespace string) {
	k.namespace = namespace
}

func (k *keyImpl) String() string {
	return toOriginalKey(k).String()
}

func (k *keyImpl) GobEncode() ([]byte, error) {
	return toOriginalKey(k).GobEncode()
}

func (k *keyImpl) GobDecode(buf []byte) error {
	origKey := &datastore.Key{}
	err := origKey.GobDecode(buf)
	if err != nil {
		return err
	}

	k.kind = origKey.Kind()
	k.id = origKey.IntID()
	k.name = origKey.StringID()
	k.parent = toWrapperKey(k.ctx, origKey.Parent())
	k.namespace = origKey.Namespace()

	return nil
}

func (k *keyImpl) MarshalJSON() ([]byte, error) {
	return toOriginalKey(k).MarshalJSON()
}

func (k *keyImpl) UnmarshalJSON(buf []byte) error {
	origKey := &datastore.Key{}
	err := origKey.UnmarshalJSON(buf)
	if err != nil {
		return err
	}

	k.kind = origKey.Kind()
	k.id = origKey.IntID()
	k.name = origKey.StringID()
	k.parent = toWrapperKey(k.ctx, origKey.Parent())
	k.namespace = origKey.Namespace()

	return nil
}

func (k *keyImpl) Encode() string {
	return toOriginalKey(k).Encode()
}

func (k *keyImpl) Equal(o w.Key) bool {
	var a w.Key = k
	var b = o
	for {
		if a == nil && b == nil {
			return true
		} else if a != nil && b == nil {
			return false
		} else if a == nil && b != nil {
			return false
		}
		if a.Kind() != b.Kind() || a.Name() != b.Name() || a.ID() != b.ID() || a.Namespace() != b.Namespace() {
			return false
		}

		// NOTE Don't checking appID. align to Cloud Datastore API.

		a = a.ParentKey()
		b = b.ParentKey()
	}
}

func (k *keyImpl) Incomplete() bool {
	return k.Name() == "" && k.ID() == 0
}

func (p *pendingKeyImpl) StoredContext() context.Context {
	return context.WithValue(p.ctx, contextPendingKey{}, p)
}