func writeWithReverses(ctx context.Context, tbl *table.KVProto, src, kind string, targets []string) error {
if err := tbl.Put(ctx, []byte(src+tempTableKeySep+kind+tempTableKeySep), &srvpb.EdgeSet_Group{
Kind: kind,
TargetTicket: targets,
}); err != nil {
return fmt.Errorf("error writing edges group: %v", err)
}
revGroup := &srvpb.EdgeSet_Group{
Kind: schema.MirrorEdge(kind),
TargetTicket: []string{src},
}
for _, tgt := range targets {
if err := tbl.Put(ctx, []byte(tgt+tempTableKeySep+revGroup.Kind+tempTableKeySep+src), revGroup); err != nil {
return fmt.Errorf("error writing rev edges group: %v", err)
}
}
return nil
}
func createDecorationFragments(ctx context.Context, edges <-chan *srvpb.Edge, fragments *table.KVProto) error {
fdb := &assemble.DecorationFragmentBuilder{
Output: func(ctx context.Context, file string, fragment *srvpb.FileDecorations) error {
key := file + tempTableKeySep
if len(fragment.Decoration) != 0 {
key += fragment.Decoration[0].Anchor.Ticket
}
return fragments.Put(ctx, []byte(key), fragment)
},
}
for e := range edges {
if err := fdb.AddEdge(ctx, e); err != nil {
for range edges { // drain input channel
}
return err
}
}
return fdb.Flush(ctx)
}
func writeEdge(ctx context.Context, edges *table.KVProto, e *srvpb.Edge) error {
return edges.Put(ctx, []byte(e.Source.Ticket+tempTableKeySep+e.Kind+tempTableKeySep+e.Target.Ticket), e)
}
