Initial commit

.gitignore

+.idea

main.go

+package main
+
+import (
+	"encoding/hex"
+	"fmt"
+	"github.com/satori/go.uuid"
+	"github.com/tarantool/go-tarantool"
+	"math/rand"
+	"time"
+	"log"
+	"strings"
+	"sync"
+	"os"
+)
+
+func bin2hex(bin []byte) string {
+	hexData := make([]byte, len(bin)*2)
+	hex.Encode(hexData, bin)
+	return string(hexData)
+}
+
+func newVid() string {
+	u := uuid.NewV4()
+	return bin2hex(u[:])
+
+}
+func queryItems(conn *tarantool.Connection) []string {
+	result, err := conn.Select("visitorid_matching", "primary", 1000000, 500000, tarantool.IterAll, []interface{}{})
+	if err != nil {
+		panic(err)
+	}
+	var items []string
+	for _, row := range result.Data {
+		rowa := row.([]interface{})
+		items = append(items, rowa[2].(string))
+	}
+	return items
+}
+func genItems(items []string, cnt int) []string {
+	result := make([]string, cnt)
+	for i := range result {
+		if rand.Intn(2) == 0 {
+			result[i] = items[rand.Intn(len(items))]
+		} else {
+			result[i] = newVid()
+		}
+	}
+	return result
+}
+func GetTTConnn(url string) *tarantool.Connection {
+	opts := tarantool.Opts{
+		Reconnect: time.Second * 10,
+	}
+	parts := strings.SplitN(url, "@", 2)
+	var addr string
+	if len(parts) == 1 {
+		addr = parts[0]
+	} else {
+		addr = parts[1]
+		parts = strings.SplitN(parts[0], ":", 2)
+		opts.User = parts[0]
+		if len(parts) > 1 {
+			opts.Pass = parts[1]
+		}
+	}
+	ttConn, err := tarantool.Connect(addr, opts)
+	if err != nil {
+		log.Fatal(err)
+	}
+	return ttConn
+}
+
+func main() {
+	if len(os.Args) < 2 {
+		log.Fatal("Usage: tarantool_test url")
+	}
+	conn := GetTTConnn(os.Args[1])
+	if conn == nil {
+		log.Fatal("connection error")
+		return
+	}
+	var tt float64
+	var tmx float64
+	existingVids := queryItems(conn)
+	fmt.Println("Items queried")
+	items := genItems(existingVids, 100000)
+	for _, item := range items {
+		t1 := time.Now()
+		_, err := conn.Select("visitorid_matching", "primary", 0, 1, tarantool.IterEq, []interface{}{"host1", "host2", item})
+		if err != nil {
+			log.Fatal(err)
+		}
+		tm := time.Now().Sub(t1).Seconds()
+		if tm > tmx {
+			tmx = tm
+		}
+		tt += tm
+	}
+	fmt.Println("sync max =", tmx*1000)
+	fmt.Println("sync avg =", tt*1000/float64(len(items)))
+
+	items = genItems(existingVids, 100000)
+	tt = 0
+	tmx = 0
+	var trmx,trsum float64
+	type asyncItem struct {
+		T time.Time
+		F *tarantool.Future
+	}
+	fchan := make(chan asyncItem, 100000)
+	t1 := time.Now()
+	go func() {
+		for _, item := range items {
+			fchan <- asyncItem{time.Now(),conn.SelectAsync("visitorid_matching", "primary", 0, 1, tarantool.IterEq, []interface{}{"host1", "host2", item})}
+			time.Sleep(time.Microsecond*50)
+		}
+		close(fchan)
+	}()
+	t2 := time.Now()
+	for item := range fchan {
+		item.F.Get()
+		t3 := time.Now()
+		tm := t3.Sub(t2).Seconds()
+		if tm > tmx {
+			tmx = tm
+		}
+		t2 = t3
+		tm = t3.Sub(item.T).Seconds()
+		trsum += tm
+		if tm > trmx {
+			trmx = tm
+		}
+	}
+	fmt.Println("async max =", tmx*1000)
+	fmt.Println("async avg =", time.Now().Sub(t1).Seconds()*1000/float64(len(items)))
+	tt = 0
+	tmx = 0
+	items = genItems(existingVids, 100000)
+	var mu sync.Mutex
+	for _, item := range items {
+		go func() {
+			t1 := time.Now()
+			_, err := conn.Select("visitorid_matching", "primary", 0, 1, tarantool.IterEq, []interface{}{"host1", "host2", item})
+			if err != nil {
+				log.Fatal(err)
+			}
+			tm := time.Now().Sub(t1).Seconds()
+			mu.Lock()
+			if tm > tmx {
+				tmx = tm
+			}
+			tt += tm
+			mu.Unlock()
+		}()
+		time.Sleep(time.Millisecond)
+	}
+	fmt.Println("mixed max =", tmx*1000)
+	fmt.Println("mixed avg =", tt*1000/float64(len(items)))
+}