1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
package com.yahoo.vespa.hosted.node.verification.hardware.benchmarks;
import com.yahoo.vespa.hosted.node.verification.commons.CommandExecutor;
import com.yahoo.vespa.hosted.node.verification.commons.parser.OutputParser;
import com.yahoo.vespa.hosted.node.verification.commons.parser.ParseInstructions;
import com.yahoo.vespa.hosted.node.verification.commons.parser.ParseResult;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* Created by olaa on 10/07/2017.
*/
public class DiskBenchmark implements Benchmark {
private static final String DISK_BENCHMARK_COMMAND = "(dd if=/dev/zero of=/tmp/output conv=fdatasync bs=4G count=4; rm -f /tmp/output;) 2>&1 | grep bytes | awk '{ print $8 \" \" $9 }'";
private static final String KILO_BYTE_SEARCH_WORD = "kB/s";
private static final String MEGA_BYTE_SEARCH_WORD = "MB/s";
private static final String GIGA_BYTE_SEARCH_WORD = "GB/s";
private static final String SPLIT_REGEX_STRING = " ";
private static final int SEARCH_ELEMENT_INDEX = 1;
private static final int RETURN_ELEMENT_INDEX = 0;
private static final Logger logger = Logger.getLogger(DiskBenchmark.class.getName());
private final BenchmarkResults benchmarkResults;
private final CommandExecutor commandExecutor;
public DiskBenchmark(BenchmarkResults benchmarkResults, CommandExecutor commandExecutor) {
this.benchmarkResults = benchmarkResults;
this.commandExecutor = commandExecutor;
}
public void doBenchmark() {
try {
ArrayList<String> commandOutput = commandExecutor.executeCommand(DISK_BENCHMARK_COMMAND);
ParseResult parseResult = parseDiskSpeed(commandOutput);
setDiskSpeed(parseResult);
} catch (IOException e) {
logger.log(Level.WARNING, "Failed to perform disk benchmark", e);
}
}
protected ParseResult parseDiskSpeed(ArrayList<String> commandOutput) {
ArrayList<String> searchWords = new ArrayList<>(Arrays.asList(KILO_BYTE_SEARCH_WORD, MEGA_BYTE_SEARCH_WORD, GIGA_BYTE_SEARCH_WORD));
ParseInstructions parseInstructions = new ParseInstructions(SEARCH_ELEMENT_INDEX, RETURN_ELEMENT_INDEX, SPLIT_REGEX_STRING, searchWords);
return OutputParser.parseSingleOutput(parseInstructions, commandOutput);
}
protected void setDiskSpeed(ParseResult parseResult) {
double diskSpeedMBs = getDiskSpeedInMBs(parseResult);
benchmarkResults.setDiskSpeedMbs(diskSpeedMBs);
}
protected double getDiskSpeedInMBs(ParseResult parseResult) {
double diskSpeedMBs = 0;
double convertKBsToMBs = 1 / 1000.0;
double convertGBsToMBs = 1000.0;
double convertMbsToMBs = 1.0;
String diskSpeed = parseResult.getValue();
if (checkSpeedValidity(diskSpeed)) {
switch (parseResult.getSearchWord()) {
case KILO_BYTE_SEARCH_WORD:
diskSpeedMBs = convertToMBs(diskSpeed, convertKBsToMBs);
break;
case MEGA_BYTE_SEARCH_WORD:
diskSpeedMBs = convertToMBs(diskSpeed, convertMbsToMBs);
break;
case GIGA_BYTE_SEARCH_WORD:
diskSpeedMBs = convertToMBs(diskSpeed, convertGBsToMBs);
break;
default:
throw new RuntimeException("Invalid ParseResult searchWord: " + parseResult.getSearchWord());
}
}
return diskSpeedMBs;
}
protected boolean checkSpeedValidity(String speed) {
try {
Double.parseDouble(speed);
} catch (NullPointerException | NumberFormatException e) {
return false;
}
return true;
}
protected Double convertToMBs(String speed, double numberToConvert) {
double speedMbs = Double.parseDouble(speed);
return speedMbs * numberToConvert;
}
}
|