houston server config에 storagepath 추가

This commit is contained in:
2023-06-13 10:10:30 +09:00
parent b14ad791df
commit 5c00ff73d7
6 changed files with 43 additions and 25 deletions

View File

@ -6,7 +6,6 @@ import (
"errors"
"fmt"
"io"
"io/fs"
"net/http"
"net/url"
"os"
@ -81,11 +80,11 @@ func unzip(fname string) error {
filePath := path.Join(verpath, name)
if f.FileInfo().IsDir() {
os.MkdirAll(filePath, os.ModePerm)
os.MkdirAll(filePath, 0775)
continue
}
if err := os.MkdirAll(path.Dir(filePath), os.ModePerm); err != nil {
if err := os.MkdirAll(path.Dir(filePath), 0775); err != nil {
return err
}
@ -133,7 +132,7 @@ func untar(fname string) error {
switch header.Typeflag {
case tar.TypeDir:
if err := os.MkdirAll(path.Join(verpath, header.Name), 0755); err != nil {
if err := os.MkdirAll(path.Join(verpath, header.Name), 0775); err != nil {
return err
}
case tar.TypeReg:
@ -171,7 +170,7 @@ func (hc *houstonClient) prepareDeploy(name string, version string) (destPath st
verpath := path.Join(name, version)
if _, err := os.Stat(verpath); os.IsNotExist(err) {
// 없네? 만들면 된다.
err = os.MkdirAll(verpath, fs.FileMode(os.O_WRONLY))
err = os.MkdirAll(verpath, 0775)
if err != nil {
return "", err
}

View File

@ -177,6 +177,8 @@ func prepareProcessLaunch(req *shared.StartProcessRequest) *procmeta {
if err == nil && fi.IsDir() {
args[0] = "./" + path.Clean(strings.TrimPrefix(args[0], "/"))
os.Chmod(path.Join(verpath, args[0]), 0777)
cmd := exec.Command(args[0], args[1:]...)
cmd.Dir = verpath
stdin, _ := cmd.StdinPipe()
@ -204,7 +206,7 @@ func (hc *houstonClient) launch(meta *procmeta) error {
return err
}
err = os.MkdirAll(path.Join(meta.cmd.Dir, "logs"), os.ModePerm)
err = os.MkdirAll(path.Join(meta.cmd.Dir, "logs"), 0775)
if err != nil {
return err
}

View File

@ -73,7 +73,7 @@ func main() {
}
if ent.IsDir() {
if err := os.MkdirAll(ent.Name(), os.ModePerm); err != nil {
if err := os.MkdirAll(ent.Name(), 0775); err != nil {
panic(err)
}
} else {

View File

@ -29,7 +29,7 @@ func (h *houstonHandler) GetDeploySources(w http.ResponseWriter, r *http.Request
files, err := os.ReadDir("deploys")
if err != nil {
if errors.Is(err, fs.ErrNotExist) {
err = os.MkdirAll("deploys", os.ModePerm)
err = os.MkdirAll("deploys", 0775)
}
}
@ -96,7 +96,7 @@ func (h *houstonHandler) UploadDeploySource(w http.ResponseWriter, r *http.Reque
filename = path.Join("deploys", name, version, name+ext)
}
if err = os.MkdirAll(path.Dir(filename), os.ModePerm); err != nil {
if err = os.MkdirAll(path.Dir(filename), 0775); err != nil {
logger.Error(err)
w.WriteHeader(http.StatusInternalServerError)
return

View File

@ -43,16 +43,22 @@ func NewHoustonHandler() HoustonServerWithHandler {
}
func (h *houstonHandler) RegisterHandlers(serveMux *http.ServeMux, prefix string) error {
logger.Println("houstonHandler registed")
storagePath := loadConfig().StoragePath
logger.Println("houstonHandler registed. storage_path :", storagePath)
if len(storagePath) > 0 {
storagePath = storagePath + "/"
}
if len(prefix) > 0 {
prefix = "/" + prefix
}
serveMux.Handle(prefix, h)
fsx := http.FileServer(http.Dir("deploys"))
fsx := http.FileServer(http.Dir(storagePath + "deploys"))
serveMux.Handle(fmt.Sprintf("%s/deploys/", prefix), http.StripPrefix(fmt.Sprintf("%s/deploys/", prefix), fsx))
ufsx := http.FileServer(http.Dir("downloads"))
ufsx := http.FileServer(http.Dir(storagePath + "downloads"))
serveMux.Handle(fmt.Sprintf("%s/downloads/", prefix), http.StripPrefix(fmt.Sprintf("%s/downloads/", prefix), ufsx))
serveMux.HandleFunc(fmt.Sprintf("%s/upload", prefix), func(w http.ResponseWriter, r *http.Request) {
@ -69,8 +75,8 @@ func (h *houstonHandler) RegisterHandlers(serveMux *http.ServeMux, prefix string
name := r.Header.Get("Houston-Service-Name")
version := r.Header.Get("Houston-Service-Version")
filename := r.Header.Get("Houston-Service-Filename")
dir := fmt.Sprintf("downloads/%s/%s", name, version)
if err := os.MkdirAll(dir, os.ModePerm); err == nil {
dir := fmt.Sprintf(storagePath+"downloads/%s/%s", name, version)
if err := os.MkdirAll(dir, 0775); err == nil {
file, _ := os.Create(path.Join(dir, filename))
if file != nil {
defer file.Close()

View File

@ -21,7 +21,8 @@ type HoustonServer interface {
}
type serverConfig struct {
GrpcPort int `json:"grpc_port"`
GrpcPort int `json:"grpc_port"`
StoragePath string `json:"storage_path"`
}
type totalConfig struct {
@ -110,17 +111,27 @@ type Operation interface {
Hosts() map[string]hostSnapshot
}
func NewServer() HoustonServer {
port := 8080
if bt, err := os.ReadFile("config.json"); err == nil {
var config totalConfig
if err := json.Unmarshal(bt, &config); err == nil {
if config.Server.GrpcPort != 0 {
port = config.Server.GrpcPort
}
}
func loadConfig() serverConfig {
var config totalConfig
config.Server.GrpcPort = 8080
configFile, err := os.Open("config.json")
if err != nil {
logger.Error(err)
return config.Server
}
defer configFile.Close()
dec := json.NewDecoder(configFile)
err = dec.Decode(&config)
if err != nil {
logger.Error(err)
return config.Server
}
return config.Server
}
func NewServer() HoustonServer {
var opts []grpc.ServerOption
grpcServer := grpc.NewServer(opts...)
@ -133,7 +144,7 @@ func NewServer() HoustonServer {
rpcServer: grpcServer,
os: os,
ms: ms,
port: port,
port: loadConfig().GrpcPort,
}
}