initial commit

Author: gkorland

.circleci/config.yml

@@ -0,0 +1,38 @@
+# Golang CircleCI 2.0 configuration file
+#
+# Check https://circleci.com/docs/2.0/language-go/ for more details
+version: 2
+jobs:
+  build:
+    docker:
+      # specify the version
+      - image: circleci/golang:1.12
+
+      - image: redislabs/rebloom:edge
+        port: 6379:6379
+
+    working_directory: /go/src/github.com/RedisBloom/redisbloom-go
+    steps:
+      - checkout
+      - run: go get -v -t -d ./...
+
+      #run tests with coverage
+      - run: go test -v -race -coverprofile=coverage.txt -covermode=atomic
+      - run: go tool cover -func=coverage.txt
+      - run: bash <(curl -s https://codecov.io/bash) -t ${CODECOV_TOKEN}
+
+workflows:
+  version: 2
+  commit:
+    jobs:
+      - build
+  nightly:
+    triggers:
+      - schedule:
+          cron: "0 0 * * *"
+          filters:
+            branches:
+              only:
+                - master
+    jobs:
+      - build

.gitignore

@@ -13,3 +13,5 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+
+.project

Gopkg.lock

@@ -0,0 +1,39 @@
+# Gopkg.toml example
+#
+# Refer to https://golang.github.io/dep/docs/Gopkg.toml.html
+# for detailed Gopkg.toml documentation.
+#
+# required = ["github.com/user/thing/cmd/thing"]
+# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
+#
+# [[constraint]]
+#   name = "github.com/user/project"
+#   version = "1.0.0"
+#
+# [[constraint]]
+#   name = "github.com/user/project2"
+#   branch = "dev"
+#   source = "github.com/myfork/project2"
+#
+# [[override]]
+#   name = "github.com/x/y"
+#   version = "2.4.0"
+#
+# [prune]
+#   non-go = false
+#   go-tests = true
+#   unused-packages = true
+
+
+[[constraint]]
+  name = "github.com/gomodule/redigo"
+  version = "2.0.0"
+
+[[constraint]]
+  name = "github.com/stretchr/testify"
+  version = "1.3.0"
+
+[prune]
+  go-tests = true
+  unused-packages = true
+  

Gopkg.toml

@@ -0,0 +1,78 @@
+[![license](https://img.shields.io/github/license/RedisBloom/redisbloom-go.svg)](https://github.com/RedisBloom/redisbloom-go)
+[![CircleCI](https://circleci.com/gh/RedisBloom/redisbloom-go.svg?style=svg)](https://circleci.com/gh/RedisBloom/redisbloom-go)
+[![GitHub issues](https://img.shields.io/github/release/RedisBloom/redisbloom-go.svg)](https://github.com/RedisBloom/redisbloom-go/releases/latest)
+[![Codecov](https://codecov.io/gh/RedisBloom/redisbloom-go/branch/master/graph/badge.svg)](https://codecov.io/gh/RedisBloom/redisbloom-go)
+[![GoDoc](https://godoc.org/github.com/RedisBloom/redisbloom-go?status.svg)](https://godoc.org/github.com/RedisBloom/redisbloom-go)
+
+
+# redisbloom-go
+
+Go client for RedisBloom (https://github.com/RedisBloom/redisbloom), based on redigo.
+
+## Installing
+
+```sh
+$ go get github.com/RedisBloom/redisbloom-go
+```
+
+## Running tests
+
+A simple test suite is provided, and can be run with:
+
+```sh
+$ RedisBloom_TEST_PASSWORD="" go test
+```
+
+The tests expect a Redis server with the RedisBloom module loaded to be available at localhost:6379
+
+## Example Code
+
+```go
+package main 
+
+import (
+        "fmt"
+        RedisBloom "github.com/RedisBloom/redisbloom-go"
+)
+
+func main() {
+		// Connect to localhost with no password
+        var client = RedisBloom.NewClient("localhost:6379", "nohelp", nil)
+        var keyname = "mytest"
+        _, haveit := client.Info(keyname)
+        if haveit != nil {
+			client.CreateKeyWithOptions(keyname, RedisBloom.DefaultCreateOptions)
+			client.CreateKeyWithOptions(keyname+"_avg", RedisBloom.DefaultCreateOptions)
+			client.CreateRule(keyname, RedisBloom.AvgAggregation, 60, keyname+"_avg")
+        }
+		// Add sample with timestamp from server time and value 100
+        // TS.ADD mytest * 100 
+        _, err := client.AddAutoTs(keyname, 100)
+        if err != nil {
+                fmt.Println("Error:", err)
+        }
+}
+```
+
+## Supported RedisBloom Commands
+
+| Command | Recommended API and godoc  |
+| :---          |  ----: |
+| [TS.CREATE](https://oss.redislabs.com/RedisBloom/commands/#tscreate) |   [CreateKeyWithOptions](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.CreateKeyWithOptions)          |
+| [TS.ALTER](https://oss.redislabs.com/RedisBloom/commands/#tsalter) |   N/A          |
+| [TS.ADD](https://oss.redislabs.com/RedisBloom/commands/#tsadd) |   <ul><li>[Add](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.Add)</li><li>[AddAutoTs](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.AddAutoTs)</li><li>[AddWithOptions](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.AddWithOptions)</li><li>[AddAutoTsWithOptions](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.AddWithOptions)</li> </ul>          |
+| [TS.MADD](https://oss.redislabs.com/RedisBloom/commands/#tsmadd) |    N/A |
+| [TS.INCRBY/TS.DECRBY](https://oss.redislabs.com/RedisBloom/commands/#tsincrbytsdecrby) |    N/A         |
+| [TS.CREATERULE](https://oss.redislabs.com/RedisBloom/commands/#tscreaterule) |   [CreateRule](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.CreateRule)          |
+| [TS.DELETERULE](https://oss.redislabs.com/RedisBloom/commands/#tsdeleterule) |   [DeleteRule](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.DeleteRule)          |
+| [TS.RANGE](https://oss.redislabs.com/RedisBloom/commands/#tsrange) |   [RangeWithOptions](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.RangeWithOptions)          |
+| [TS.MRANGE](https://oss.redislabs.com/RedisBloom/commands/#tsmrange) |   [MultiRangeWithOptions](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.MultiRangeWithOptions)          |
+| [TS.GET](https://oss.redislabs.com/RedisBloom/commands/#tsget) |   [Get](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.Get)          |
+| [TS.MGET](https://oss.redislabs.com/RedisBloom/commands/#tsmget) |   [MultiGet](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.MultiGet)          |
+| [TS.INFO](https://oss.redislabs.com/RedisBloom/commands/#tsinfo) |   [Info](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.Info)          |
+| [TS.QUERYINDEX](https://oss.redislabs.com/RedisBloom/commands/#tsqueryindex) |    N/A |
+
+
+## License
+
+redisbloom-go is distributed under the Apache-2 license - see [LICENSE](LICENSE)

README.md

@@ -0,0 +1,54 @@
+package redis_timeseries_go
+
+import (
+	"strings"
+	"github.com/gomodule/redigo/redis"
+)
+
+// TODO: refactor this hard limit and revise client locking
+// Client Max Connections
+var maxConns = 500
+
+// Client is an interface to time series redis commands
+type Client struct {
+	Pool ConnPool
+	Name string
+}
+
+// NewClient creates a new client connecting to the redis host, and using the given name as key prefix.
+// Addr can be a single host:port pair, or a comma separated list of host:port,host:port...
+// In the case of multiple hosts we create a multi-pool and select connections at random
+func NewClient(addr, name string, authPass *string) *Client {
+	addrs := strings.Split(addr, ",")
+	var pool ConnPool
+	if len(addrs) == 1 {
+		pool = NewSingleHostPool(addrs[0], authPass)
+	} else {
+		pool = NewMultiHostPool(addrs, authPass)
+	}
+	ret := &Client{
+		Pool: pool,
+		Name: name,
+	}
+	return ret
+}
+
+// Add - Add (or create and add) a new value to the filter
+// args:
+// key - the name of the filter 
+// item - the item to add 
+func (client *Client) Add(key string, item string) (exists bool, err error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	return redis.Bool(conn.Do("BF.ADD", key, item))
+}
+
+// Exists - Determines whether an item may exist in the Bloom Filter or not. 
+// args:
+// key - the name of the filter 
+// item - the item to check for 
+func (client *Client) Exists(key string, item string) (exists bool, err error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	return redis.Bool(conn.Do("BF.EXISTS", key, item))
+}

client.go

@@ -0,0 +1,68 @@
+package redis_timeseries_go
+
+import (
+	"os"
+//	"reflect"
+	"testing"
+	"time"
+
+//	"github.com/gomodule/redigo/redis"
+	"github.com/stretchr/testify/assert"
+)
+
+func createClient() *Client {
+	valueh, exists := os.LookupEnv("REDISTIMESERIES_TEST_HOST")
+	host := "localhost:6379"
+	if exists && valueh != "" {
+		host = valueh
+	}
+	valuep, exists := os.LookupEnv("REDISTIMESERIES_TEST_PASSWORD")
+	password := "SUPERSECRET"
+	var ptr *string = nil
+	if exists {
+		password = valuep
+	}
+	if len(password) > 0 {
+		ptr = &password
+	}
+	return NewClient(host, "test_client", ptr)
+}
+
+var client = createClient()
+var _ = client.FlushAll()
+
+var defaultDuration, _ = time.ParseDuration("1h")
+var tooShortDuration, _ = time.ParseDuration("10ms")
+
+func (client *Client) FlushAll() (err error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	_, err = conn.Do("FLUSHALL")
+	return err
+}
+
+func TestAdd(t *testing.T) {
+	client.FlushAll()
+	key := "test_ADD"
+	value := "test_ADD_value";
+	exists, err := client.Add(key, value)
+	assert.Nil(t, err)
+	assert.True(t, exists)
+	
+	exists, err = client.Add(key, value)
+	assert.Nil(t, err)
+	assert.False(t, exists)
+}
+
+func TestExists(t *testing.T) {
+	client.FlushAll()
+	client.Add("test_ADD", "test_EXISTS")
+	
+	exists, err := client.Exists("test_ADD", "test_EXISTS")
+	assert.Nil(t, err)
+	assert.True(t, exists)
+	
+	exists, err = client.Exists("test_ADD", "test_EXISTS1")
+	assert.Nil(t, err)
+	assert.False(t, exists)
+}