Пример #1
0
func lookupPagedEdgeSets(ctx context.Context, tbl table.ProtoBatch, keys [][]byte) (<-chan edgeSetResult, error) {
	rs, err := tbl.LookupBatch(ctx, keys, (*srvpb.PagedEdgeSet)(nil))
	if err != nil {
		return nil, err
	}
	ch := make(chan edgeSetResult)
	go func() {
		defer close(ch)
		for r := range rs {
			if r.Err == table.ErrNoSuchKey {
				log.Printf("Could not locate edges with key %q", r.Key)
				ch <- edgeSetResult{Err: r.Err}
				continue
			} else if r.Err != nil {
				ticket := strings.TrimPrefix(string(r.Key), edgeSetsTablePrefix)
				ch <- edgeSetResult{
					Err: fmt.Errorf("edges lookup error (ticket %q): %v", ticket, r.Err),
				}
				continue
			}

			ch <- edgeSetResult{PagedEdgeSet: r.Value.(*srvpb.PagedEdgeSet)}
		}
	}()
	return ch, nil
}
Пример #2
0
func lookupNodes(ctx context.Context, tbl table.ProtoBatch, keys [][]byte) (<-chan nodeResult, error) {
	rs, err := tbl.LookupBatch(ctx, keys, (*srvpb.Node)(nil))
	if err != nil {
		return nil, err
	}
	ch := make(chan nodeResult)
	go func() {
		defer close(ch)
		for r := range rs {
			if r.Err == table.ErrNoSuchKey {
				log.Printf("Could not locate node with key %q", r.Key)
				ch <- nodeResult{Err: r.Err}
				continue
			} else if r.Err != nil {
				ticket := strings.TrimPrefix(string(r.Key), nodesTablePrefix)
				ch <- nodeResult{
					Err: fmt.Errorf("lookup error for node %q: %v", ticket, r.Err),
				}
				continue
			}

			ch <- nodeResult{Node: r.Value.(*srvpb.Node)}
		}
	}()
	return ch, nil
}