efn阿克说 发表于 2017-12-12 17:31:37

InfluxDB和MySQL的读写对比测试

package main  

  
import (
  

"log"  
"time"
  
"fmt"
  
"math/rand"
  
"github.com/influxdata/influxdb/client/v2"
  
)
  

  
const (
  
MyDB = "testInfluxdb"
  
username = "root"
  
password = ""
  
)
  

  
func queryDB(clnt client.Client, cmd string) (res []client.Result, err error) {
  
q := client.Query{
  
Command:cmd,
  
Database: MyDB,
  
}
  
if response, err := clnt.Query(q); err == nil {
  
if response.Error() != nil {
  
return res, response.Error()
  
}
  
res = response.Results
  
} else {
  
return res, err
  
}
  
return res, nil
  
}
  

  
func writePoints(clnt client.Client,num int) {
  
sampleSize := 1 * 10000
  
rand.Seed(42)
  
t := num
  
bp, _ := client.NewBatchPoints(client.BatchPointsConfig{
  
Database:MyDB,
  
Precision: "us",
  
})
  

  
for i := 0; i < sampleSize; i++ {
  
t += 1
  
tags := mapstring{
  
"system_name": fmt.Sprintf("sys_%d",i%10),
  
"site_name":fmt.Sprintf("s_%d", (t+i) % 10),
  
"equipment_name":fmt.Sprintf("e_%d",t % 10),
  
}
  
fields := mapinterface{}{
  
"value" : fmt.Sprintf("%d",rand.Int()),
  
}
  
pt, err := client.NewPoint("monitorStatus", tags, fields,time.Now())
  
if err != nil {
  
log.Fatalln("Error: ", err)
  
}
  
bp.AddPoint(pt)
  
}
  

  
err := clnt.Write(bp)
  
if err != nil {
  
log.Fatal(err)
  
}
  

  
//fmt.Printf("%d task done\n",num)
  
}
  

  
func main() {
  
// Make client
  
c, err := client.NewHTTPClient(client.HTTPConfig{
  
Addr: "http://localhost:8086",
  
Username: username,
  
Password: password,
  
})
  

  
if err != nil {
  
log.Fatalln("Error: ", err)
  
}
  
_, err = queryDB(c, fmt.Sprintf("CREATE DATABASE %s", MyDB))
  
if err != nil {
  
log.Fatal(err)
  
}
  

  
i := 1
  
for i <= 10000 {
  
defer writePoints(c,i)
  
//fmt.Printf("i=%d\n",i)
  
i += 1
  
}
  
//fmt.Printf("task done : i=%d \n",i)
  

  
}
页: [1]
查看完整版本: InfluxDB和MySQL的读写对比测试