net.oneandone.stool.util.Pool Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of main Show documentation
Show all versions of main Show documentation
Stool's main component. Java Library, cli, setup code.
/**
* Copyright 1&1 Internet AG, https://github.com/1and1/
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package net.oneandone.stool.util;
import net.oneandone.stool.stage.Stage;
import net.oneandone.sushi.cli.ArgumentException;
import net.oneandone.sushi.fs.LineFormat;
import net.oneandone.sushi.fs.LineReader;
import net.oneandone.sushi.fs.file.FileNode;
import java.io.IOException;
import java.io.Reader;
import java.net.ServerSocket;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
public class Pool {
private static final LineFormat FMT = new LineFormat(LineFormat.LF_SEPARATOR, LineFormat.Trim.ALL, LineFormat.excludes(true));
public static Pool loadOpt(FileNode file, int first, int last, FileNode backstages) throws IOException {
Pool result;
String line;
result = new Pool(file, first, last, backstages);
if (file.exists()) {
try (Reader in = file.createReader(); LineReader src = new LineReader(in, FMT)) {
while (true) {
line = src.next();
if (line == null) {
break;
}
result.vhosts.add(Vhost.forLine(file.getWorld(), line));
}
}
}
return result;
}
//--
private final FileNode file;
private final int first;
private final int last;
private final List vhosts;
private final FileNode backstages;
public Pool(FileNode file, int first, int last, FileNode backstages) {
if (first % 2 != 0) {
throw new IllegalArgumentException("even port expected: " + first);
}
if (last % 2 != 1) {
throw new IllegalArgumentException("odd port expected: " + last);
}
this.file = file;
this.first = first;
this.last = last;
this.backstages = backstages;
this.vhosts = new ArrayList<>();
}
public FileNode getFile() {
return file;
}
// TODO: ugly reference to stage ...
public Ports allocate(Stage stage, Map fixed) throws IOException {
// maps vhosts to docroots
LinkedHashSet names;
Map nameDocroots;
Vhost previous;
Vhost found;
String stageName;
List result;
Integer port; // null for not fixed
Vhost modified;
FileNode docroot;
gc();
names = new LinkedHashSet<>();
names.add(Ports.STOP_WRAPPER);
names.add(Ports.JMX_DEBUG);
names.addAll(fixed.keySet());
nameDocroots = new LinkedHashMap<>();
nameDocroots.putAll(stage.selectedVhosts());
nameDocroots.putAll(stage.extensions().vhosts(stage));
names.addAll(nameDocroots.keySet());
stageName = stage.getName();
result = new ArrayList<>();
for (String name : names) {
docroot = nameDocroots.get(name);
port = fixed.get(name);
previous = lookup(name, stageName);
if (previous != null) {
modified = previous.set(port, docroot);
if (modified == null) {
// no changes
result.add(previous);
} else {
if (port != null) {
if (used(port)) {
throw new ArgumentException("port already reserved: " + port);
}
checkFree(port);
}
result.add(modified);
vhosts.remove(previous);
vhosts.add(modified);
}
} else {
if (port == null) {
found = allocate(name, stageName, docroot);
} else {
found = allocate(port, name, stageName, docroot);
if (found.even != port) {
throw new ArgumentException("port already in use: " + port);
}
}
result.add(found);
}
}
// gc this stage, i.e. remove all vhosts that are now unused
for (int i = vhosts.size() - 1; i >= 0; i--) {
if (stageName.equals(vhosts.get(i).stage) && !result.contains(vhosts.get(i))) {
vhosts.remove(i);
}
}
save();
return new Ports(result);
}
private void gc() {
Set stages;
stages = new HashSet<>();
for (Vhost vhost : vhosts) {
stages.add(vhost.stage);
}
for (String stage : stages) {
if (!backstages.join(stage).isDirectory()) {
gc(stage);
}
}
}
private void gc(String stage) {
Vhost vhost;
for (int i = vhosts.size() - 1; i >= 0; i--) {
vhost = vhosts.get(i);
if (stage.equals(vhost.stage)) {
vhosts.remove(i);
}
}
}
public Vhost lookup(String name, String stage) {
for (Vhost vhost : vhosts) {
if (name.equals(vhost.name) && stage.equals(vhost.stage)) {
return vhost;
}
}
return null;
}
public void save() throws IOException {
List lines;
lines = new ArrayList<>();
for (Vhost vhost : vhosts) {
lines.add(vhost.toLine());
}
Files.stoolFile(file.writeLines(lines));
}
private Vhost allocate(String name, String stage, FileNode docroot) throws IOException {
return allocate(forName(name, stage), name, stage, docroot);
}
private Vhost allocate(int start, String name, String stage, FileNode docroot) throws IOException {
int current;
Vhost result;
if ((start < first) || (start > last)) {
throw new IllegalArgumentException("ports out of range: " + start);
}
if (start % 2 != 0) {
throw new IllegalArgumentException("even port expected: " + start);
}
current = start;
do {
if (!used(current)) {
checkFree(current);
checkFree(current + 1);
result = new Vhost(current, name, stage, docroot);
vhosts.add(result);
return result;
}
current += 2;
if (current > last) {
current = first;
}
} while (current != start);
throw new IOException("cannot find free port in range [" + first + ", " + last + "[");
}
public int temp() throws IOException {
int current;
for (current = first; current <= last; current += 2) {
if (!used(current)) {
return current;
}
}
throw new IOException("cannot find free port in range [" + first + ", " + last + "[");
}
public void add(Vhost vhost) {
vhosts.add(vhost);
}
private boolean used(int even) {
for (Vhost vhost : vhosts) {
if (vhost.even == even) {
return true;
}
}
return false;
}
private int forName(String name, String stage) {
int result;
result = (name + stage).hashCode();
if (result < 0) {
result = -result;
if (result < 0) { // happens for Integer.MIN_VALUE
result = 0;
}
}
return ((result % (last - first + 1)) + first) & 0xfffffffe;
}
//--
/**
* See http://stackoverflow.com/questions/434718/sockets-discover-port-availability-using-java
*/
public static void checkFree(int port) throws IOException {
boolean available;
ServerSocket socket;
socket = null;
available = false;
try {
socket = new ServerSocket(port);
available = true;
} catch (IOException e) {
// fall-through
} finally {
// Clean up
if (socket != null) {
try {
socket.close();
} catch (IOException e) {
/* should not be thrown */
e.printStackTrace();
}
}
}
if (!available) {
throw new IOException("port already in use: " + port);
}
}
public Ports stageOpt(String stage) {
List result;
result = new ArrayList<>();
for (Vhost vhost : vhosts) {
if (stage.equals(vhost.stage)) {
result.add(vhost);
}
}
return result.isEmpty() ? null : new Ports(result);
}
}