featurebase/dispatch/dispatch.go

package dispatch

import (
	"log"
	"pilosa/core"
	"pilosa/db"
	"pilosa/query"

	"github.com/davecgh/go-spew/spew"
)

type Dispatch struct {
	service *core.Service
}

func (self *Dispatch) Init() error {
	log.Println("Starting Dispatcher")
	return nil
}

func (self *Dispatch) Close() {
	log.Println("Shutting down Dispatcher")
}

// The Local Route

func (self *Dispatch) Run() {
	log.Println("Dispatch Run...")
	for {
		message := self.service.Transport.Receive()
		switch data := message.Data.(type) {
		case core.BatchRequest:
			response := db.Message{Data: core.BatchResponse{Id: data.Id}}
			self.service.Index.LoadBitmap(data.Fragment_id, data.Bitmap_id, data.Compressed_bitmap, data.Filter)
			self.service.Transport.Send(&response, data.Source)
		case core.BitsRequest:
			var results []core.SBResult
			for _, v := range data.Bits {
				result, _ := self.service.Index.SetBit(v.Fragment_id, v.Bitmap_id, v.Profile_id, uint64(v.Filter))
				//jbundle := core.SBResult{v.Bitmap_id, ''v.Frame, v.Filter, v.Profile_id, result}
				bundle := core.SBResult{v.Bitmap_id, v.Frame, v.Filter, v.Profile_id, result}
				results = append(results, bundle)
			}
			response := db.Message{Data: core.BitsResponse{Id: &data.QueryId, Items: results}}
			self.service.Transport.Send(&response, &data.ReturnProcessId)
		case core.PingRequest:
			pong := db.Message{Data: core.PongRequest{Id: data.Id}}
			self.service.Transport.Send(&pong, data.Source)
		case db.HoldResult:
			self.service.Hold.Set(data.ResultId(), data.ResultData(), 30)
		case query.PortableQueryStep:
			go self.service.Executor.NewJob(message)
		case core.TopFill:
			go self.service.TopFillHandler(message)
		case core.BitsResponse:
			self.service.Hold.Set(data.ResultId(), data.ResultData(), 30)
		default:
			println("Dispatch Unhandled")
			spew.Dump(data)
			log.Println("Unprocessed message", data)
		}
	}
}

func NewDispatch(service *core.Service) *Dispatch {
	return &Dispatch{service}
}