//  Copyright (c) 2016 Couchbase, Inc.
//  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 moss

import (
	"github.com/couchbase/moss"

	"github.com/blevesearch/bleve/index/store"
)

type Batch struct {
	store   *Store
	merge   *store.EmulatedMerge
	batch   moss.Batch
	buf     []byte // Non-nil when using pre-alloc'ed / NewBatchEx().
	bufUsed int
}

func (b *Batch) Set(key, val []byte) {
	var err error
	if b.buf != nil {
		b.bufUsed += len(key) + len(val)
		err = b.batch.AllocSet(key, val)
	} else {
		err = b.batch.Set(key, val)
	}

	if err != nil {
		b.store.Logf("bleve moss batch.Set err: %v", err)
	}
}

func (b *Batch) Delete(key []byte) {
	var err error
	if b.buf != nil {
		b.bufUsed += len(key)
		err = b.batch.AllocDel(key)
	} else {
		err = b.batch.Del(key)
	}

	if err != nil {
		b.store.Logf("bleve moss batch.Delete err: %v", err)
	}
}

func (b *Batch) Merge(key, val []byte) {
	if b.buf != nil {
		b.bufUsed += len(key) + len(val)
	}
	b.merge.Merge(key, val)
}

func (b *Batch) Reset() {
	err := b.Close()
	if err != nil {
		b.store.Logf("bleve moss batch.Close err: %v", err)
		return
	}

	batch, err := b.store.ms.NewBatch(0, 0)
	if err == nil {
		b.batch = batch
		b.merge = store.NewEmulatedMerge(b.store.mo)
		b.buf = nil
		b.bufUsed = 0
	}
}

func (b *Batch) Close() error {
	b.merge = nil
	err := b.batch.Close()
	b.batch = nil
	return err
}
