Exemplo n.º 1
0
func (m *MessageHandler) prepareForStorage(o *blob.Blob) error {
	if o.Type != github.EvtPullRequest || o.HasAttribute(LabelsAttribute) {
		return nil
	}
	number := o.Data.Get("number").MustInt()
	log.Debugf("fetching labels for %s #%d", m.repo.PrettyName(), number)
	l, _, err := m.client.Issues.ListLabelsByIssue(m.repo.User, m.repo.Repo, number, &gh.ListOptions{})
	if err != nil {
		return fmt.Errorf("retrieve labels for issue %d: %v", number, err)
	}

	// TODO This is terrible
	var b bytes.Buffer
	var d []interface{}
	if err := json.NewEncoder(&b).Encode(l); err != nil {
		return fmt.Errorf("serializing labels: %v", err)
	}
	if err := json.Unmarshal(b.Bytes(), &d); err != nil {
		return fmt.Errorf("unserializing labels: %v", err)
	}

	o.Push(LabelsAttribute, d)
	return nil
}
Exemplo n.º 2
0
// Index stores the blob into the specified storage under the provided id for
// a given repository.
func (b *simpleBlobStore) Store(storage Storage, repo *Repository, blob *blob.Blob) error {
	switch storage {
	// Live is an index containing the webhook events. In this particular case,
	// we use the delivery id as the document index.
	//
	// When storing a live event, we always update the next two indices.
	case StoreLiveEvent:
		liveIndex := repo.LiveIndexForTimestamp(blob.Timestamp)
		log.Debugf("store live event to %s/%s/%s", liveIndex, blob.Type, blob.ID)
		if err := b.indexer.Index(liveIndex, blob); err != nil {
			return fmt.Errorf("store live event %s data: %v", blob.ID, err)
		}
		// Before falling through, replace the blob with the snapshot data from
		// the event, if any.
		if blob = blob.Snapshot(); blob == nil {
			return nil
		}
		fallthrough
	// Current state is an index containing the last version of items at a
	// given moment in time, and is updated at a frequency configured by the
	// user.
	//
	// When storing a current state, we always update the next index.
	case StoreCurrentState:
		stateIndex := repo.StateIndexForTimestamp(blob.Timestamp)
		log.Debugf("store current state to %s/%s/%s", stateIndex, blob.Type, blob.ID)
		if err := b.indexer.Index(stateIndex, blob); err != nil {
			return fmt.Errorf("store current state %s data: %v", blob.ID, err)
		}
		fallthrough
	// Snapshot is an index containing the last version of all items, opened or
	// closed.
	case StoreSnapshot:
		log.Debugf("store snapshot to %s/%s/%s", repo.SnapshotIndex(), blob.Type, blob.ID)
		if err := b.indexer.Index(repo.SnapshotIndex(), blob); err != nil {
			return fmt.Errorf("store snapshot %s data: %v", blob.ID, err)
		}
	}
	return nil
}