You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
1.2 KiB
Go
86 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/c2h5oh/datasize"
|
|
|
|
)
|
|
|
|
|
|
func (qc QuotaConfig) ParsedSizeBytes() int64 {
|
|
var v datasize.ByteSize
|
|
|
|
err := v.UnmarshalText([]byte(qc.DefaultSize))
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return int64(v)
|
|
}
|
|
|
|
func (qc QuotaConfig) ParsedDuration() time.Duration {
|
|
fmt.Sprintf("duration str: %s", qc.DefaultDuration)
|
|
|
|
d, err := time.ParseDuration(qc.DefaultDuration)
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if d < time.Second {
|
|
panic("choose a longer default")
|
|
}
|
|
|
|
return d
|
|
}
|
|
|
|
type QuotaConfig struct {
|
|
DefaultSize string `toml:"default_size"`
|
|
DefaultDuration string `toml:"default_duration"`
|
|
}
|
|
|
|
|
|
type ServerConfig struct {
|
|
Port int64
|
|
Host string
|
|
}
|
|
|
|
|
|
type FileConfig struct {
|
|
Path string
|
|
}
|
|
|
|
|
|
type DatabaseConfig struct {
|
|
Backend string
|
|
Connection string
|
|
}
|
|
|
|
|
|
type Configuration struct {
|
|
Server ServerConfig
|
|
File FileConfig
|
|
Database DatabaseConfig
|
|
Quota QuotaConfig
|
|
}
|
|
|
|
func (c *Configuration) bindAddress() string {
|
|
return fmt.Sprintf("%s:%d", c.Server.Host, c.Server.Port)
|
|
}
|
|
|
|
|
|
func ConfigFromToml(toml_data string) (*Configuration, error) {
|
|
var config Configuration
|
|
|
|
_, err := toml.Decode(toml_data, &config)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &config, nil
|
|
}
|