示例#1
0
文件: job.go 项目: funkygao/gafka
func (this *Client) AddJob(payload []byte, delay string, opt PubOption) (jobId string, err error) {
	buf := mpool.BytesBufferGet()
	defer mpool.BytesBufferPut(buf)

	buf.Reset()
	buf.Write(payload)

	var req *http.Request
	var u url.URL
	u.Scheme = this.cf.Pub.Scheme
	u.Host = this.cf.Pub.Endpoint
	u.Path = fmt.Sprintf("/v1/jobs/%s/%s", opt.Topic, opt.Ver)
	q := u.Query()
	q.Set("delay", delay)
	u.RawQuery = q.Encode()

	req, err = http.NewRequest("POST", u.String(), buf)
	if err != nil {
		return
	}

	req.Header.Set("AppId", this.cf.AppId)
	req.Header.Set("Pubkey", this.cf.Secret)

	var response *http.Response
	response, err = this.pubConn.Do(req)
	if err != nil {
		return
	}

	// TODO if 201, needn't read body
	var b []byte
	b, err = ioutil.ReadAll(response.Body)
	if err != nil {
		return
	}

	// reuse the connection
	response.Body.Close()

	if response.StatusCode != http.StatusCreated && response.StatusCode != http.StatusAccepted {
		return "", errors.New(string(b))
	}

	jobId = response.Header.Get(gateway.HttpHeaderJobId)

	if this.cf.Debug {
		log.Printf("--> [%s]", response.Status)
		log.Printf("JobId:%s", response.Header.Get(gateway.HttpHeaderJobId))
	}

	return
}
示例#2
0
文件: dummy.go 项目: funkygao/gafka
func (this *dummyStore) KafkaTopic(appid string, topic string, ver string) (r string) {
	b := mpool.BytesBufferGet()
	b.Reset()
	b.WriteString(appid)
	b.WriteString(".")
	b.WriteString(topic)
	b.WriteString(".")
	b.WriteString(ver)
	r = b.String()
	mpool.BytesBufferPut(b)
	return
}
示例#3
0
func BenchmarkManualCreateJson(b *testing.B) {
	for i := 0; i < b.N; i++ {
		buffer := mpool.BytesBufferGet()

		buffer.Reset()
		buffer.WriteString(`{"partition":`)
		buffer.WriteString(strconv.Itoa(int(6)))
		buffer.WriteString(`,"offset":`)
		buffer.WriteString(strconv.Itoa(int(7)))
		buffer.WriteString(`}`)

		mpool.BytesBufferPut(buffer)
	}
}
示例#4
0
func (this *WebhookExecutor) pushToEndpoint(msg *sarama.ConsumerMessage, uri string) (ok bool) {
	log.Debug("%s sending[%s] %s", this.topic, uri, string(msg.Value))

	if this.circuits[uri].Open() {
		log.Warn("%s %s circuit open", this.topic, uri)
		return false
	}

	body := mpool.BytesBufferGet()
	defer mpool.BytesBufferPut(body)

	body.Reset()
	body.Write(msg.Value)

	// TODO user defined post body schema, e,g. ElasticSearch
	req, err := http.NewRequest("POST", uri, body)
	if err != nil {
		this.circuits[uri].Fail()
		return false
	}

	req.Header.Set(gateway.HttpHeaderOffset, strconv.FormatInt(msg.Offset, 10))
	req.Header.Set(gateway.HttpHeaderPartition, strconv.FormatInt(int64(msg.Partition), 10))
	req.Header.Set("User-Agent", this.userAgent)
	req.Header.Set("X-App-Signature", this.appSignature)
	response, err := this.httpClient.Do(req)
	if err != nil {
		log.Error("%s %s %s", this.topic, uri, err)
		this.circuits[uri].Fail()
		return false
	}

	io.Copy(ioutil.Discard, response.Body)
	response.Body.Close()

	if response.StatusCode >= 300 {
		this.circuits[uri].Fail()
		log.Error("%s %s response: %s", this.topic, uri, http.StatusText(response.StatusCode))
		return
	}

	// audit
	log.Info("pushed %s/%d %d", this.topic, msg.Partition, msg.Offset)
	return true
}
示例#5
0
文件: api.go 项目: funkygao/gafka
func (this *mysqlStore) KafkaTopic(appid string, topic string, ver string) (r string) {
	b := mpool.BytesBufferGet()
	b.Reset()
	b.WriteString(appid)
	b.WriteByte('.')
	b.WriteString(topic)
	b.WriteByte('.')
	b.WriteString(ver)
	if len(ver) > 2 {
		// ver starts with 'v1', from 'v10' on, will use obfuscation
		b.WriteByte('.')

		// can't use app secret as part of cookie: what if user changes his secret?
		// FIXME user can guess the cookie if they know the algorithm in advance
		cookie := adler32.Checksum([]byte(appid + topic))
		b.WriteString(strconv.Itoa(int(cookie % 1000)))
	}
	r = b.String()
	mpool.BytesBufferPut(b)
	return
}
示例#6
0
func (this *Agent) Callback(endpoint string, msg *sarama.ConsumerMessage) error {
	buf := mpool.BytesBufferGet()
	defer mpool.BytesBufferPut(buf)

	buf.Reset()
	buf.Write(msg.Value)

	req, err := http.NewRequest("POST", endpoint, buf)
	if err != nil {
		return err
	}

	// setup http headers
	req.Header.Set("User-Agent", UserAgent)

	// invoke the callback over the wire
	response, err := this.callbackConn.Do(req)
	if err != nil {
		return err
	}

	b, err := ioutil.ReadAll(response.Body)
	if err != nil {
		return err
	}

	// reuse the connection
	response.Body.Close()

	if response.StatusCode != http.StatusCreated {
		return errors.New(string(b))
	}

	if this.debug {
		log.Debug("%s -> %s", endpoint, string(msg.Value))
	}

	return nil
}
示例#7
0
文件: pub.go 项目: funkygao/gafka
// Pub publish a keyed message to specified versioned topic.
func (this *Client) Pub(key string, msg []byte, opt PubOption) (err error) {
	buf := mpool.BytesBufferGet()
	defer mpool.BytesBufferPut(buf)

	buf.Reset()
	buf.Write(msg)

	var req *http.Request
	var u url.URL
	u.Scheme = this.cf.Pub.Scheme
	u.Host = this.cf.Pub.Endpoint
	u.Path = fmt.Sprintf("/v1/msgs/%s/%s", opt.Topic, opt.Ver)
	q := u.Query()
	q.Set("key", key)
	if opt.AckAll {
		q.Set("ack", "all")
	}
	if opt.Async {
		q.Set("async", "1")
	}
	u.RawQuery = q.Encode()

	req, err = http.NewRequest("POST", u.String(), buf)
	if err != nil {
		return
	}

	req.Header.Set(gateway.HttpHeaderAppid, this.cf.AppId)
	req.Header.Set(gateway.HttpHeaderPubkey, this.cf.Secret)
	if opt.Tag != "" {
		req.Header.Set(gateway.HttpHeaderMsgTag, opt.Tag)
	}

	var response *http.Response
	response, err = this.pubConn.Do(req)
	// when you get a redirection failure both response and err will be non-nil
	if response != nil {
		// reuse the connection
		defer response.Body.Close()
	}
	if err != nil {
		return
	}

	// TODO if 201, needn't read body
	var b []byte
	b, err = ioutil.ReadAll(response.Body)
	if err != nil {
		return
	}

	if response.StatusCode != http.StatusCreated && response.StatusCode != http.StatusAccepted {
		return errors.New(string(b))
	}

	if this.cf.Debug {
		log.Printf("--> [%s]", response.Status)
		log.Printf("Partition:%s Offset:%s",
			response.Header.Get(gateway.HttpHeaderPartition),
			response.Header.Get(gateway.HttpHeaderOffset))
	}

	return nil
}