public boolean update(Weight weight, Map<String, Object> context) {
if (!super.currentUser.hasServicePrivilege(weight.getService())) {
context.put("message", getMessage("HaveNoServicePrivilege", weight.getService()));
return false;
}
weight.setAddress(Tool.getIP(weight.getAddress()));
overrideService.updateOverride(OverrideUtils.weightToOverride(weight));
return true;
}
public void index(Map<String, Object> context) {
final String service = StringUtils.trimToNull((String) context.get("service"));
String address = (String) context.get("address");
address = Tool.getIP(address);
List<Weight> weights;
if (service != null && service.length() > 0) {
weights = OverrideUtils.overridesToWeights(overrideService.findByService(service));
} else if (address != null && address.length() > 0) {
weights = OverrideUtils.overridesToWeights(overrideService.findByAddress(address));
} else {
weights = OverrideUtils.overridesToWeights(overrideService.findAll());
}
context.put("weights", weights);
}
/**
* load页面供新增操作
*
* @param context
*/
public void add(Map<String, Object> context) {
String service = (String) context.get("service");
if (service != null && service.length() > 0 && !service.contains("*")) {
List<Provider> providerList = providerService.findByService(service);
List<String> addressList = new ArrayList<String>();
for (Provider provider : providerList) {
addressList.add(provider.getUrl().split("://")[1].split("/")[0]);
}
context.put("addressList", addressList);
context.put("service", service);
context.put("methods", CollectionUtils.sort(providerService.findMethodsByService(service)));
} else {
List<String> serviceList = Tool.sortSimpleName(providerService.findServices());
context.put("serviceList", serviceList);
}
if (context.get("input") != null) context.put("input", context.get("input"));
}
public boolean create(Map<String, Object> context) throws Exception {
String addr = (String) context.get("address");
String services = (String) context.get("multiservice");
if (services == null || services.trim().length() == 0) {
services = (String) context.get("service");
}
String weight = (String) context.get("weight");
int w = Integer.parseInt(weight);
Set<String> addresses = new HashSet<String>();
BufferedReader reader = new BufferedReader(new StringReader(addr));
while (true) {
String line = reader.readLine();
if (null == line) break;
String[] split = line.split("[\\s,;]+");
for (String s : split) {
if (s.length() == 0) continue;
String ip = s;
String port = null;
if (s.indexOf(":") != -1) {
ip = s.substring(0, s.indexOf(":"));
port = s.substring(s.indexOf(":") + 1, s.length());
if (port.trim().length() == 0) port = null;
}
if (!IP_PATTERN.matcher(ip).matches()) {
context.put("message", "illegal IP: " + s);
return false;
}
if (LOCAL_IP_PATTERN.matcher(ip).matches() || ALL_IP_PATTERN.matcher(ip).matches()) {
context.put("message", "local IP or any host ip is illegal: " + s);
return false;
}
if (port != null) {
if (!NumberUtils.isDigits(port)) {
context.put("message", "illegal port: " + s);
return false;
}
}
addresses.add(s);
}
}
Set<String> aimServices = new HashSet<String>();
reader = new BufferedReader(new StringReader(services));
while (true) {
String line = reader.readLine();
if (null == line) break;
String[] split = line.split("[\\s,;]+");
for (String s : split) {
if (s.length() == 0) continue;
if (!super.currentUser.hasServicePrivilege(s)) {
context.put("message", getMessage("HaveNoServicePrivilege", s));
return false;
}
aimServices.add(s);
}
}
for (String aimService : aimServices) {
for (String a : addresses) {
Weight wt = new Weight();
wt.setUsername((String) context.get("operator"));
wt.setAddress(Tool.getIP(a));
wt.setService(aimService);
wt.setWeight(w);
overrideService.saveOverride(OverrideUtils.weightToOverride(wt));
}
}
return true;
}
/**
* load页面供新增操作
*
* @param context
*/
public void multiadd(Map<String, Object> context) {
List<String> serviceList = Tool.sortSimpleName(providerService.findServices());
context.put("serviceList", serviceList);
}