forked from vieter-v/vieter
Some logging improvements; fixed segfault
parent
af86d91d3f
commit
3ada24ef3b
4
Makefile
4
Makefile
|
@ -2,6 +2,10 @@
|
||||||
run:
|
run:
|
||||||
API_KEY=test REPO_DIR=data LOG_LEVEL=DEBUG v -cg run vieter
|
API_KEY=test REPO_DIR=data LOG_LEVEL=DEBUG v -cg run vieter
|
||||||
|
|
||||||
|
.PHONY: run-prod
|
||||||
|
run-prod:
|
||||||
|
API_KEY=test REPO_DIR=data LOG_LEVEL=DEBUG v -prod run vieter
|
||||||
|
|
||||||
.PHONY: watch
|
.PHONY: watch
|
||||||
watch:
|
watch:
|
||||||
API_KEY=test REPO_DIR=data LOG_LEVEL=DEBUG v watch run vieter
|
API_KEY=test REPO_DIR=data LOG_LEVEL=DEBUG v watch run vieter
|
||||||
|
|
|
@ -1,9 +0,0 @@
|
||||||
module fibonacci
|
|
||||||
|
|
||||||
pub fn fib(i int) int {
|
|
||||||
if i <= 1 {
|
|
||||||
return i
|
|
||||||
}
|
|
||||||
|
|
||||||
return fib(i - 1) + fib(i - 2)
|
|
||||||
}
|
|
|
@ -8,7 +8,7 @@ import repo
|
||||||
|
|
||||||
const port = 8000
|
const port = 8000
|
||||||
|
|
||||||
const buf_size = 100_000
|
const buf_size = 1_000_000
|
||||||
|
|
||||||
const db_name = 'pieter.db.tar.gz'
|
const db_name = 'pieter.db.tar.gz'
|
||||||
|
|
||||||
|
@ -80,21 +80,23 @@ fn main() {
|
||||||
exit_with_message(1, 'No repo directory was configured.')
|
exit_with_message(1, 'No repo directory was configured.')
|
||||||
}
|
}
|
||||||
|
|
||||||
// We create the upload directory during startup
|
repo := repo.Repo{
|
||||||
if !os.is_dir(repo_dir) {
|
dir: repo_dir
|
||||||
os.mkdir_all(repo_dir) or {
|
name: db_name
|
||||||
exit_with_message(2, "Failed to create repo directory '$repo_dir'.")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.info("Created repo directory '$repo_dir'")
|
// We create the upload directory during startup
|
||||||
|
if !os.is_dir(repo.pkg_dir()) {
|
||||||
|
os.mkdir_all(repo.pkg_dir()) or {
|
||||||
|
exit_with_message(2, "Failed to create repo directory '$repo.pkg_dir()'.")
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.info("Created package directory '$repo.pkg_dir()'.")
|
||||||
}
|
}
|
||||||
|
|
||||||
web.run(&App{
|
web.run(&App{
|
||||||
logger: logger
|
logger: logger
|
||||||
api_key: key
|
api_key: key
|
||||||
repo: repo.Repo{
|
repo: repo
|
||||||
dir: repo_dir
|
|
||||||
name: db_name
|
|
||||||
}
|
|
||||||
}, port)
|
}, port)
|
||||||
}
|
}
|
||||||
|
|
|
@ -4,16 +4,24 @@ import os
|
||||||
|
|
||||||
const pkgs_subpath = 'pkgs'
|
const pkgs_subpath = 'pkgs'
|
||||||
|
|
||||||
|
// Dummy struct to work around the fact that you can only share structs, maps &
|
||||||
|
// arrays
|
||||||
|
pub struct Dummy { x int }
|
||||||
|
|
||||||
// Handles management of a repository. Package files are stored in '$dir/pkgs'
|
// Handles management of a repository. Package files are stored in '$dir/pkgs'
|
||||||
// & moved there if necessary.
|
// & moved there if necessary.
|
||||||
pub struct Repo {
|
pub struct Repo {
|
||||||
mut:
|
mut:
|
||||||
mutex shared int = 0
|
mutex shared Dummy
|
||||||
pub:
|
pub:
|
||||||
dir string [required]
|
dir string [required]
|
||||||
name string [required]
|
name string [required]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn (r &Repo) pkg_dir() string {
|
||||||
|
return os.join_path_single(r.dir, pkgs_subpath)
|
||||||
|
}
|
||||||
|
|
||||||
// Returns path to the given package, prepended with the repo's path.
|
// Returns path to the given package, prepended with the repo's path.
|
||||||
pub fn (r &Repo) pkg_path(pkg string) string {
|
pub fn (r &Repo) pkg_path(pkg string) string {
|
||||||
return os.join_path(r.dir, pkgs_subpath, pkg)
|
return os.join_path(r.dir, pkgs_subpath, pkg)
|
||||||
|
|
|
@ -3,11 +3,26 @@ module main
|
||||||
import web
|
import web
|
||||||
import os
|
import os
|
||||||
import repo
|
import repo
|
||||||
|
import time
|
||||||
|
|
||||||
|
const prefixes = ['B', 'KB', 'MB', 'GB']
|
||||||
|
|
||||||
|
fn pretty_bytes(bytes int) string {
|
||||||
|
mut i := 0
|
||||||
|
mut n := f32(bytes)
|
||||||
|
|
||||||
|
for n >= 1024 {
|
||||||
|
i++
|
||||||
|
n /= 1024
|
||||||
|
}
|
||||||
|
|
||||||
|
return '${n:.2}${prefixes[i]}'
|
||||||
|
}
|
||||||
|
|
||||||
['/pkgs/:pkg'; put]
|
['/pkgs/:pkg'; put]
|
||||||
fn (mut app App) put_package(pkg string) web.Result {
|
fn (mut app App) put_package(pkg string) web.Result {
|
||||||
if app.repo.exists(pkg) {
|
if app.repo.exists(pkg) {
|
||||||
app.lwarn("Tried to upload duplicate package '$pkg'")
|
app.lwarn("Duplicate package '$pkg'")
|
||||||
|
|
||||||
return app.text('File already exists.')
|
return app.text('File already exists.')
|
||||||
}
|
}
|
||||||
|
@ -15,20 +30,27 @@ fn (mut app App) put_package(pkg string) web.Result {
|
||||||
pkg_path := app.repo.pkg_path(pkg)
|
pkg_path := app.repo.pkg_path(pkg)
|
||||||
|
|
||||||
if length := app.req.header.get(.content_length) {
|
if length := app.req.header.get(.content_length) {
|
||||||
app.ldebug("Uploading $length bytes to package '$pkg'")
|
app.ldebug("Uploading $length (${pretty_bytes(length.int())}) bytes to package '$pkg'.")
|
||||||
println(pkg_path)
|
|
||||||
|
// This is used to time how long it takes to upload a file
|
||||||
|
mut sw := time.new_stopwatch(time.StopWatchOptions{auto_start: true})
|
||||||
|
|
||||||
reader_to_file(mut app.reader, length.int(), pkg_path) or {
|
reader_to_file(mut app.reader, length.int(), pkg_path) or {
|
||||||
app.lwarn("Failed to upload package '$pkg'")
|
app.lwarn("Failed to upload package '$pkg'")
|
||||||
|
|
||||||
return app.text('Failed to upload file.')
|
return app.text('Failed to upload file.')
|
||||||
}
|
}
|
||||||
|
|
||||||
|
sw.stop()
|
||||||
|
app.ldebug("Upload of package '$pkg' completed in ${sw.elapsed().seconds():.3}s.")
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
app.lwarn("Tried to upload package '$pkg' without specifying a Content-Length.")
|
app.lwarn("Tried to upload package '$pkg' without specifying a Content-Length.")
|
||||||
return app.text("Content-Type header isn't set.")
|
return app.text("Content-Type header isn't set.")
|
||||||
}
|
}
|
||||||
|
|
||||||
app.repo.add_package(pkg_path) or {
|
app.repo.add_package(pkg_path) or {
|
||||||
app.linfo("Failed to add package '$pkg' to database.")
|
app.lwarn("Failed to add package '$pkg' to database.")
|
||||||
|
|
||||||
os.rm(pkg_path) or { println('Failed to remove $pkg_path') }
|
os.rm(pkg_path) or { println('Failed to remove $pkg_path') }
|
||||||
|
|
||||||
|
@ -36,7 +58,6 @@ fn (mut app App) put_package(pkg string) web.Result {
|
||||||
}
|
}
|
||||||
|
|
||||||
app.linfo("Added '$pkg' to repository.")
|
app.linfo("Added '$pkg' to repository.")
|
||||||
app.linfo("Uploaded package '$pkg'.")
|
|
||||||
|
|
||||||
return app.text('Package added successfully.')
|
return app.text('Package added successfully.')
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue