コード例 #1
0
ファイル: submit.go プロジェクト: banyue/periodic
func SubmitJob(entryPoint string, job driver.Job) {
	parts := strings.SplitN(entryPoint, "://", 2)
	c, err := net.Dial(parts[0], parts[1])
	if err != nil {
		log.Fatal(err)
	}
	conn := protocol.NewClientConn(c)
	defer conn.Close()
	err = conn.Send(protocol.TYPE_CLIENT.Bytes())
	if err != nil {
		log.Fatal(err)
	}
	var msgId = []byte("100")
	buf := bytes.NewBuffer(nil)
	buf.Write(msgId)
	buf.Write(protocol.NULL_CHAR)
	buf.WriteByte(byte(protocol.SUBMIT_JOB))
	buf.Write(protocol.NULL_CHAR)
	buf.Write(job.Bytes())
	err = conn.Send(buf.Bytes())
	if err != nil {
		log.Fatal(err)
	}
	payload, err := conn.Receive()
	if err != nil {
		log.Fatal(err)
	}
	_, cmd, _ := protocol.ParseCommand(payload)
	fmt.Printf("%s\n", cmd.String())
}
コード例 #2
0
ファイル: drop.go プロジェクト: 4honor/periodic
func DropFunc(entryPoint, Func string) {
	parts := strings.SplitN(entryPoint, "://", 2)
	c, err := net.Dial(parts[0], parts[1])
	if err != nil {
		log.Fatal(err)
	}
	conn := protocol.Conn{Conn: c}
	defer conn.Close()
	err = conn.Send(protocol.TYPE_CLIENT.Bytes())
	if err != nil {
		log.Fatal(err)
	}
	var msgId = []byte("100")
	buf := bytes.NewBuffer(nil)
	buf.Write(msgId)
	buf.Write(protocol.NULL_CHAR)
	buf.WriteByte(byte(protocol.DROP_FUNC))
	buf.Write(protocol.NULL_CHAR)
	buf.WriteString(Func)
	err = conn.Send(buf.Bytes())
	if err != nil {
		log.Fatal(err)
	}
	payload, err := conn.Receive()
	if err != nil {
		log.Fatal(err)
	}
	_, cmd, _ := protocol.ParseCommand(payload)
	fmt.Printf("%s\n", cmd)
}
コード例 #3
0
ファイル: client.go プロジェクト: monsterwof/periodic
func (c *client) handle() {
	var payload []byte
	var err error
	var msgID []byte
	var cmd protocol.Command
	var conn = c.conn
	defer func() {
		if x := recover(); x != nil {
			log.Printf("[client] painc: %v\n", x)
		}
	}()
	defer conn.Close()
	for {
		payload, err = conn.Receive()
		if err != nil {
			if err != io.EOF {
				log.Printf("clientError: %s\n", err.Error())
			}
			return
		}

		msgID, cmd, payload = protocol.ParseCommand(payload)

		switch cmd {
		case protocol.SUBMITJOB:
			err = c.handleSubmitJob(msgID, payload)
			break
		case protocol.STATUS:
			err = c.handleStatus(msgID)
			break
		case protocol.PING:
			err = c.handleCommand(msgID, protocol.PONG)
			break
		case protocol.DROPFUNC:
			err = c.handleDropFunc(msgID, payload)
			break
		case protocol.REMOVEJOB:
			err = c.handleRemoveJob(msgID, payload)
			break
		case protocol.DUMP:
			err = c.handleDump(msgID)
			break
		case protocol.LOAD:
			err = c.handleLoad(msgID, payload)
			break
		default:
			err = c.handleCommand(msgID, protocol.UNKNOWN)
			break
		}
		if err != nil {
			if err != io.EOF {
				log.Printf("clientError: %s\n", err.Error())
			}
			return
		}
	}
}
コード例 #4
0
ファイル: client.go プロジェクト: 4honor/periodic
func (client *Client) Handle() {
	var payload []byte
	var err error
	var msgId int64
	var cmd protocol.Command
	var conn = client.conn
	defer func() {
		if x := recover(); x != nil {
			log.Printf("[Client] painc: %v\n", x)
		}
	}()
	defer conn.Close()
	for {
		payload, err = conn.Receive()
		if err != nil {
			if err != io.EOF {
				log.Printf("ClientError: %s\n", err.Error())
			}
			return
		}

		msgId, cmd, payload = protocol.ParseCommand(payload)

		switch cmd {
		case protocol.SUBMIT_JOB:
			err = client.HandleSubmitJob(msgId, payload)
			break
		case protocol.STATUS:
			err = client.HandleStatus(msgId)
			break
		case protocol.PING:
			err = client.HandleCommand(msgId, protocol.PONG)
			break
		case protocol.DROP_FUNC:
			err = client.HandleDropFunc(msgId, payload)
			break
		default:
			err = client.HandleCommand(msgId, protocol.UNKNOWN)
			break
		}
		if err != nil {
			if err != io.EOF {
				log.Printf("ClientError: %s\n", err.Error())
			}
			return
		}
	}
}
コード例 #5
0
ファイル: worker.go プロジェクト: banyue/periodic
func (w *worker) handle() {
	var payload []byte
	var err error
	var conn = w.conn
	var msgId []byte
	var cmd protocol.Command
	defer func() {
		if x := recover(); x != nil {
			log.Printf("[worker] painc: %v\n", x)
		}
	}()
	defer w.Close()
	for {
		payload, err = conn.Receive()
		if err != nil {
			if err != io.EOF {
				log.Printf("workerError: %s\n", err.Error())
			}
			break
		}

		msgId, cmd, payload = protocol.ParseCommand(payload)

		switch cmd {
		case protocol.GRAB_JOB:
			err = w.handleGrabJob(msgId)
			break
		case protocol.WORK_DONE:
			jobId, _ := strconv.ParseInt(string(payload), 10, 0)
			err = w.handleDone(jobId)
			break
		case protocol.WORK_FAIL:
			jobId, _ := strconv.ParseInt(string(payload), 10, 0)
			err = w.handleFail(jobId)
			break
		case protocol.SCHED_LATER:
			parts := bytes.SplitN(payload, protocol.NULL_CHAR, 2)
			if len(parts) != 2 {
				log.Printf("Error: invalid format.")
				break
			}
			jobId, _ := strconv.ParseInt(string(parts[0]), 10, 0)
			delay, _ := strconv.ParseInt(string(parts[1]), 10, 0)
			err = w.handleSchedLater(jobId, delay)
			break
		case protocol.SLEEP:
			err = w.handleCommand(msgId, protocol.NOOP)
			break
		case protocol.PING:
			err = w.handleCommand(msgId, protocol.PONG)
			break
		case protocol.CAN_DO:
			err = w.handleCanDo(string(payload))
			break
		case protocol.CANT_DO:
			err = w.handleCanNoDo(string(payload))
			break
		default:
			err = w.handleCommand(msgId, protocol.UNKNOWN)
			break
		}
		if err != nil {
			if err != io.EOF {
				log.Printf("workerError: %s\n", err.Error())
			}
			break
		}

		if !w.alive {
			break
		}
	}
}