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
|
package io.crashbox.ci
import java.io.{
BufferedInputStream,
File,
FileInputStream,
FileOutputStream,
InputStream,
OutputStream
}
import scala.concurrent.Future
import java.util.UUID
import slick.jdbc.H2Profile.api._
class Storage(implicit core: Core) {
import core._
import Scheduler._
case class Build(
id: BuildId,
url: String,
rebuild: Int,
state: Int
)
class Builds(tag: Tag) extends Table[Build](tag, "builds") {
def id = column[BuildId]("id", O.PrimaryKey)
def url = column[String]("url")
def rebuild = column[Int]("rebuild")
def state = column[Int]("state")
def * = (id, url, rebuild, state) <> (Build.tupled, Build.unapply)
}
val builds = TableQuery[Builds]
val database = Database.forConfig("crashbox.db")
def setupDatabase(): Future[Unit] = {
log.info("Preparing build database")
val setup = DBIO.seq(
builds.schema.create
)
database.run(setup)
}
def newBuildId(): BuildId = UUID.randomUUID()
def nextBuild(url: String): Future[Build] = database.run{
builds.filter(_.url === url).map(_.rebuild).take(1).result.headOption
}.map{ no =>
Build(newBuildId(), url, no.getOrElse(0), 0)
}
def updateBuildState(buildId: BuildId, state: BuildState) = {
log.info(s"Build $buildId: state update $state")
}
private val streamsDirectory: File = new File(
config.getString("crashbox.streams.directory"))
private def logFile(buildId: BuildId, task: Int): File = {
def stringifyId(id: BuildId): String = {
val bytes = new Array[Byte](16) // 128 bits
for (i <- 0 until 8) {
bytes(i) = ((id.getLeastSignificantBits >> i) & 0xff).toByte
}
for (i <- 0 until 8) {
bytes(8 + i) = ((id.getMostSignificantBits >> i) & 0xff).toByte
}
bytes.map { byte =>
Integer.toString((byte & 0xff) + 0x100, 16)
}.mkString
}
val (dir1, tail) = stringifyId(buildId).splitAt(2)
val (dir2, dir3) = tail.splitAt(2)
new File(streamsDirectory, s"$dir1/$dir2/$dir3/$task")
}
def saveLog(buildId: BuildId, task: Int): OutputStream = {
val file = logFile(buildId, task)
file.getParentFile.mkdirs()
file.createNewFile()
file.setWritable(true)
new FileOutputStream(file)
}
def readLog(buildId: BuildId, task: Int): InputStream = {
new FileInputStream(logFile(buildId, task))
}
}
|