Cian-H/simple_blender_server
1
0
Fork 0
mirror of https://github.com/Cian-H/simple_blender_server.git synced 2025-12-23 05:41:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improved file handling to allow for parallel model generation

Browse Source
This commit is contained in:
Cian Hughes
2025-05-21 10:58:01 +01:00
parent 17f1ba0426
commit d54213daa8
1 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
main.go
View File

@@ -9,7 +9,6 @@ import (
"net/http" "net/http"
"os" "os"
"os/exec" "os/exec"
"path/filepath"
"text/template" "text/template"
) )
@@ -26,8 +25,6 @@ func main() {
} }
} }
const ModelPath = "./model.glb"
type Request struct { type Request struct {
ModelCode string `json:"model_code"` ModelCode string `json:"model_code"`
} }
@@ -60,16 +57,18 @@ func handleCreateModel(w http.ResponseWriter, r *http.Request) {
log.Printf("Successfully parsed request: ModelCode=\n```\n%s\n```\n", req.ModelCode) log.Printf("Successfully parsed request: ModelCode=\n```\n%s\n```\n", req.ModelCode)
absModelPath, err := filepath.Abs(ModelPath) tempFile, err := os.CreateTemp("", "model*.glb")
if err != nil { if err != nil {
log.Printf("Error getting absolute path: %v", err) log.Printf("Error creating temp file: %v", err)
http.Error(w, "Internal server error", http.StatusInternalServerError) http.Error(w, "Internal server error", http.StatusInternalServerError)
return return
} }
tempFilePath := tempFile.Name()
tempFile.Close()
log.Printf("Using temporary file path: %s", tempFilePath)
defer os.Remove(tempFilePath)
_ = os.Remove(absModelPath) model_expression, err := create_glb(req.ModelCode, tempFilePath)
model_expression, err := create_glb(req.ModelCode, absModelPath)
if err != nil { if err != nil {
http.Error(w, fmt.Sprintf("Error parsing templates: %v", err), http.StatusInternalServerError) http.Error(w, fmt.Sprintf("Error parsing templates: %v", err), http.StatusInternalServerError)
return return
@@ -103,13 +102,13 @@ func handleCreateModel(w http.ResponseWriter, r *http.Request) {
log.Printf("Blender command executed; output: %s", string(output)) log.Printf("Blender command executed; output: %s", string(output))
if _, err := os.Stat(absModelPath); os.IsNotExist(err) { if _, err := os.Stat(tempFilePath); os.IsNotExist(err) {
log.Printf("Error: GLB file was not created at %s", absModelPath) log.Printf("Error: GLB file was not created at %s", tempFilePath)
http.Error(w, "Failed to generate GLB file", http.StatusInternalServerError) http.Error(w, "Failed to generate GLB file", http.StatusInternalServerError)
return return
} }
glbData, err := os.ReadFile(absModelPath) glbData, err := os.ReadFile(tempFilePath)
if err != nil { if err != nil {
log.Printf("Error reading GLB file: %v", err) log.Printf("Error reading GLB file: %v", err)
http.Error(w, "Error reading generated GLB file", http.StatusInternalServerError) http.Error(w, "Error reading generated GLB file", http.StatusInternalServerError)
@@ -119,7 +118,7 @@ func handleCreateModel(w http.ResponseWriter, r *http.Request) {
log.Printf("GLB file read successfully, size: %d bytes", len(glbData)) log.Printf("GLB file read successfully, size: %d bytes", len(glbData))
w.Header().Set("Content-Type", "application/octet-stream") w.Header().Set("Content-Type", "application/octet-stream")
w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", filepath.Base(ModelPath))) w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=model.glb"))
w.Header().Set("Content-Length", fmt.Sprintf("%d", len(glbData))) w.Header().Set("Content-Length", fmt.Sprintf("%d", len(glbData)))
if _, err := w.Write(glbData); err != nil { if _, err := w.Write(glbData); err != nil {