package core

import (
	"encoding/gob"
	"pilosa/db"
	"pilosa/util"
	"time"
)

type PingRequest struct {
	Id     *util.GUID
	Source *util.GUID
}

type PongRequest struct {
	Id *util.GUID
}

func (self PongRequest) ResultId() *util.GUID {
	return self.Id
}
func (self PongRequest) ResultData() interface{} {
	return self.Id
}

func init() {
	gob.Register(PingRequest{})
	gob.Register(PongRequest{})
}

func (self *Service) Ping(process_id *util.GUID) (*time.Duration, error) {
	id := util.RandomUUID()
	ping := db.Message{Data: PingRequest{Id: &id, Source: self.Id}}
	start := time.Now()
	self.Transport.Send(&ping, process_id)
	_, err := self.Hold.Get(&id, 60)
	if err != nil {
		return nil, err
	}
	end := time.Now()
	dur := end.Sub(start)
	return &dur, nil
}