Skip to content

M
mogo
Commit 96b623d9 authored by luxq's avatar luxq
Browse files
Options

update test

parent abd2c29a
Hide whitespace changes
Inline Side-by-side
Showing with 18 additions and 16 deletions
db/workerinfo_test.go
View file @ 96b623d9
...@@ -17,7 +17,8 @@ import ( ...@@ -17,7 +17,8 @@ import (
) )
var ( var (
idlist = make([]string, 0, 1000000) maxModelId = 10000
idlist = make([]string, 0, 1000000)
//workers = make([]*DbWorkerInfo, 0, 1000000) //workers = make([]*DbWorkerInfo, 0, 1000000)
database = "test" database = "test"
collection = "workers" collection = "workers"
...@@ -197,7 +198,7 @@ func generateAHardware() *types.HardwareInfo { ...@@ -197,7 +198,7 @@ func generateAHardware() *types.HardwareInfo {
func generateAInstallModel() *types.InstalledModel { func generateAInstallModel() *types.InstalledModel {
return &types.InstalledModel{ return &types.InstalledModel{
ModelID: getRandId(100), ModelID: getRandId(maxModelId),
DiskSize: 101, DiskSize: 101,
InstalledTime: time.Now().Unix(), InstalledTime: time.Now().Unix(),
LastRunTime: time.Now().Unix(), LastRunTime: time.Now().Unix(),
...@@ -206,7 +207,7 @@ func generateAInstallModel() *types.InstalledModel { ...@@ -206,7 +207,7 @@ func generateAInstallModel() *types.InstalledModel {
func generateARunningModel() *types.RunningModel { func generateARunningModel() *types.RunningModel {
return &types.RunningModel{ return &types.RunningModel{
ModelID: getRandId(100), ModelID: getRandId(maxModelId),
GpuSeq: rand.Intn(3), GpuSeq: rand.Intn(3),
GpuRAM: generateAGpuRam(), GpuRAM: generateAGpuRam(),
StartedTime: time.Now().Unix(), StartedTime: time.Now().Unix(),
...@@ -217,16 +218,17 @@ func generateARunningModel() *types.RunningModel { ...@@ -217,16 +218,17 @@ func generateARunningModel() *types.RunningModel {
} }
func generateAModel() *types.ModelInfo { func generateAModel() *types.ModelInfo {
return &types.ModelInfo{ m := &types.ModelInfo{
InstalledModels: []*types.InstalledModel{ InstalledModels: make([]*types.InstalledModel, 0, 1000),
generateAInstallModel(), RunningModels: make([]*types.RunningModel, 0, 1000),
generateAInstallModel(), }
generateAInstallModel(), for i := 0; i < 500; i++ {
}, m.InstalledModels = append(m.InstalledModels, generateAInstallModel())
RunningModels: []*types.RunningModel{ if len(m.RunningModels) < 500 {
generateARunningModel(), m.RunningModels = append(m.RunningModels, generateARunningModel())
}, }
} }
return m
} }
func BenchmarkGenerateWorker(b *testing.B) { func BenchmarkGenerateWorker(b *testing.B) {
...@@ -393,7 +395,7 @@ func BenchmarkDbWorker_FindWorkerByInstallModelAndSortByGpuRam(b *testing.B) { ...@@ -393,7 +395,7 @@ func BenchmarkDbWorker_FindWorkerByInstallModelAndSortByGpuRam(b *testing.B) {
defer db.client.Disconnect(context.Background()) defer db.client.Disconnect(context.Background())
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
installedModelId := getRandId(100) installedModelId := getRandId(maxModelId)
performance := generateAGpuPerformance() performance := generateAGpuPerformance()
ram := generateAGpuRam() ram := generateAGpuRam()
if w, err := db.FindWorkerByInstallModelAndSortByGpuRam(context.Background(), installedModelId, performance, ram, 10); err != nil { if w, err := db.FindWorkerByInstallModelAndSortByGpuRam(context.Background(), installedModelId, performance, ram, 10); err != nil {
...@@ -413,7 +415,7 @@ func BenchmarkDbWorker_FindWorkerByInstallModelAndSortByGpuRam_Parallel(b *testi ...@@ -413,7 +415,7 @@ func BenchmarkDbWorker_FindWorkerByInstallModelAndSortByGpuRam_Parallel(b *testi
defer db.client.Disconnect(context.Background()) defer db.client.Disconnect(context.Background())
b.RunParallel(func(pb *testing.PB) { b.RunParallel(func(pb *testing.PB) {
for pb.Next() { for pb.Next() {
installedModelId := getRandId(100) installedModelId := getRandId(maxModelId)
performance := generateAGpuPerformance() performance := generateAGpuPerformance()
ram := generateAGpuRam() ram := generateAGpuRam()
if w, err := db.FindWorkerByInstallModelAndSortByGpuRam(context.Background(), installedModelId, performance, ram, 10); err != nil { if w, err := db.FindWorkerByInstallModelAndSortByGpuRam(context.Background(), installedModelId, performance, ram, 10); err != nil {
...@@ -434,7 +436,7 @@ func BenchmarkDbWorker_FindWorkerByRunningModelAndSortByWaitTime(b *testing.B) { ...@@ -434,7 +436,7 @@ func BenchmarkDbWorker_FindWorkerByRunningModelAndSortByWaitTime(b *testing.B) {
defer db.client.Disconnect(context.Background()) defer db.client.Disconnect(context.Background())
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
runningModelId := getRandId(100) runningModelId := getRandId(maxModelId)
if w, err := db.FindWorkerByRunningModelAndSortByWaitTime(context.Background(), runningModelId, 10); err != nil { if w, err := db.FindWorkerByRunningModelAndSortByWaitTime(context.Background(), runningModelId, 10); err != nil {
panic(fmt.Sprintf("find worker failed with err:%s", err)) panic(fmt.Sprintf("find worker failed with err:%s", err))
} else if len(w) == 0 { } else if len(w) == 0 {
...@@ -452,7 +454,7 @@ func BenchmarkDbWorker_FindWorkerByRunningModelAndSortByWaitTime_Parallel(b *tes ...@@ -452,7 +454,7 @@ func BenchmarkDbWorker_FindWorkerByRunningModelAndSortByWaitTime_Parallel(b *tes
defer db.client.Disconnect(context.Background()) defer db.client.Disconnect(context.Background())
b.RunParallel(func(pb *testing.PB) { b.RunParallel(func(pb *testing.PB) {
for pb.Next() { for pb.Next() {
runningModelId := getRandId(100) runningModelId := getRandId(maxModelId)
if w, err := db.FindWorkerByRunningModelAndSortByWaitTime(context.Background(), runningModelId, 10); err != nil { if w, err := db.FindWorkerByRunningModelAndSortByWaitTime(context.Background(), runningModelId, 10); err != nil {
panic(fmt.Sprintf("find worker failed with err:%s", err)) panic(fmt.Sprintf("find worker failed with err:%s", err))
} else if len(w) == 0 { } else if len(w) == 0 {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment