1 /* 2 * Copyright (c) 2017, 2018, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 package jdk.test.lib.containers.cgroup; 25 26 import java.io.IOException; 27 import java.nio.file.Files; 28 import java.nio.file.Paths; 29 import java.util.ArrayList; 30 import java.util.List; 31 import java.util.Optional; 32 import java.util.stream.Collectors; 33 import java.util.stream.IntStream; 34 import java.util.stream.Stream; 35 36 import jdk.test.lib.Asserts; 37 38 39 // A simple CPU sets reader and parser 40 public class CPUSetsReader { 41 public static String PROC_SELF_STATUS_PATH = "/proc/self/status"; 42 43 // Test the parser 44 public static void test() { 45 assertParse("0-7", "0,1,2,3,4,5,6,7"); 46 assertParse("1,3,6", "1,3,6"); 47 assertParse("0,2-4,6,10-11", "0,2,3,4,6,10,11"); 48 assertParse("0", "0"); 49 } 50 51 52 private static void assertParse(String cpuSet, String expectedResult) { 53 Asserts.assertEquals(listToString(parseCpuSet(cpuSet)), expectedResult); 54 } 55 56 public static int getNumCpus() { 57 String path = "/proc/cpuinfo"; 58 try(Stream<String> stream = Files.lines(Paths.get(path))) { 59 return (int) stream.filter(line -> line.startsWith("processor")).count(); 60 } catch (IOException e) { 61 return 0; 62 } 63 } 64 65 66 public static String readFromProcStatus(String setType) { 67 String path = PROC_SELF_STATUS_PATH; 68 Optional<String> o = Optional.empty(); 69 70 System.out.println("readFromProcStatus() entering for: " + setType); 71 72 try (Stream<String> stream = Files.lines(Paths.get(path))) { 73 o = stream 74 .filter(line -> line.contains(setType)) 75 .findFirst(); 76 } catch (IOException e) { 77 return null; 78 } 79 80 if (!o.isPresent()) { 81 return null; // entry not found 82 } 83 84 String[] parts = o.get().replaceAll("\\s", "").split(":"); 85 86 // Should be 2 parts, before and after ":" 87 Asserts.assertEquals(parts.length, 2); 88 89 String result = parts[1]; 90 System.out.println("readFromProcStatus() returning: " + result); 91 return result; 92 } 93 94 95 public static List<Integer> parseCpuSet(String value) { 96 ArrayList<Integer> result = new ArrayList<Integer>(); 97 98 try { 99 String[] commaSeparated = value.split(","); 100 101 for (String item : commaSeparated) { 102 if (item.contains("-")) { 103 addRange(result, item); 104 } else { 105 result.add(Integer.parseInt(item)); 106 } 107 } 108 } catch (Exception e) { 109 System.err.println("Exception in getMaxCpuSets(): " + e); 110 return null; 111 } 112 113 return result; 114 } 115 116 private static void addRange(ArrayList<Integer> list, String s) { 117 String[] range = s.split("-"); 118 if (range.length != 2) { 119 throw new RuntimeException("Range should only contain two items, but contains " 120 + range.length + " items"); 121 } 122 123 int min = Integer.parseInt(range[0]); 124 int max = Integer.parseInt(range[1]); 125 126 if (min >= max) { 127 String msg = String.format("min is greater or equals to max, min = %d, max = %d", 128 min, max); 129 throw new RuntimeException(msg); 130 } 131 132 for (int i = min; i <= max; i++) { 133 list.add(i); 134 } 135 } 136 137 138 // Convert list of integers to string with comma-separated values 139 public static String listToString(List<Integer> list) { 140 return listToString(list, Integer.MAX_VALUE); 141 } 142 143 // Convert list of integers to a string with comma-separated values; 144 // include up to maxCount. 145 public static String listToString(List<Integer> list, int maxCount) { 146 return list.stream() 147 .limit(maxCount) 148 .map(Object::toString) 149 .collect(Collectors.joining(",")); 150 } 151 152 public static String numberToString(int num) { 153 return IntStream.range(0, num).boxed().map(Object::toString).collect(Collectors.joining(",")); 154 } 155 }