httprunner/hrp/internal/boomer/message.go

package boomer

const (
	typeClientReady      = "client_ready"
	typeClientStopped    = "client_stopped"
	typeHeartbeat        = "heartbeat"
	typeSpawning         = "spawning"
	typeSpawningComplete = "spawning_complete"
	typeQuit             = "quit"
	typeException        = "exception"
)

type message interface {
}

type genericMessage struct {
	Type   string           `json:"type,omitempty"`
	Data   map[string]int64 `json:"data,omitempty"`
	NodeID string           `json:"node_id,omitempty"`
	Tasks  []byte           `json:"tasks,omitempty"`
}

func newGenericMessage(t string, data map[string]int64, nodeID string) (msg *genericMessage) {
	return &genericMessage{
		Type:   t,
		Data:   data,
		NodeID: nodeID,
	}
}

func newQuitMessage(nodeID string) (msg *genericMessage) {
	return &genericMessage{
		Type:   "quit",
		NodeID: nodeID,
	}
}

func newSpawnMessageToWorker(t string, data map[string]int64, tasks []byte) (msg *genericMessage) {
	return &genericMessage{
		Type:  t,
		Data:  data,
		Tasks: tasks,
	}
}

func newClientReadyMessageToMaster(nodeID string) (msg *genericMessage) {
	return &genericMessage{
		Type:   "client_ready",
		NodeID: nodeID,
	}
}