func newService(t *testing.T, entries []*spb.Entry) *GraphStoreService {
gs := inmemory.Create()
for req := range graphstore.BatchWrites(channelEntries(entries), 64) {
if err := gs.Write(ctx, req); err != nil {
t.Fatalf("Failed to write entries: %v", err)
}
}
return NewGraphStoreService(gs)
}
"strings"
"syscall"
"kythe.io/kythe/go/services/graphstore"
"kythe.io/kythe/go/storage/inmemory"
"golang.org/x/net/context"
)
// Handler returns a graphstore.Service based on the given specification.
// See also: Register(string, Handler).
type Handler func(spec string) (graphstore.Service, error)
var (
handlers = map[string]Handler{
"in-memory": func(_ string) (graphstore.Service, error) { return inmemory.Create(), nil },
}
defaultHandlerKind string
)
// Register exposes the given Handler to ParseGraphStore. Each string starting
// with kind+":" will be passed to the given Handler. A kind can only be
// registered once.
func Register(kind string, h Handler) {
if _, exists := handlers[kind]; exists {
log.Fatalf("gsutil Handler for kind %q already exists", kind)
}
handlers[kind] = h
}
// RegisterDefault gives ParseGraphStore a fallback kind if not given any