From 2b0b848fba6510c805ff4d6f36d1da553cbca5f0 Mon Sep 17 00:00:00 2001 From: Siddhartha Basu Date: Wed, 26 Sep 2018 23:07:32 +0000 Subject: [PATCH] feat: add new method to generate random string --- generate/rand.go | 28 ++++++++++++++++++++++++++++ storage/arangodb/datasource.go | 6 +++++- 2 files changed, 33 insertions(+), 1 deletion(-) create mode 100644 generate/rand.go diff --git a/generate/rand.go b/generate/rand.go new file mode 100644 index 0000000..8b7d411 --- /dev/null +++ b/generate/rand.go @@ -0,0 +1,28 @@ +package generate + +import ( + "math/rand" + "time" +) + +const ( + charSet = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" +) + +var seedRand *rand.Rand = rand.New(rand.NewSource(time.Now().UnixNano())) + +func stringWithCharset(length int, charset string) string { + var b []byte + for i := 0; i < length; i++ { + b = append( + b, + charset[seedRand.Intn(len(charset))], + ) + } + return string(b) +} + +// RandString generates a random string of a specific length +func RandString(length int) string { + return stringWithCharset(length, charSet) +} diff --git a/storage/arangodb/datasource.go b/storage/arangodb/datasource.go index 7c21ba1..774aa84 100644 --- a/storage/arangodb/datasource.go +++ b/storage/arangodb/datasource.go @@ -182,7 +182,11 @@ func (a *arangoSource) UpdateOboGraphInfo(g graph.OboGraph) error { } func (a *arangoSource) SaveOrUpdateTerms(g graph.OboGraph) (int, int, error) { - return 0, 0, nil + id, err := a.graphDocId(g) + if err != nil { + return 0, 0, err + } + return 1, 1, nil } func (a *arangoSource) SaveNewRelationships(g graph.OboGraph) (int, error) {