gocryptotrader/exchanges/stream/buffer/buffer.go

package buffer

import (
	"errors"
	"fmt"
	"sort"

	"github.com/thrasher-corp/gocryptotrader/currency"
	"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
	"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
)

const packageError = "websocket orderbook buffer error: %w"

var (
	errUnsetExchangeName            = errors.New("exchange name unset")
	errUnsetDataHandler             = errors.New("datahandler unset")
	errIssueBufferEnabledButNoLimit = errors.New("buffer enabled but no limit set")
	errUpdateIsNil                  = errors.New("update is nil")
	errUpdateNoTargets              = errors.New("update bid/ask targets cannot be nil")
)

// Setup sets private variables
func (w *Orderbook) Setup(obBufferLimit int,
	bufferEnabled,
	sortBuffer,
	sortBufferByUpdateIDs,
	updateEntriesByID bool, exchangeName string, dataHandler chan interface{}) error {
	if exchangeName == "" {
		return fmt.Errorf(packageError, errUnsetExchangeName)
	}
	if dataHandler == nil {
		return fmt.Errorf(packageError, errUnsetDataHandler)
	}
	if bufferEnabled && obBufferLimit < 1 {
		return fmt.Errorf(packageError, errIssueBufferEnabledButNoLimit)
	}
	w.obBufferLimit = obBufferLimit
	w.bufferEnabled = bufferEnabled
	w.sortBuffer = sortBuffer
	w.sortBufferByUpdateIDs = sortBufferByUpdateIDs
	w.updateEntriesByID = updateEntriesByID
	w.exchangeName = exchangeName
	w.dataHandler = dataHandler
	w.ob = make(map[currency.Code]map[currency.Code]map[asset.Item]*orderbookHolder)
	return nil
}

// validate validates update against setup values
func (w *Orderbook) validate(u *Update) error {
	if u == nil {
		return fmt.Errorf(packageError, errUpdateIsNil)
	}
	if len(u.Bids) == 0 && len(u.Asks) == 0 {
		return fmt.Errorf(packageError, errUpdateNoTargets)
	}
	return nil
}

// Update updates a local buffer using bid targets and ask targets then updates
// main orderbook
// Volume == 0; deletion at price target
// Price target not found; append of price target
// Price target found; amend volume of price target
func (w *Orderbook) Update(u *Update) error {
	if err := w.validate(u); err != nil {
		return err
	}
	w.m.Lock()
	defer w.m.Unlock()
	obLookup, ok := w.ob[u.Pair.Base][u.Pair.Quote][u.Asset]
	if !ok {
		return fmt.Errorf("ob.Base could not be found for Exchange %s CurrencyPair: %s AssetType: %s",
			w.exchangeName,
			u.Pair,
			u.Asset)
	}

	if w.bufferEnabled {
		processed, err := w.processBufferUpdate(obLookup, u)
		if err != nil {
			return err
		}

		if !processed {
			return nil
		}
	} else {
		err := w.processObUpdate(obLookup, u)
		if err != nil {
			return err
		}
	}

	err := obLookup.ob.Process()
	if err != nil {
		return err
	}

	// Process in data handler
	select {
	case w.dataHandler <- obLookup.ob:
	default:
	}
	return nil
}

// processBufferUpdate stores update into buffer, when buffer at capacity as
// defined by w.obBufferLimit it well then sort and apply updates.
func (w *Orderbook) processBufferUpdate(o *orderbookHolder, u *Update) (bool, error) {
	*o.buffer = append(*o.buffer, *u)
	if len(*o.buffer) < w.obBufferLimit {
		return false, nil
	}

	if w.sortBuffer {
		// sort by last updated to ensure each update is in order
		if w.sortBufferByUpdateIDs {
			sort.Slice(*o.buffer, func(i, j int) bool {
				return (*o.buffer)[i].UpdateID < (*o.buffer)[j].UpdateID
			})
		} else {
			sort.Slice(*o.buffer, func(i, j int) bool {
				return (*o.buffer)[i].UpdateTime.Before((*o.buffer)[j].UpdateTime)
			})
		}
	}
	for i := range *o.buffer {
		err := w.processObUpdate(o, &(*o.buffer)[i])
		if err != nil {
			return false, err
		}
	}
	// clear buffer of old updates
	*o.buffer = nil
	return true, nil
}

// processObUpdate processes updates either by its corresponding id or by
// price level
func (w *Orderbook) processObUpdate(o *orderbookHolder, u *Update) error {
	o.ob.LastUpdateID = u.UpdateID
	if w.updateEntriesByID {
		return o.updateByIDAndAction(u)
	}
	return o.updateByPrice(u)
}

// updateByPrice ammends amount if match occurs by price, deletes if amount is
// zero or less and inserts if not found.
func (o *orderbookHolder) updateByPrice(updts *Update) error {
askUpdates:
	for j := range updts.Asks {
		for target := range o.ob.Asks {
			if o.ob.Asks[target].Price == updts.Asks[j].Price {
				if updts.Asks[j].Amount == 0 {
					o.ob.Asks = append(o.ob.Asks[:target], o.ob.Asks[target+1:]...)
					continue askUpdates
				}
				o.ob.Asks[target].Amount = updts.Asks[j].Amount
				continue askUpdates
			}
		}
		if updts.Asks[j].Amount <= 0 {
			continue
		}
		insertAsk(updts.Asks[j], &o.ob.Asks)
		if updts.MaxDepth != 0 && len(o.ob.Asks) > updts.MaxDepth {
			o.ob.Asks = o.ob.Asks[:updts.MaxDepth]
		}
	}
bidUpdates:
	for j := range updts.Bids {
		for target := range o.ob.Bids {
			if o.ob.Bids[target].Price == updts.Bids[j].Price {
				if updts.Bids[j].Amount == 0 {
					o.ob.Bids = append(o.ob.Bids[:target], o.ob.Bids[target+1:]...)
					continue bidUpdates
				}
				o.ob.Bids[target].Amount = updts.Bids[j].Amount
				continue bidUpdates
			}
		}
		if updts.Bids[j].Amount <= 0 {
			continue
		}
		insertBid(updts.Bids[j], &o.ob.Bids)
		if updts.MaxDepth != 0 && len(o.ob.Bids) > updts.MaxDepth {
			o.ob.Bids = o.ob.Bids[:updts.MaxDepth]
		}
	}
	return nil
}

// updateByIDAndAction will receive an action to execute against the orderbook
// it will then match by IDs instead of price to perform the action
func (o *orderbookHolder) updateByIDAndAction(updts *Update) (err error) {
	switch updts.Action {
	case Amend:
		err = applyUpdates(updts.Bids, o.ob.Bids)
		if err != nil {
			return err
		}
		err = applyUpdates(updts.Asks, o.ob.Asks)
		if err != nil {
			return err
		}
	case Delete:
		// edge case for Bitfinex as their streaming endpoint duplicates deletes
		bypassErr := o.ob.ExchangeName == "Bitfinex" && o.ob.IsFundingRate
		err = deleteUpdates(updts.Bids, &o.ob.Bids, bypassErr)
		if err != nil {
			return fmt.Errorf("%s %s %v", o.ob.AssetType, o.ob.Pair, err)
		}
		err = deleteUpdates(updts.Asks, &o.ob.Asks, bypassErr)
		if err != nil {
			return fmt.Errorf("%s %s %v", o.ob.AssetType, o.ob.Pair, err)
		}
	case Insert:
		insertUpdatesBid(updts.Bids, &o.ob.Bids)
		insertUpdatesAsk(updts.Asks, &o.ob.Asks)
	case UpdateInsert:
	updateBids:
		for x := range updts.Bids {
			for target := range o.ob.Bids { // First iteration finds ID matches
				if o.ob.Bids[target].ID == updts.Bids[x].ID {
					if o.ob.Bids[target].Price != updts.Bids[x].Price {
						// Price change occurred so correct bid alignment is
						// needed - delete instance and insert into correct
						// price level
						o.ob.Bids = append(o.ob.Bids[:target], o.ob.Bids[target+1:]...)
						break
					}
					o.ob.Bids[target].Amount = updts.Bids[x].Amount
					continue updateBids
				}
			}
			insertBid(updts.Bids[x], &o.ob.Bids)
		}
	updateAsks:
		for x := range updts.Asks {
			for target := range o.ob.Asks {
				if o.ob.Asks[target].ID == updts.Asks[x].ID {
					if o.ob.Asks[target].Price != updts.Asks[x].Price {
						// Price change occurred so correct ask alignment is
						// needed - delete instance and insert into correct
						// price level
						o.ob.Asks = append(o.ob.Asks[:target], o.ob.Asks[target+1:]...)
						break
					}
					o.ob.Asks[target].Amount = updts.Asks[x].Amount
					continue updateAsks
				}
			}
			insertAsk(updts.Asks[x], &o.ob.Asks)
		}
	default:
		return fmt.Errorf("invalid action [%s]", updts.Action)
	}
	return nil
}

// applyUpdates amends amount by ID and returns an error if not found
func applyUpdates(updts, book []orderbook.Item) error {
updates:
	for x := range updts {
		for y := range book {
			if book[y].ID == updts[x].ID {
				book[y].Amount = updts[x].Amount
				continue updates
			}
		}
		return fmt.Errorf("update cannot be applied id: %d not found",
			updts[x].ID)
	}
	return nil
}

// deleteUpdates removes updates from orderbook and returns an error if not
// found
func deleteUpdates(updt []orderbook.Item, book *[]orderbook.Item, bypassErr bool) error {
updates:
	for x := range updt {
		for y := range *book {
			if (*book)[y].ID == updt[x].ID {
				*book = append((*book)[:y], (*book)[y+1:]...) // nolint:gocritic
				continue updates
			}
		}
		// bypassErr is for expected duplication from endpoint.
		if !bypassErr {
			return fmt.Errorf("update cannot be deleted id: %d not found",
				updt[x].ID)
		}
	}
	return nil
}

func insertAsk(updt orderbook.Item, book *[]orderbook.Item) {
	for target := range *book {
		if updt.Price < (*book)[target].Price {
			insertItem(updt, book, target)
			return
		}
	}
	*book = append(*book, updt)
}

func insertBid(updt orderbook.Item, book *[]orderbook.Item) {
	for target := range *book {
		if updt.Price > (*book)[target].Price {
			insertItem(updt, book, target)
			return
		}
	}
	*book = append(*book, updt)
}

// insertUpdatesBid inserts on **correctly aligned** book at price level
func insertUpdatesBid(updt []orderbook.Item, book *[]orderbook.Item) {
updates:
	for x := range updt {
		for target := range *book {
			if updt[x].Price > (*book)[target].Price {
				insertItem(updt[x], book, target)
				continue updates
			}
		}
		*book = append(*book, updt[x])
	}
}

// insertUpdatesBid inserts on **correctly aligned** book at price level
func insertUpdatesAsk(updt []orderbook.Item, book *[]orderbook.Item) {
updates:
	for x := range updt {
		for target := range *book {
			if updt[x].Price < (*book)[target].Price {
				insertItem(updt[x], book, target)
				continue updates
			}
		}
		*book = append(*book, updt[x])
	}
}

// insertItem inserts item in slice by target element this is an optimization
// to reduce the need for sorting algorithms
func insertItem(update orderbook.Item, book *[]orderbook.Item, target int) {
	// TODO: extend slice by incoming update length before this gets hit
	*book = append(*book, orderbook.Item{})
	copy((*book)[target+1:], (*book)[target:])
	(*book)[target] = update
}

// LoadSnapshot loads initial snapshot of ob data from websocket
func (w *Orderbook) LoadSnapshot(book *orderbook.Base) error {
	w.m.Lock()
	defer w.m.Unlock()

	err := book.Process()
	if err != nil {
		return err
	}

	m1, ok := w.ob[book.Pair.Base]
	if !ok {
		m1 = make(map[currency.Code]map[asset.Item]*orderbookHolder)
		w.ob[book.Pair.Base] = m1
	}
	m2, ok := m1[book.Pair.Quote]
	if !ok {
		m2 = make(map[asset.Item]*orderbookHolder)
		m1[book.Pair.Quote] = m2
	}
	m3, ok := m2[book.AssetType]
	if !ok {
		m3 = &orderbookHolder{ob: book, buffer: &[]Update{}}
		m2[book.AssetType] = m3
	} else {
		m3.ob.LastUpdateID = book.LastUpdateID
		m3.ob.Bids = book.Bids
		m3.ob.Asks = book.Asks
	}
	w.dataHandler <- book
	return nil
}

// GetOrderbook returns orderbook stored in current buffer
func (w *Orderbook) GetOrderbook(p currency.Pair, a asset.Item) *orderbook.Base {
	w.m.Lock()
	defer w.m.Unlock()
	ptr, ok := w.ob[p.Base][p.Quote][a]
	if !ok {
		return nil
	}
	cpy := *ptr.ob
	cpy.Asks = append(cpy.Asks[:0:0], cpy.Asks...)
	cpy.Bids = append(cpy.Bids[:0:0], cpy.Bids...)
	return &cpy
}

// FlushBuffer flushes w.ob data to be garbage collected and refreshed when a
// connection is lost and reconnected
func (w *Orderbook) FlushBuffer() {
	w.m.Lock()
	w.ob = make(map[currency.Code]map[currency.Code]map[asset.Item]*orderbookHolder)
	w.m.Unlock()
}

// FlushOrderbook flushes independent orderbook
func (w *Orderbook) FlushOrderbook(p currency.Pair, a asset.Item) error {
	w.m.Lock()
	defer w.m.Unlock()
	book, ok := w.ob[p.Base][p.Quote][a]
	if !ok {
		return fmt.Errorf("orderbook not associated with pair: [%s] and asset [%s]", p, a)
	}
	book.ob.Bids = nil
	book.ob.Asks = nil
	return nil
}