-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* DockerAPIを操作するためのクライアントを追加 #162 * jesterがdevelだと動かなかったのでバージョン固定 #162 * 不要になったテストコードを削除 #162 * Nimコンパイラのバージョンアップ #162 * ビルド用のコンテナでdevelを使うように変更 #162 * テスト方法を修正 #162 * bigendianで読み取るように修正 * ディレクトリ作成 * コンテナを削除 * 削除をremoverで * fix * 削除のログを追加 * 一旦無効化 * ネットワーク制限を追加 * 非同期処理に変更 * Revert "非同期処理に変更" This reverts commit a7a8bb3. * getLogs follow=falseにして自前でタイムアウトするように変更 * デバッグコードを削除 * ciフラグありのときはテストをスキップ * ciでのテスト時はciスキップ
- Loading branch information
Showing
9 changed files
with
291 additions
and
150 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,153 @@ | ||
| import httpclient, os, json, strformat, streams, endians | ||
| from strutils import join | ||
|
|
||
| import status | ||
|
|
||
| type | ||
| DockerClient = ref object | ||
| client: HttpClient | ||
| url: string | ||
| Mount = object | ||
| Target, Source, Type: string | ||
| ReadOnly: bool | ||
| HostConfig = object | ||
| Memory: int64 | ||
| OomKillDisable: bool | ||
| PidsLimit: int64 | ||
| Mounts: seq[Mount] | ||
| NetworkMode: string | ||
|
|
||
| proc newClient*(): DockerClient = | ||
| let | ||
| client = newHttpClient(timeout = 10 * 1000) | ||
| url = "http://localhost:2376" | ||
| return DockerClient(client: client, url: url) | ||
|
|
||
| proc createContainer*(self: DockerClient, name: string, image: string, cmds: seq[string], script = "", mediaDir = "", imageDir = ""): Response = | ||
| var self = self | ||
| self.client.headers = newHttpHeaders({ "Content-Type": "application/json" }) | ||
| let url = &"{self.url}/containers/create?name={name}" | ||
|
|
||
| var body = %*{ | ||
| "Image": image, | ||
| "Cmd": cmds, | ||
| "Tty": false, | ||
| } | ||
|
|
||
| var hostconf = HostConfig( | ||
| Memory: 256 * 1024 * 1024, # 256MB | ||
| OomKillDisable: true, | ||
| PidsLimit: 1024, | ||
| NetworkMode: "none", | ||
| ) | ||
| var mounts: seq[Mount] | ||
| if script != "": | ||
| mounts.add(Mount(Target: "/tmp/exec.sh", Source: script, Type: "bind", ReadOnly: true)) | ||
| if mediaDir != "": | ||
| mounts.add(Mount(Target: "/media", Source: mediaDir, Type: "bind", ReadOnly: true)) | ||
| if imageDir != "": | ||
| mounts.add(Mount(Target: "/images", Source: imageDir, Type: "bind", ReadOnly: false)) | ||
|
|
||
| if 1 <= mounts.len: | ||
| hostconf.Mounts = mounts | ||
|
|
||
| body["HostConfig"] = % hostconf | ||
|
|
||
| self.client.post(url = url, body = $body) | ||
|
|
||
| proc startContainer*(self: DockerClient, name: string): Response = | ||
| var self = self | ||
| self.client.headers = newHttpHeaders({ "Content-Type": "application/json" }) | ||
| let url = &"{self.url}/containers/{name}/start" | ||
| self.client.post(url = url) | ||
|
|
||
| proc killContainer*(self: DockerClient, name: string): Response = | ||
| var self = self | ||
| let url = &"{self.url}/containers/{name}/kill" | ||
| self.client.post(url = url) | ||
|
|
||
| proc removeContainer*(self: DockerClient, name: string): Response = | ||
| var self = self | ||
| let url = &"{self.url}/containers/{name}?v=true&force=true" | ||
| self.client.delete(url = url) | ||
|
|
||
| proc inspectContainer*(self: DockerClient, name: string): Response = | ||
| var self = self | ||
| let url = &"{self.url}/containers/{name}/json" | ||
| self.client.get(url = url) | ||
|
|
||
| proc parseLog(s: string): string = | ||
| var strm = newStringStream(s) | ||
| defer: strm.close | ||
| var lines: seq[string] | ||
| while not strm.atEnd: | ||
| # 1 = stdout, 2 = stderr | ||
| if strm.readUint8() notin [1'u8, 2]: | ||
| break | ||
|
|
||
| # 3byteは使わないので捨てる | ||
| discard strm.readUint8() | ||
| discard strm.readUint8() | ||
| discard strm.readUint8() | ||
|
|
||
| # Bigendianで読み取る | ||
| var src = strm.readUint32().int | ||
| var n: int | ||
| bigEndian32(addr(n), addr(src)) | ||
|
|
||
| lines.add(strm.readStr(n)) | ||
| result = lines.join | ||
|
|
||
| proc getLog(self: DockerClient, name: string, stdout = false, stderr = false): Response = | ||
| let url = &"{self.url}/containers/{name}/logs?stdout={stdout}&stderr={stderr}&follow=false" | ||
| self.client.get(url = url) | ||
|
|
||
| proc getStdoutLog*(self: DockerClient, name: string): Response = | ||
| self.getLog(name = name, stdout = true, stderr = false) | ||
|
|
||
| proc getStderrLog*(self: DockerClient, name: string): Response = | ||
| self.getLog(name = name, stdout = false, stderr = true) | ||
|
|
||
| proc waitFinish(self: DockerClient, name: string) = | ||
| const timeout = 10000 | ||
| const unit = 200 | ||
| var elapsed: int | ||
| while true: | ||
| let resp = self.inspectContainer(name = name) | ||
| if not resp.code.is2xx: | ||
| return | ||
| let running = resp.body.parseJson["State"]["Running"].getBool | ||
| if not running: break | ||
|
|
||
| sleep unit | ||
| elapsed += unit | ||
| if timeout <= elapsed: break | ||
|
|
||
| proc runContainer*(self: DockerClient, name: string, image: string, cmds: seq[string], script = "", mediaDir = "", imageDir = ""): (string, string, int, string) = | ||
| var resp: Response | ||
| resp = self.createContainer(name = name, image = image, cmds = cmds, script = script, mediaDir = mediaDir, imageDir = imageDir) | ||
| if not resp.code.is2xx: | ||
| return ("", "", statusSystemError, &"failed to call 'createContainer': cmds={cmds} resp.body={resp.body}") | ||
|
|
||
| resp = self.startContainer(name = name) | ||
| if not resp.code.is2xx: | ||
| return ("", "", statusSystemError, &"failed to call 'startContainer': cmds={cmds} resp.body={resp.body}") | ||
|
|
||
| self.waitFinish(name = name) | ||
|
|
||
| var stdoutStr: string | ||
| resp = self.getStdoutLog(name = name) | ||
| if not resp.code.is2xx: | ||
| return ("", "", statusSystemError, &"failed to call 'getStdoutLog': cmds={cmds} resp.body={resp.body}") | ||
| stdoutStr = resp.body.parseLog | ||
|
|
||
| var stderrStr: string | ||
| resp = self.getStderrLog(name = name) | ||
| if not resp.code.is2xx: | ||
| return ("", "", statusSystemError, &"failed to call 'getStderrLog': cmds={cmds} resp.body={resp.body}") | ||
| stderrStr = resp.body.parseLog | ||
|
|
||
| discard self.killContainer(name = name) | ||
| discard self.removeContainer(name = name) | ||
|
|
||
| return (stdoutStr, stderrStr, statusOk, "") |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| const | ||
| statusOk* = 0 | ||
| statusTimeout* = 1 | ||
| statusSystemError* = 100 | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.