github.com/eth-easl/loader@v0.0.0-20230908084258-8a37e1d94279/pkg/metric/collect.go (about) 1 /* 2 * MIT License 3 * 4 * Copyright (c) 2023 EASL and the vHive community 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining a copy 7 * of this software and associated documentation files (the "Software"), to deal 8 * in the Software without restriction, including without limitation the rights 9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 10 * copies of the Software, and to permit persons to whom the Software is 11 * furnished to do so, subject to the following conditions: 12 * 13 * The above copyright notice and this permission notice shall be included in all 14 * copies or substantial portions of the Software. 15 * 16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 22 * SOFTWARE. 23 */ 24 25 package metric 26 27 import ( 28 "encoding/json" 29 "os/exec" 30 31 log "github.com/sirupsen/logrus" 32 ) 33 34 func ScrapeDeploymentScales() []DeploymentScale { 35 cmd := exec.Command("python3", "pkg/metric/scrape_scales.py") 36 out, err := cmd.CombinedOutput() 37 if err != nil { 38 log.Warn("Fail to scrape deployment scales: ", err) 39 } 40 41 var results []DeploymentScale 42 err = json.Unmarshal(out, &results) 43 if err != nil { 44 log.Warn("Fail to parse deployment scales: ", string(out[:]), err) 45 } 46 47 return results 48 } 49 50 func ScrapeKnStats() KnStats { 51 cmd := exec.Command( 52 "python3", 53 "pkg/metric/scrape_kn.py", 54 ) 55 out, err := cmd.CombinedOutput() 56 if err != nil { 57 log.Warn("Fail to scrape Knative: ", err) 58 } 59 60 var result KnStats 61 err = json.Unmarshal(out, &result) 62 if err != nil { 63 log.Warn("Fail to parse Knative: ", string(out[:]), err) 64 } 65 66 return result 67 } 68 69 func ScrapeClusterUsage() ClusterUsage { 70 cmd := exec.Command("python3", "pkg/metric/scrape_infra.py") 71 out, err := cmd.CombinedOutput() 72 if err != nil { 73 log.Warn("Fail to scrape cluster usage: ", err) 74 } 75 76 var result ClusterUsage 77 err = json.Unmarshal(out, &result) 78 if err != nil { 79 log.Warn("Fail to parse cluster usage: ", string(out[:]), err) 80 } 81 82 return result 83 }