rows.go

// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.

package pouchdb

import (
	"encoding/json"
	"io"
	"strings"

	"github.com/gopherjs/gopherjs/js"

	"github.com/go-kivik/kivik/v4/driver"
	"github.com/go-kivik/pouchdb/v4/bindings"
)

var jsJSON *js.Object

func init() {
	jsJSON = js.Global.Get("JSON")
}

type rows struct {
	*js.Object
	Off   int64  `js:"offset"`
	TRows int64  `js:"total_rows"`
	USeq  string `js:"update_seq"`
}

var _ driver.Rows = &rows{}

func (r *rows) Close() error {
	r.Delete("rows") // Free up memory used by any remaining rows
	return nil
}

func (r *rows) Next(row *driver.Row) (err error) {
	defer bindings.RecoverError(&err)
	if r.Get("rows") == js.Undefined || r.Get("rows").Length() == 0 {
		return io.EOF
	}
	next := r.Get("rows").Call("shift")
	row.ID = next.Get("id").String()
	row.Key = json.RawMessage(jsJSON.Call("stringify", next.Get("key")).String())
	row.Value = strings.NewReader(jsJSON.Call("stringify", next.Get("value")).String())
	if doc := next.Get("doc"); doc != js.Undefined {
		row.Doc = strings.NewReader(jsJSON.Call("stringify", doc).String())
	}
	return nil
}

func (r *rows) Offset() int64 {
	return r.Off
}

func (r *rows) TotalRows() int64 {
	return r.TRows
}

func (r *rows) UpdateSeq() string {
	return "" // PouchDB doesn't support this option
}