Skip to content

Commit

Permalink
split kv.go
Browse files Browse the repository at this point in the history
  • Loading branch information
syumai committed Jan 12, 2025
1 parent 43b36c4 commit c5873af
Show file tree
Hide file tree
Showing 6 changed files with 252 additions and 225 deletions.
16 changes: 16 additions & 0 deletions cloudflare/kv/delete.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
package kv

import (
"github.com/syumai/workers/internal/jsutil"
)

// Delete deletes key-value pair specified by the key.
// - if a network error happens, returns error.
func (ns *Namespace) Delete(key string) error {
p := ns.instance.Call("delete", key)
_, err := jsutil.AwaitPromise(p)
if err != nil {
return err
}
return nil
}
48 changes: 48 additions & 0 deletions cloudflare/kv/get.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
package kv

import (
"io"
"syscall/js"

"github.com/syumai/workers/internal/jsutil"
)

// GetOptions represents Cloudflare KV namespace get options.
// - https://github.com/cloudflare/workers-types/blob/3012f263fb1239825e5f0061b267c8650d01b717/index.d.ts#L930
type GetOptions struct {
CacheTTL int
}

func (opts *GetOptions) toJS(type_ string) js.Value {
obj := jsutil.NewObject()
obj.Set("type", type_)
if opts == nil {
return obj
}
if opts.CacheTTL != 0 {
obj.Set("cacheTtl", opts.CacheTTL)
}
return obj
}

// GetString gets string value by the specified key.
// - if a network error happens, returns error.
func (ns *Namespace) GetString(key string, opts *GetOptions) (string, error) {
p := ns.instance.Call("get", key, opts.toJS("text"))
v, err := jsutil.AwaitPromise(p)
if err != nil {
return "", err
}
return v.String(), nil
}

// GetReader gets stream value by the specified key.
// - if a network error happens, returns error.
func (ns *Namespace) GetReader(key string, opts *GetOptions) (io.Reader, error) {
p := ns.instance.Call("get", key, opts.toJS("stream"))
v, err := jsutil.AwaitPromise(p)
if err != nil {
return nil, err
}
return jsutil.ConvertReadableStreamToReadCloser(v), nil
}
225 changes: 0 additions & 225 deletions cloudflare/kv/kv.go

This file was deleted.

101 changes: 101 additions & 0 deletions cloudflare/kv/list.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,101 @@
package kv

import (
"fmt"
"syscall/js"

"github.com/syumai/workers/internal/jsutil"
)

// ListOptions represents Cloudflare KV namespace list options.
// - https://github.com/cloudflare/workers-types/blob/3012f263fb1239825e5f0061b267c8650d01b717/index.d.ts#L946
type ListOptions struct {
Limit int
Prefix string
Cursor string
}

func (opts *ListOptions) toJS() js.Value {
if opts == nil {
return js.Undefined()
}
obj := jsutil.NewObject()
if opts.Limit != 0 {
obj.Set("limit", opts.Limit)
}
if opts.Prefix != "" {
obj.Set("prefix", opts.Prefix)
}
if opts.Cursor != "" {
obj.Set("cursor", opts.Cursor)
}
return obj
}

// ListKey represents Cloudflare KV namespace list key.
// - https://github.com/cloudflare/workers-types/blob/3012f263fb1239825e5f0061b267c8650d01b717/index.d.ts#L940
type ListKey struct {
Name string
// Expiration is an expiration of KV value cache. The value `0` means no expiration.
Expiration int
// Metadata map[string]any // TODO: implement
}

// toListKey converts JavaScript side's KVNamespaceListKey to *ListKey.
// - https://github.com/cloudflare/workers-types/blob/3012f263fb1239825e5f0061b267c8650d01b717/index.d.ts#L940
func toListKey(v js.Value) (*ListKey, error) {
expVal := v.Get("expiration")
var exp int
if !expVal.IsUndefined() {
exp = expVal.Int()
}
return &ListKey{
Name: v.Get("name").String(),
Expiration: exp,
// Metadata // TODO: implement. This may return an error, so this func signature has an error in return parameters.
}, nil
}

// ListResult represents Cloudflare KV namespace list result.
// - https://github.com/cloudflare/workers-types/blob/3012f263fb1239825e5f0061b267c8650d01b717/index.d.ts#L952
type ListResult struct {
Keys []*ListKey
ListComplete bool
Cursor string
}

// toListResult converts JavaScript side's KVNamespaceListResult to *ListResult.
// - https://github.com/cloudflare/workers-types/blob/3012f263fb1239825e5f0061b267c8650d01b717/index.d.ts#L952
func toListResult(v js.Value) (*ListResult, error) {
keysVal := v.Get("keys")
keys := make([]*ListKey, keysVal.Length())
for i := 0; i < len(keys); i++ {
key, err := toListKey(keysVal.Index(i))
if err != nil {
return nil, fmt.Errorf("error converting to ListKey: %w", err)
}
keys[i] = key
}

cursorVal := v.Get("cursor")
var cursor string
if !cursorVal.IsUndefined() {
cursor = cursorVal.String()
}

return &ListResult{
Keys: keys,
ListComplete: v.Get("list_complete").Bool(),
Cursor: cursor,
}, nil
}

// List lists keys stored into the KV namespace.
func (ns *Namespace) List(opts *ListOptions) (*ListResult, error) {
p := ns.instance.Call("list", opts.toJS())
v, err := jsutil.AwaitPromise(p)
if err != nil {
return nil, err
}
return toListResult(v)
}
Loading

0 comments on commit c5873af

Please sign in to comment.