staticrouter
This commit is contained in:
parent
440f0a8378
commit
6e3fe5e9ef
88
core/routers/staticrouter.go
Normal file
88
core/routers/staticrouter.go
Normal file
|
|
@ -0,0 +1,88 @@
|
||||||
|
package routers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
// StaticRouter is a filesystem-based router for static files
|
||||||
|
type StaticRouter struct {
|
||||||
|
rootDir string // Root directory containing files
|
||||||
|
routes map[string]string // Direct mapping from URL path to file path
|
||||||
|
mu sync.RWMutex // Lock for concurrent access to routes
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewStaticRouter creates a new StaticRouter instance
|
||||||
|
func NewStaticRouter(rootDir string) (*StaticRouter, error) {
|
||||||
|
// Verify root directory exists
|
||||||
|
info, err := os.Stat(rootDir)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if !info.IsDir() {
|
||||||
|
return nil, errors.New("root path is not a directory")
|
||||||
|
}
|
||||||
|
|
||||||
|
r := &StaticRouter{
|
||||||
|
rootDir: rootDir,
|
||||||
|
routes: make(map[string]string),
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build routes
|
||||||
|
if err := r.buildRoutes(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildRoutes scans the root directory and builds the routing map
|
||||||
|
func (r *StaticRouter) buildRoutes() error {
|
||||||
|
return filepath.Walk(r.rootDir, func(path string, info os.FileInfo, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Skip directories
|
||||||
|
if info.IsDir() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get relative path for URL
|
||||||
|
relPath, err := filepath.Rel(r.rootDir, path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Convert to URL path with forward slashes for consistency
|
||||||
|
urlPath := "/" + strings.ReplaceAll(relPath, "\\", "/")
|
||||||
|
|
||||||
|
// Add to routes map
|
||||||
|
r.routes[urlPath] = path
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Match finds a file path for the given URL path
|
||||||
|
func (r *StaticRouter) Match(path string) (string, bool) {
|
||||||
|
r.mu.RLock()
|
||||||
|
defer r.mu.RUnlock()
|
||||||
|
|
||||||
|
filePath, found := r.routes[path]
|
||||||
|
return filePath, found
|
||||||
|
}
|
||||||
|
|
||||||
|
// Refresh rebuilds the router by rescanning the root directory
|
||||||
|
func (r *StaticRouter) Refresh() error {
|
||||||
|
r.mu.Lock()
|
||||||
|
defer r.mu.Unlock()
|
||||||
|
|
||||||
|
// Clear routes
|
||||||
|
r.routes = make(map[string]string)
|
||||||
|
|
||||||
|
// Rebuild routes
|
||||||
|
return r.buildRoutes()
|
||||||
|
}
|
||||||
150
core/routers/staticrouter_test.go
Normal file
150
core/routers/staticrouter_test.go
Normal file
|
|
@ -0,0 +1,150 @@
|
||||||
|
package routers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func setupStaticFiles(t *testing.T) (string, func()) {
|
||||||
|
// Create a temporary directory
|
||||||
|
tempDir, err := os.MkdirTemp("", "staticrouter-test")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create temp directory: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create file structure
|
||||||
|
files := map[string]string{
|
||||||
|
"index.html": "<html>Home</html>",
|
||||||
|
"about.html": "<html>About</html>",
|
||||||
|
"api/index.json": `{"version": "1.0"}`,
|
||||||
|
"users/index.html": "<html>Users</html>",
|
||||||
|
"users/123/profile.html": "<html>User Profile</html>",
|
||||||
|
"posts/hello-world/comments.html": "<html>Post Comments</html>",
|
||||||
|
"docs/v1/api.html": "<html>API Docs</html>",
|
||||||
|
}
|
||||||
|
|
||||||
|
for path, content := range files {
|
||||||
|
filePath := filepath.Join(tempDir, path)
|
||||||
|
|
||||||
|
// Create directories
|
||||||
|
err := os.MkdirAll(filepath.Dir(filePath), 0755)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create directory %s: %v", filepath.Dir(filePath), err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create file
|
||||||
|
err = os.WriteFile(filePath, []byte(content), 0644)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create file %s: %v", filePath, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Return cleanup function
|
||||||
|
cleanup := func() {
|
||||||
|
os.RemoveAll(tempDir)
|
||||||
|
}
|
||||||
|
|
||||||
|
return tempDir, cleanup
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStaticRouterInitialization(t *testing.T) {
|
||||||
|
rootDir, cleanup := setupStaticFiles(t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
router, err := NewStaticRouter(rootDir)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create router: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if router == nil {
|
||||||
|
t.Fatal("Router is nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStaticRouteMatching(t *testing.T) {
|
||||||
|
rootDir, cleanup := setupStaticFiles(t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
router, err := NewStaticRouter(rootDir)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create router: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
path string
|
||||||
|
wantFound bool
|
||||||
|
wantHandler string
|
||||||
|
}{
|
||||||
|
{"/index.html", true, filepath.Join(rootDir, "index.html")},
|
||||||
|
{"/about.html", true, filepath.Join(rootDir, "about.html")},
|
||||||
|
{"/api/index.json", true, filepath.Join(rootDir, "api/index.json")},
|
||||||
|
{"/users/index.html", true, filepath.Join(rootDir, "users/index.html")},
|
||||||
|
{"/users/123/profile.html", true, filepath.Join(rootDir, "users/123/profile.html")},
|
||||||
|
{"/posts/hello-world/comments.html", true, filepath.Join(rootDir, "posts/hello-world/comments.html")},
|
||||||
|
{"/docs/v1/api.html", true, filepath.Join(rootDir, "docs/v1/api.html")},
|
||||||
|
|
||||||
|
// Non-existent routes
|
||||||
|
{"/nonexistent.html", false, ""},
|
||||||
|
{"/api/nonexistent.json", false, ""},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.path, func(t *testing.T) {
|
||||||
|
filePath, found := router.Match(tt.path)
|
||||||
|
|
||||||
|
if found != tt.wantFound {
|
||||||
|
t.Errorf("Match() found = %v, want %v", found, tt.wantFound)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !found {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if filePath != tt.wantHandler {
|
||||||
|
t.Errorf("Match() handler = %v, want %v", filePath, tt.wantHandler)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//TestStaticParamExtraction has been removed since we no longer extract parameters
|
||||||
|
|
||||||
|
func TestStaticRefresh(t *testing.T) {
|
||||||
|
rootDir, cleanup := setupStaticFiles(t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
router, err := NewStaticRouter(rootDir)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create router: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add a new file
|
||||||
|
newFilePath := filepath.Join(rootDir, "new.html")
|
||||||
|
err = os.WriteFile(newFilePath, []byte("<html>New</html>"), 0644)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Before refresh, file should not be found
|
||||||
|
_, found := router.Match("/new.html")
|
||||||
|
if found {
|
||||||
|
t.Errorf("New file should not be found before refresh")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Refresh router
|
||||||
|
err = router.Refresh()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to refresh router: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// After refresh, file should be found
|
||||||
|
filePath, found := router.Match("/new.html")
|
||||||
|
if !found {
|
||||||
|
t.Errorf("New file should be found after refresh")
|
||||||
|
}
|
||||||
|
|
||||||
|
if filePath != newFilePath {
|
||||||
|
t.Errorf("Expected path %s, got %s", newFilePath, filePath)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user