package discovery

import (
	"sync"

	"git.loafle.net/overflow_scanner/probe/discovery/types"
)

type ofDiscoveryMessage struct {
	request     types.DiscoveryRequest
	messageType types.DiscoveryMessageType
	data        interface{}
	err         error
}

func (dm *ofDiscoveryMessage) Request() types.DiscoveryRequest {
	return dm.request
}

func (dm *ofDiscoveryMessage) Type() types.DiscoveryMessageType {
	return dm.messageType
}

func (dm *ofDiscoveryMessage) Data() interface{} {
	return dm.data
}

func (dm *ofDiscoveryMessage) Error() error {
	return dm.err
}

func (dm *ofDiscoveryMessage) Release() {
	releaseDiscoveryMessage(dm)
}

func RetainDiscoveryMessage(request types.DiscoveryRequest, messageType types.DiscoveryMessageType, data interface{}, err error) *ofDiscoveryMessage {
	return retainDiscoveryMessage(request, messageType, data, err)
}

func ReleaseDiscoveryMessage(dm *ofDiscoveryMessage) {
	releaseDiscoveryMessage(dm)
}

var discoveryMessagePool sync.Pool

func retainDiscoveryMessage(request types.DiscoveryRequest, messageType types.DiscoveryMessageType, data interface{}, err error) *ofDiscoveryMessage {
	v := discoveryMessagePool.Get()
	var dm *ofDiscoveryMessage
	if v == nil {
		dm = &ofDiscoveryMessage{}
	} else {
		dm = v.(*ofDiscoveryMessage)
	}

	dm.request = request
	dm.messageType = messageType
	dm.data = data
	dm.err = err

	return dm
}

func releaseDiscoveryMessage(dm *ofDiscoveryMessage) {
	dm.request = nil
	dm.messageType = types.DiscoveryMessageTypeNone
	dm.data = nil
	dm.err = nil

	discoveryMessagePool.Put(dm)
}