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
/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

下载后使用即可:

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))
}
}
}