LevelDB性能测试|Golang调用LevelDB

不同方式使用压力测试

  1. 用ssdb,TCP连接方式调用,底层存储levelDB
  2. 直接调用Cgo的levelDB (必须保证串行)
  3. 直接调用Golang的LevelDB (必须保证串行)

开始:

go test  -v -test.run="DB.*" -test.bench="DB.*" -test.count=1 -test.benchtime=3s
go test  -v -test.run="Raws.*" -test.bench="Raws.*" -test.count=1 -test.benchtime=3s
go test  -v -test.run="Normal.*" -test.bench="Normal.*" -test.count=1 -test.benchtime=3s

性能对比:

1.调用SSDB:随机读写,和顺序读写差异不大,网络延迟是主要问题。

goos: linux
goarch: amd64
pkg: common/ssdb
Benchmark_DBSXSSDBSET     	   30000	    127546 ns/op
Benchmark_DBSXSSDBGET     	   50000	    118855 ns/op
Benchmark_DBRandomSSDBSET 	   30000	    128268 ns/op
Benchmark_DBRandomSSDBGET 	   50000	    119668 ns/op
PASS
ok  	common/ssdb	24.545s

2.直接调用LevelDB,每次都打开文件,导致特别慢:

goos: linux
goarch: amd64
pkg: common/ssdb
Benchmark_RawsSXSET     	     100	  38748277 ns/op
Benchmark_RawsSXGET     	     200	  27432834 ns/op
Benchmark_RawsRandomSET 	     100	  39496210 ns/op
Benchmark_RawsRandomGET 	     200	  27987023 ns/op
PASS
ok  	common/ssdb	24.637s

改为只打开一次文件:

goos: linux
goarch: amd64
pkg: common/ssdb
Benchmark_RawsSXSET     	 1000000	      3142 ns/op
Benchmark_RawsSXGET     	 3000000	      1856 ns/op
Benchmark_RawsRandomSET 	 1000000	      3892 ns/op
Benchmark_RawsRandomGET 	 3000000	      1467 ns/op
PASS
ok  	common/ssdb	24.073s
  1. 调用Golang LevelDB
goos: linux
goarch: amd64
pkg: common/ssdb
Benchmark_NormalSXSET     	     500	   8888019 ns/op
Benchmark_NormalSXGET     	  500000	      6632 ns/op
Benchmark_NormalRandomSET 	     500	   9006333 ns/op
Benchmark_NormalRandomGET 	  500000	      6449 ns/op
PASS
ok  	common/ssdb	17.501s

Golang调用Cgo LevelDB

levelDB Golang实现的库性能有问题?那么Golang调用C++ levelDB库:

环境准备:

需要先升级cmake,要求版本3.9以上

wget https://cmake.org/files/v3.9/cmake-3.9.2.tar.gz
tar xvf cmake-3.9.2.tar.gz
cd cmake-3.9.2
./bootstrap --prefix=/usr
make
sudo make install
cmake --version

下载并编译levelDB C库:

git clone https://github.com/google/leveldb
cd leveldb
git checkout v1.20
make

动态库在:

ls out-shared/libleveldb.so.1.20

静态库在:

ls out-static/libleveldb.a

安装:

# cp leveldb header file
sudo cp -r include/ /usr/include/

# cp lib to /usr/lib/
sudo cp out-shared/libleveldb.so.1.20 /usr/lib/

# create link
sudo ln -s /usr/lib/libleveldb.so.1.20 /usr/lib/libleveldb.so.1
sudo ln -s /usr/lib/libleveldb.so.1.20 /usr/lib/libleveldb.so

# update lib cache
sudo ldconfig

ls /usr/lib/libleveldb.so*

# 显示下面 3 个文件即安装成功
/usr/lib/libleveldb.so.1.20
/usr/lib/libleveldb.so.1
/usr/lib/libleveldb.so

我们先用C++来写一个程序先/root/test.cc

#include <iostream>
#include <cassert>
#include <cstdlib>
#include <string>
// 包含必要的头文件
#include <leveldb/db.h>

using namespace std;

int main(void)
{
    leveldb::DB *db = nullptr;
    leveldb::Options options;

    // 如果数据库不存在就创建
    options.create_if_missing = true;

    // 创建的数据库在 /tmp/testdb
    leveldb::Status status = leveldb::DB::Open(options, "/tmp/testdb", &db);
    assert(status.ok());

    std::string key = "A";
    std::string value = "a";
    std::string get_value;

    // 写入 key1 -> value1
    leveldb::Status s = db->Put(leveldb::WriteOptions(), key, value);

    // 写入成功,就读取 key:people 对应的 value
    if (s.ok())
        s = db->Get(leveldb::ReadOptions(), "A", &get_value);

    // 读取成功就输出
    if (s.ok())
        cout << get_value << endl;
    else
        cout << s.ToString() << endl;

    delete db;

    return 0;
}

静态编译:

cp out-static/libleveldb.a /root/
g++ test.cc -o test ./libleveldb.a -lpthread

动态:

g++ test.cc -o test -lpthread -lleveldb

接下来参考:https://github.com/jmhodges/levigo/blob/master/examples/comparator_example.go

下载后使用即可:

CGO_CFLAGS="-I/usr/include" CGO_LDFLAGS="-L/usr/lib" go get github.com/jmhodges/levigo

Golang调用ssdb

使用ssdb,用tcp方式请求,底层存储为levelDB.

wget --no-check-certificate https://github.com/ideawu/ssdb/archive/master.zip
unzip master.zip
cd ssdb-master
apt install autoconf
make
# 将安装在 /usr/local/ssdb 目录下
sudo make install

mkdir /root/ssdb
cp ssdb-server /root/ssdb
cp ssdb.conf /root/ssdb
cd /root/ssdb
mkdir var

编辑ssdb.conf

# ssdb-server config
# MUST indent by TAB!

# absolute path, or relative to path of this file, directory must exists
work_dir = ./var
pidfile = ./var/ssdb.pid

server:
        ip: 0.0.0.0
        port: 8888
        # bind to public ip
        #ip: 0.0.0.0
        # format: allow|deny: all|ip_prefix
        # multiple allows or denys is supported
        #deny: all
        #allow: 127.0.0.1
        #allow: 192.168
        # auth password must be at least 32 characters
        #auth: very-strong-password
        #readonly: yes
        # in ms, to log slowlog with WARN level
        #slowlog_timeout: 5

replication:
        binlog: yes
        # Limit sync speed to *MB/s, -1: no limit
        sync_speed: -1
        slaveof:
                # to identify a master even if it moved(ip, port changed)
                # if set to empty or not defined, ip:port will be used.
                #id: svc_2
                # sync|mirror, default is sync
                #type: sync
                #host: localhost
                #port: 8889

logger:
        level: debug
        output: /root/ssdb/log.txt
        rotate:
                size: 1000000000

leveldb:
        # in MB
        cache_size: 500
        # in MB
        write_buffer_size: 64
        # in MB/s
        compaction_speed: 1000
        # yes|no
        compression: yes

启动:

# 启动主库, 此命令会阻塞住命令行
./ssdb-server ssdb.conf

# 或者启动为后台进程(不阻塞命令行)
./ssdb-server -d ssdb.conf

# 停止 ssdb-server
./ssdb-server ssdb.conf -s stop
# 对于旧版本
kill `cat ./var/ssdb.pid`

# 重启
./ssdb-server ssdb.conf -s restart

下载后使用:

go get -v github.com/seefan/gossdb

代码:

Golang levelDB:

package bench

import (
	"github.com/syndtr/goleveldb/leveldb"
	"github.com/syndtr/goleveldb/leveldb/opt"
	"sync"
)

type LevelDBClient struct {
	db *leveldb.DB
	*sync.Mutex
}

func (x *LevelDBClient) init(dir string) error {
	//os.Mkdir(dir, 777)
	db, err := leveldb.OpenFile(dir, nil)
	if err != nil {
		return err
	}
	x.Mutex = new(sync.Mutex)
	x.db = db
	return nil
}

func (x *LevelDBClient) Set(key, value []byte) error {
	x.Lock()
	defer x.Unlock()
	return x.db.Put(key, value, &opt.WriteOptions{NoWriteMerge: true, Sync: true})
}

func (x *LevelDBClient) Get(key []byte) ([]byte, error) {
	x.Lock()
	defer x.Unlock()
	return x.db.Get(key, &opt.ReadOptions{DontFillCache: true})
}

Cgo levelDB:

package bench

import (
	"github.com/jmhodges/levigo"
	"sync"
)

type RawLevelDBClient struct {
	db *levigo.DB
	ro *levigo.ReadOptions
	wb *levigo.WriteOptions

	*sync.Mutex
}

func (x *RawLevelDBClient) init(dir string) error {
	opts := levigo.NewOptions()
	//opts.SetCache(levigo.NewLRUCache(3 << 30))
	opts.SetCreateIfMissing(true)
	db, err := levigo.Open(dir, opts)

	if err != nil {
		return err
	}

	x.Mutex = new(sync.Mutex)
	x.db = db
	x.ro = levigo.NewReadOptions()
	x.wb = levigo.NewWriteOptions()
	return nil
}

func (x *RawLevelDBClient) Get(key []byte) (data []byte, err error) {
	x.Lock()
	defer x.Unlock()
	data, err = x.db.Get(x.ro, key)

	if err != nil {
		return
	}
	return
}

func (x *RawLevelDBClient) Set(key []byte, value []byte) (err error) {
	x.Lock()
	defer x.Unlock()
	err = x.db.Put(x.wb, key, value)
	return
}

测试文件(三种方式):

package bench

import (
	"testing"
	"github.com/seefan/gossdb"
	"github.com/seefan/gossdb/conf"
	"fmt"
	"math/rand"
)

func Benchmark_DBSXSSDBSET(b *testing.B) {
	pool, err := gossdb.NewPool(&conf.Config{
		Host:             "192.168.153.15",
		Port:             8888,
		MinPoolSize:      5,
		MaxPoolSize:      50,
		AcquireIncrement: 5,
		RetryEnabled:     true,
	})
	if err != nil {
		fmt.Println("xxx")
		return
	}
	defer pool.Close()

	for i := 0; i < b.N; i++ {
		c, err := pool.NewClient()
		if err != nil {
			if c != nil {
				c.Close()
			}
			fmt.Println(err.Error())
			return
		}
		err = c.Set(fmt.Sprintf("%d", i), fmt.Sprintf("%d", i))
		if err != nil {
			fmt.Println(err.Error())
		}
		if c != nil {
			c.Close()
		}
	}
}

func Benchmark_DBSXSSDBGET(b *testing.B) {
	pool, err := gossdb.NewPool(&conf.Config{
		Host:             "192.168.153.15",
		Port:             8888,
		MinPoolSize:      5,
		MaxPoolSize:      50,
		AcquireIncrement: 5,
		RetryEnabled:     true,
	})
	if err != nil {
		fmt.Println("xxx")
		return
	}
	defer pool.Close()

	for i := 0; i < b.N; i++ {
		c, err := pool.NewClient()
		if err != nil {
			if c != nil {
				c.Close()
			}
			fmt.Println(err.Error())
			return
		}
		_, err = c.Get(fmt.Sprintf("%d", i))
		if err != nil {
			fmt.Println(err.Error())
		}
		if c != nil {
			c.Close()
		}
	}
}

func Benchmark_DBRandomSSDBSET(b *testing.B) {
	pool, err := gossdb.NewPool(&conf.Config{
		Host:             "192.168.153.15",
		Port:             8888,
		MinPoolSize:      5,
		MaxPoolSize:      50,
		AcquireIncrement: 5,
		RetryEnabled:     true,
	})
	if err != nil {
		fmt.Println("xxx")
		return
	}
	defer pool.Close()
	for i := 0; i < b.N; i++ {
		c, err := pool.NewClient()
		if err != nil {
			if c != nil {
				c.Close()
			}
			fmt.Println(err.Error())
			return
		}
		dudu := uint(rand.Intn(5))
		err = c.Set(fmt.Sprintf("%d", dudu), fmt.Sprintf("%d", dudu))
		if err != nil {
			fmt.Println(err.Error())
		}
		if c != nil {
			c.Close()
		}
	}
}

func Benchmark_DBRandomSSDBGET(b *testing.B) {
	pool, err := gossdb.NewPool(&conf.Config{
		Host:             "192.168.153.15",
		Port:             8888,
		MinPoolSize:      5,
		MaxPoolSize:      50,
		AcquireIncrement: 5,
		RetryEnabled:     true,
	})
	if err != nil {
		fmt.Println("xxx")
		return
	}
	defer pool.Close()
	for i := 0; i < b.N; i++ {
		c, err := pool.NewClient()
		if err != nil {
			if c != nil {
				c.Close()
			}
			fmt.Println(err.Error())
			return
		}
		_, err = c.Get(fmt.Sprintf("%d", uint(rand.Intn(5))))
		if err != nil {
			fmt.Println(err.Error())
		}
		if c != nil {
			c.Close()
		}
	}
}

var ldb = new(RawLevelDBClient)
var db = new(LevelDBClient)

func init() {
	dir := "/data/leveldb"
	err := ldb.init(dir)
	if err != nil {
		panic(err)
	}

	dir = "/data/leveldb1"
	err = db.init(dir)
	if err != nil {
		panic(err)
	}

}

func Benchmark_RawsSXSET(b *testing.B) {
	for i := 0; i < b.N; i++ {
		dudu := fmt.Sprintf("%d", i)
		err := ldb.Set([]byte(dudu), []byte(dudu))
		if err != nil {
			fmt.Println(err.Error())
		}
	}
}
func Benchmark_RawsSXGET(b *testing.B) {

	for i := 0; i < b.N; i++ {
		dudu := fmt.Sprintf("%d", i)
		_, err := ldb.Get([]byte(dudu))
		if err != nil {
			fmt.Println(err.Error())
		} else {
			//fmt.Println(string(v))
		}
	}
}
func Benchmark_RawsRandomSET(b *testing.B) {
	for i := 0; i < b.N; i++ {
		dudu := fmt.Sprintf("%d", uint(rand.Intn(5)))
		err := ldb.Set([]byte(dudu), []byte(dudu))
		if err != nil {
			fmt.Println(err.Error())
		}
	}
}
func Benchmark_RawsRandomGET(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := ldb.Get([]byte(fmt.Sprintf("%d", uint(rand.Intn(5)))))
		if err != nil {
			//fmt.Println(err.Error())
		} else {
			//fmt.Println(string(v))
		}
	}
}

func Benchmark_NormalSXSET(b *testing.B) {
	for i := 0; i < b.N; i++ {
		dudu := fmt.Sprintf("%d", i)
		err := db.Set([]byte(dudu), []byte(dudu))
		if err != nil {
			//fmt.Println(err.Error())
		} else {

		}
	}
}
func Benchmark_NormalSXGET(b *testing.B) {

	for i := 0; i < b.N; i++ {
		dudu := fmt.Sprintf("%d", i)
		_, err := db.Get([]byte(dudu))
		if err != nil {
			//fmt.Println(err.Error())
		} else {
			//fmt.Println(string(v))
		}
	}
}
func Benchmark_NormalRandomSET(b *testing.B) {
	for i := 0; i < b.N; i++ {
		dudu := fmt.Sprintf("%d", uint(rand.Intn(5)))
		err := db.Set([]byte(dudu), []byte(dudu))
		if err != nil {
			//fmt.Println(err.Error())
		}
	}
}
func Benchmark_NormalRandomGET(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, err := db.Get([]byte(fmt.Sprintf("%d", uint(rand.Intn(5)))))
		if err != nil {
			//fmt.Println(err.Error())
		} else {
			//fmt.Println(string(v))
		}
	}
}