function.ts 6.95 KB
Newer Older
chicm-ms's avatar
chicm-ms committed
1
import * as JSON5 from 'json5';
Lijiao's avatar
Lijiao committed
2
3
import axios from 'axios';
import { MANAGER_IP } from './const';
4
import { MetricDataRecord, FinalType, TableObj } from './interface';
Lijiao's avatar
Lijiao committed
5

6
async function requestAxios(url: string): Promise<any> {
Lijiaoa's avatar
Lijiaoa committed
7
8
9
10
11
12
13
14
15
16
17
18
    const response = await axios.get(url);
    if (response.status === 200) {
        if (response.data.error !== undefined) {
            throw new Error(`API ${url} ${response.data.error}`);
        } else {
            return response.data as any;
        }
    } else {
        throw new Error(`API ${url} ${response.status} error`);
    }
}

Lijiao's avatar
Lijiao committed
19
const convertTime = (num: number): string => {
20
21
22
    if (num <= 0) {
        return '0';
    }
Lijiao's avatar
Lijiao committed
23
24
25
26
27
28
29
30
31
32
    if (num % 3600 === 0) {
        return num / 3600 + 'h';
    } else {
        const hour = Math.floor(num / 3600);
        const min = Math.floor(num / 60 % 60);
        return hour > 0 ? `${hour}h ${min}min` : `${min}min`;
    }
};

// trial's duration, accurate to seconds for example 10min 30s
Lijiao's avatar
Lijiao committed
33
const convertDuration = (num: number): string => {
34
35
36
    if (num < 1) {
        return '0s';
    }
Lijiao's avatar
Lijiao committed
37
    const hour = Math.floor(num / 3600);
38
    const minute = Math.floor(num / 60 % 60);
Lijiao's avatar
Lijiao committed
39
    const second = Math.floor(num % 60);
40
    const result: string[] = [];
41
42
43
44
45
46
47
48
    if (hour > 0) {
        result.push(`${hour}h`);
    }
    if (minute > 0) {
        result.push(`${minute}min`);
    }
    if (second > 0) {
        result.push(`${second}s`);
Lijiao's avatar
Lijiao committed
49
    }
50
    return result.join(' ');
Lijiao's avatar
Lijiao committed
51
52
};

53
54
55
56
57
58
59
60
function parseMetrics(metricData: string): any {
    if (metricData.includes('NaN')) {
        return JSON5.parse(JSON5.parse(metricData));
    } else {
        return JSON.parse(JSON.parse(metricData));
    }
}

61
62
63
64
const isArrayType = (list: any): boolean | undefined => {
    return Array.isArray(list);
}

Lijiao's avatar
Lijiao committed
65
// get final result value
66
// draw Accuracy point graph
Lijiao's avatar
Lijiao committed
67
const getFinalResult = (final?: MetricDataRecord[]): number => {
Lijiao's avatar
Lijiao committed
68
69
70
    let acc;
    let showDefault = 0;
    if (final) {
71
        acc = parseMetrics(final[final.length - 1].data);
72
        if (typeof (acc) === 'object' && !isArrayType(acc)) {
Lijiao's avatar
Lijiao committed
73
74
75
            if (acc.default) {
                showDefault = acc.default;
            }
76
        } else if (typeof (acc) === 'number') {
Lijiao's avatar
Lijiao committed
77
            showDefault = acc;
78
79
        } else {
            showDefault = NaN;
Lijiao's avatar
Lijiao committed
80
81
82
83
84
85
        }
        return showDefault;
    } else {
        return 0;
    }
};
86

87
// get final result value // acc obj
88
const getFinal = (final?: MetricDataRecord[]): FinalType | undefined => {
89
90
    let showDefault: FinalType;
    if (final) {
91
        showDefault = parseMetrics(final[final.length - 1].data);
92
        if (typeof showDefault === 'number') {
93
94
95
96
            if(!isNaN(showDefault)){
                showDefault = { default: showDefault };
                return showDefault;
            }
97
98
99
100
101
        } else if (isArrayType(showDefault)) {
            // not support final type
            return undefined;
        } else if (typeof showDefault === 'object' && showDefault.hasOwnProperty('default')) {
            return showDefault;
102
103
104
105
106
107
        }
    } else {
        return undefined;
    }
};

Lijiao's avatar
Lijiao committed
108
// detail page table intermediate button
Lijiao's avatar
Lijiao committed
109
const intermediateGraphOption = (intermediateArr: number[], id: string): any => {
Lijiao's avatar
Lijiao committed
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
    const sequence: number[] = [];
    const lengthInter = intermediateArr.length;
    for (let i = 1; i <= lengthInter; i++) {
        sequence.push(i);
    }
    return {
        title: {
            text: id,
            left: 'center',
            textStyle: {
                fontSize: 16,
                color: '#333',
            }
        },
        tooltip: {
            trigger: 'item'
        },
        xAxis: {
128
            // name: '#Intermediate result',
Lijiao's avatar
Lijiao committed
129
130
131
            data: sequence
        },
        yAxis: {
Lijiao's avatar
Lijiao committed
132
            name: 'Default metric',
Lijiao's avatar
Lijiao committed
133
            type: 'value',
Lijiaoa's avatar
Lijiaoa committed
134
135
            data: intermediateArr,
            scale: true
Lijiao's avatar
Lijiao committed
136
137
138
139
140
141
142
143
144
145
        },
        series: [{
            symbolSize: 6,
            type: 'scatter',
            data: intermediateArr
        }]
    };
};

// kill job
Lijiao's avatar
Lijiao committed
146
const killJob = (key: number, id: string, status: string, updateList?: Function): void => {
Lijiao's avatar
Lijiao committed
147
148
149
150
151
152
153
154
    axios(`${MANAGER_IP}/trial-jobs/${id}`, {
        method: 'DELETE',
        headers: {
            'Content-Type': 'application/json;charset=utf-8'
        }
    })
        .then(res => {
            if (res.status === 200) {
155
156
                // TODO: use Message.txt to tooltip
                alert('Cancel the job successfully');
Lijiao's avatar
Lijiao committed
157
                // render the table
158
159
160
                if (updateList) {
                    updateList();  // FIXME
                }
Lijiao's avatar
Lijiao committed
161
            } else {
162
                alert('fail to cancel the job');
Lijiao's avatar
Lijiao committed
163
164
165
166
167
            }
        })
        .catch(error => {
            if (error.response.status === 500) {
                if (error.response.data.error) {
168
169
                    alert(123);
                    // message.error(error.response.data.error);
Lijiao's avatar
Lijiao committed
170
                } else {
171
172
                    alert(234);
                    // message.error('500 error, fail to cancel the job');
Lijiao's avatar
Lijiao committed
173
174
175
176
177
                }
            }
        });
};

178
const filterByStatus = (item: TableObj): boolean => {
179
180
181
182
    return item.status === 'SUCCEEDED';
};

// a waittiong trial may havn't start time 
183
const filterDuration = (item: TableObj): boolean => {
184
185
186
    return item.status !== 'WAITING';
};

Lijiao's avatar
Lijiao committed
187
const downFile = (content: string, fileName: string): void => {
v-liguo's avatar
v-liguo committed
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
    const aTag = document.createElement('a');
    const isEdge = navigator.userAgent.indexOf('Edge') !== -1 ? true : false;
    const file = new Blob([content], { type: 'application/json' });
    aTag.download = fileName;
    aTag.href = URL.createObjectURL(file);
    aTag.click();
    if (!isEdge) {
        URL.revokeObjectURL(aTag.href);
    }
    if (navigator.userAgent.indexOf('Firefox') > -1) {
        const downTag = document.createElement('a');
        downTag.addEventListener('click', function () {
            downTag.download = fileName;
            downTag.href = URL.createObjectURL(file);
        });
chicm-ms's avatar
chicm-ms committed
203
        const eventMouse = document.createEvent('MouseEvents');
v-liguo's avatar
v-liguo committed
204
205
206
207
208
        eventMouse.initEvent('click', false, false);
        downTag.dispatchEvent(eventMouse);
    }
};

209
210
211
212
213
// function formatTimestamp(timestamp?: number, placeholder?: string = 'N/A'): string {
function formatTimestamp(timestamp?: number, placeholder?: string): string {
    if (placeholder === undefined) {
        placeholder = 'N/A';
    }
214
215
216
217
    return timestamp ? new Date(timestamp).toLocaleString('en-US') : placeholder;
}

function metricAccuracy(metric: MetricDataRecord): number {
chicm-ms's avatar
chicm-ms committed
218
    const data = parseMetrics(metric.data);
219
220
221
222
223
224
    // return typeof data === 'number' ? data : NaN;
    if (typeof data === 'number') {
        return data;
    } else {
        return data.default;
    }
225
226
227
228
229
230
231
}

function formatAccuracy(accuracy: number): string {
    // TODO: how to format NaN?
    return accuracy.toFixed(6).replace(/0+$/, '').replace(/\.$/, '');
}

232
export {
v-liguo's avatar
v-liguo committed
233
    convertTime, convertDuration, getFinalResult, getFinal, downFile,
234
    intermediateGraphOption, killJob, filterByStatus, filterDuration,
235
    formatAccuracy, formatTimestamp, metricAccuracy, parseMetrics,
Lijiaoa's avatar
Lijiaoa committed
236
    isArrayType, requestAxios
237
};