blob: 32683d9029c486c3c598b52e3df3a54278677988 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
/* NSC -- new Scala compiler
* Copyright 2005-2006 LAMP/EPFL
* @author Martin Odersky
*/
// $Id$
package scala.tools.nsc
import java.io._
import java.net._
object CompileSocket {
/** The prefix of the port identification file, which is followed by the port number */
private val dirName = "scalac-compile-server-port"
/** The vm-part of the command to start a new scala compile server */
private val vmCommand = "java"
/** The class name of the scala compile server */
private val serverClass = "scala.tools.nsc.CompileServer"
/** The temporary directory in which the port identification file is stored */
private val tmpDir = {
val d = new File(System.getProperty("java.io.tmpdir"), dirName)
if (!d.isDirectory())
if (!d.mkdir()) {
System.err.println("cannot create directory "+dirName+"; exiting")
exit(1)
}
d
}
/** Maximum number of polls for an available port */
private val MaxAttempts = 100
/** Time (in ms) to sleep between two polls */
private val sleepTime = 20
/** The command which starts the compile server, given vm arguments
* @param vmArgs the argument string to be passed to the java or scala command
* the string must be either empty or start with a ' '.
*/
def serverCommand(vmArgs: String): String = vmCommand+vmArgs+" "+serverClass
/** The port identification file */
def portFile(port: int) = new File(tmpDir, port.toString())
/** Poll for a server port number; return -1 if none exists yet */
private def pollPort(): int = {
val hits = tmpDir.listFiles()
if (hits.length == 0) -1
else
try {
for (val i <- 1 until hits.length) hits(i).delete()
Integer.parseInt(hits(0).getName)
} catch {
case ex: Throwable =>
System.err.println(ex)
System.err.println("bad file in temp directory: "+hits(0).getAbsolutePath())
System.err.println("please remove the file and try again")
exit(1)
}
}
/** Get the port number to which a scala compile server is connected;
* If no server is running yet, create one
*/
def getPort(vmArgs: String): int = {
var attempts = 0;
var port = pollPort()
if (port < 0) {
Runtime.getRuntime().exec(serverCommand(vmArgs))
}
while (port < 0 && attempts < MaxAttempts) {
attempts = attempts + 1
Thread.sleep(sleepTime)
port = pollPort()
}
if (port < 0) {
System.err.println("cannot start server with command")
System.err.println(serverCommand(vmArgs))
exit(1)
}
port
}
/** Set the port number to which a scala compile server is connected */
def setPort(port: int): unit =
try {
val f = new FileOutputStream(portFile(port));
f.close()
} catch {
case ex: IOException =>
System.err.println("cannot create file: "+portFile(port).getAbsolutePath()+"; exiting")
throw new Error()
}
/** Delete the port number to which a scala compile server was connected */
def deletePort(port: int): unit = portFile(port).delete()
def getOrCreateSocket(vmArgs: String): Socket = {
def getsock(attempts: int): Socket =
if (attempts == 0) {
System.err.println("unable to establish connection to server; exiting");
exit(1)
} else {
val port = getPort(vmArgs)
val hostName = InetAddress.getLocalHost().getHostName()
try {
new Socket(hostName, port)
} catch {
case e: IOException =>
System.err.println(e)
System.err.println("...connection attempt to server at port "+port+" failed; re-trying...")
if (attempts % 2 == 0) portFile(port).delete()
Thread.sleep(100)
getsock(attempts - 1)
}
}
getsock(9)
}
def getSocket(serverAdr: String): Socket = {
val cpos = serverAdr indexOf ':'
if (cpos < 0) {
System.err.println("malformed server address: "+serverAdr+"; exiting")
exit(1)
} else {
val hostName = serverAdr.substring(0, cpos)
val port = try {
Integer.parseInt(serverAdr.substring(cpos+1))
} catch {
case ex: Throwable =>
System.err.println("malformed server address: "+serverAdr+"; exiting")
exit(1)
}
getSocket(hostName, port)
}
}
def getSocket(hostName: String, port: int): Socket =
try {
new Socket(hostName, port)
} catch {
case e: IOException =>
System.err.println(
"unable to establish connection to server "+hostName+":"+port+"; exiting")
exit(1)
}
}
|
