Compare commits
5 Commits
b7729e8b39
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| 42b0493adb | |||
| b3c1f37632 | |||
| 40ded4d4db | |||
| 291cfa33a9 | |||
| 74fef30251 |
@@ -10,6 +10,7 @@ import (
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"syscall"
|
||||
@@ -55,6 +56,11 @@ func loadConfig() AppConfig {
|
||||
m.PeerTTL = parseDuration(os.Getenv("MESH_PEER_TTL"), 2*time.Minute)
|
||||
m.PruneInterval = parseDuration(os.Getenv("MESH_PRUNE_INTERVAL"), 30*time.Second)
|
||||
|
||||
m.HelloInterval = parseDuration(os.Getenv("MESH_HELLO_INTERVAL"), 20*time.Second)
|
||||
m.HelloFanout = parseIntEnv(os.Getenv("MESH_HELLO_FANOUT"), 8)
|
||||
|
||||
m.BlobTimeout = parseDuration(os.Getenv("MESH_BLOB_TIMEOUT"), 0)
|
||||
|
||||
// Wenn keine AdvertURL gesetzt ist, versuche eine sinnvolle Herleitung:
|
||||
if strings.TrimSpace(m.AdvertURL) == "" {
|
||||
m.AdvertURL = inferAdvertURL(m.BindAddr)
|
||||
@@ -101,6 +107,18 @@ func parseBoolEnv(k string, def bool) bool {
|
||||
return v == "1" || v == "true" || v == "yes" || v == "on"
|
||||
}
|
||||
|
||||
func parseIntEnv(k string, def int) int {
|
||||
v := strings.TrimSpace(os.Getenv(k))
|
||||
if v == "" {
|
||||
return def
|
||||
}
|
||||
n, err := strconv.Atoi(v)
|
||||
if err != nil {
|
||||
return def
|
||||
}
|
||||
return n
|
||||
}
|
||||
|
||||
func splitCSV(s string) []string {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
@@ -465,11 +483,14 @@ func main() {
|
||||
|
||||
// Domain-Store (mesh-fähig)
|
||||
nodeID := strings.TrimSpace(cfg.Mesh.AdvertURL)
|
||||
st := filesvc.NewMemStore(nodeID)
|
||||
//st := filesvc.NewMemStore(nodeID)
|
||||
|
||||
// Mesh starten
|
||||
//mcfg := mesh.FromEnv()
|
||||
blobs := blobfs.New(getenvDefault("DATA_DIR", "./data"))
|
||||
dataDir := getenvDefault("DATA_DIR", "./data")
|
||||
metaPath := filepath.Join(dataDir, "meta", "items.json")
|
||||
st := filesvc.NewMemStorePersistent(nodeID, metaPath)
|
||||
|
||||
mnode, err := mesh.New(cfg.Mesh, mesh.Callbacks{
|
||||
GetSnapshot: func(ctx context.Context) (mesh.Snapshot, error) {
|
||||
@@ -620,36 +641,28 @@ func registerPublicDownloads(mux *http.ServeMux, store filesvc.MeshStore, blobs
|
||||
base = "/" + base
|
||||
}
|
||||
mux.HandleFunc(base+"/", func(w http.ResponseWriter, r *http.Request) {
|
||||
idStr := strings.TrimPrefix(r.URL.Path, base+"/")
|
||||
if idStr == "" {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
id := idStr
|
||||
if strings.TrimSpace(id) == "" {
|
||||
id := strings.TrimSpace(strings.TrimPrefix(r.URL.Path, base+"/"))
|
||||
if id == "" {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
// 1) Metadaten prüfen (nicht vorhanden oder gelöscht → 404)
|
||||
// 1) Metadaten prüfen
|
||||
it, err := store.Get(r.Context(), filesvc.ID(id))
|
||||
if err != nil || it.Deleted {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
// 2) lokal versuchen
|
||||
if rc, meta, err := blobs.Open(r.Context(), id); err == nil {
|
||||
// 2) Lokal versuchen
|
||||
if rc, meta, err := blobs.Open(context.Background(), id); err == nil {
|
||||
defer rc.Close()
|
||||
serveBlob(w, r, rc, meta, it.Name) // Download-Name aus Store!
|
||||
serveBlob(w, r, rc, meta, it.Name)
|
||||
return
|
||||
} else { // ########## HIER EVENTUELL PROBLEM!!! ###########
|
||||
// Lokal nicht vorhanden → nur aus Mesh ziehen, wenn Owner aktiv ist
|
||||
it, err := store.Get(r.Context(), filesvc.ID(id))
|
||||
if err != nil || it.Deleted {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// (Optional) Owner-Online-Check — wenn du auch bei offline Ownern liefern willst, block auskommentieren
|
||||
{
|
||||
peers := meshNode.PeerList()
|
||||
ttl := 2 * time.Minute
|
||||
if !isOwnerActive(it.Owner, peers, ttl) {
|
||||
@@ -658,26 +671,34 @@ func registerPublicDownloads(mux *http.ServeMux, store filesvc.MeshStore, blobs
|
||||
}
|
||||
}
|
||||
|
||||
// 3) aus Mesh holen (signiert) und cachen
|
||||
rrc, _, _, _, err := meshNode.FetchBlobAny(r.Context(), id)
|
||||
// 3) Aus Mesh holen — EIGENER Timeout-Kontext, NICHT r.Context()
|
||||
rrc, remoteName, _, _, err := meshNode.FetchBlobAny(context.Background(), id)
|
||||
if err != nil {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
defer rrc.Close()
|
||||
if _, err := blobs.Save(r.Context(), id, it.Name, rrc); err != nil {
|
||||
http.Error(w, "cache failed", http.StatusInternalServerError)
|
||||
|
||||
filename := strings.TrimSpace(remoteName)
|
||||
if filename == "" {
|
||||
filename = it.Name
|
||||
}
|
||||
|
||||
// 4) Lokal cachen — KEIN Request-Kontext, damit Save nicht abbricht
|
||||
if _, err := blobs.Save(context.Background(), id, filename, rrc); err != nil {
|
||||
log.Printf("[public] cache save failed id=%s name=%q: %v", id, filename, err)
|
||||
http.Error(w, "cache failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// 4) erneut lokal öffnen und streamen
|
||||
lrc, meta, err := blobs.Open(r.Context(), id)
|
||||
// 5) Erneut lokal öffnen und streamen
|
||||
lrc, meta, err := blobs.Open(context.Background(), id)
|
||||
if err != nil {
|
||||
http.Error(w, "open failed", http.StatusInternalServerError)
|
||||
http.Error(w, "open failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer lrc.Close()
|
||||
serveBlob(w, r, lrc, meta, it.Name)
|
||||
serveBlob(w, r, lrc, meta, filename)
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
@@ -2,6 +2,9 @@ package filesvc
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"slices"
|
||||
"strings"
|
||||
"sync"
|
||||
@@ -13,13 +16,82 @@ type MemStore struct {
|
||||
items map[ID]File
|
||||
self string
|
||||
// optionales Eventing
|
||||
subs []chan ChangeEvent
|
||||
subs []chan ChangeEvent
|
||||
persistPath string
|
||||
}
|
||||
|
||||
func NewMemStore(self string) *MemStore {
|
||||
return &MemStore{self: strings.TrimSpace(self), items: make(map[ID]File)}
|
||||
}
|
||||
|
||||
func NewMemStorePersistent(self, path string) *MemStore {
|
||||
m := NewMemStore(self)
|
||||
m.persistPath = strings.TrimSpace(path)
|
||||
// beim Start versuchen zu laden
|
||||
_ = m.loadFromDisk()
|
||||
return m
|
||||
}
|
||||
|
||||
// --- Persistenz-Helper (NEU) ---
|
||||
|
||||
func (m *MemStore) loadFromDisk() error {
|
||||
if m.persistPath == "" {
|
||||
return nil
|
||||
}
|
||||
f, err := os.Open(m.persistPath)
|
||||
if err != nil {
|
||||
return nil // Datei existiert beim ersten Start nicht – ok
|
||||
}
|
||||
defer f.Close()
|
||||
var snap Snapshot
|
||||
if err := json.NewDecoder(f).Decode(&snap); err != nil {
|
||||
return err
|
||||
}
|
||||
m.mu.Lock()
|
||||
for _, it := range snap.Items {
|
||||
m.items[it.ID] = it
|
||||
}
|
||||
m.mu.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *MemStore) saveLocked() error {
|
||||
if m.persistPath == "" {
|
||||
return nil
|
||||
}
|
||||
if err := os.MkdirAll(filepath.Dir(m.persistPath), 0o755); err != nil {
|
||||
return err
|
||||
}
|
||||
// Snapshot aus Map bauen
|
||||
snap := Snapshot{Items: make([]File, 0, len(m.items))}
|
||||
for _, it := range m.items {
|
||||
snap.Items = append(snap.Items, it)
|
||||
}
|
||||
// atomar schreiben
|
||||
tmp := m.persistPath + ".tmp"
|
||||
f, err := os.Create(tmp)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
enc := json.NewEncoder(f)
|
||||
enc.SetIndent("", " ")
|
||||
if err := enc.Encode(&snap); err != nil {
|
||||
f.Close()
|
||||
_ = os.Remove(tmp)
|
||||
return err
|
||||
}
|
||||
if err := f.Sync(); err != nil {
|
||||
f.Close()
|
||||
_ = os.Remove(tmp)
|
||||
return err
|
||||
}
|
||||
if err := f.Close(); err != nil {
|
||||
_ = os.Remove(tmp)
|
||||
return err
|
||||
}
|
||||
return os.Rename(tmp, m.persistPath)
|
||||
}
|
||||
|
||||
/*** Store ***/
|
||||
|
||||
func (m *MemStore) Get(_ context.Context, id ID) (File, error) {
|
||||
@@ -102,6 +174,7 @@ func (m *MemStore) Create(_ context.Context, name string) (File, error) {
|
||||
}
|
||||
it := File{ID: uid, Name: name, UpdatedAt: now, Owner: m.self}
|
||||
m.items[it.ID] = it
|
||||
_ = m.saveLocked()
|
||||
m.emit(it)
|
||||
return it, nil
|
||||
}
|
||||
@@ -124,6 +197,7 @@ func (m *MemStore) Rename(_ context.Context, id ID, newName string) (File, error
|
||||
it.Name = strings.TrimSpace(newName)
|
||||
it.UpdatedAt = time.Now().UnixNano()
|
||||
m.items[id] = it
|
||||
_ = m.saveLocked()
|
||||
m.emit(it)
|
||||
return it, nil
|
||||
}
|
||||
@@ -144,6 +218,7 @@ func (m *MemStore) Delete(_ context.Context, id ID) (File, error) {
|
||||
it.Deleted = true
|
||||
it.UpdatedAt = time.Now().UnixNano()
|
||||
m.items[id] = it
|
||||
_ = m.saveLocked()
|
||||
m.emit(it)
|
||||
return it, nil
|
||||
}
|
||||
@@ -169,6 +244,7 @@ func (m *MemStore) TakeoverOwner(_ context.Context, id ID, newOwner string) (Fil
|
||||
it.Owner = newOwner
|
||||
it.UpdatedAt = time.Now().UnixNano()
|
||||
m.items[id] = it
|
||||
_ = m.saveLocked()
|
||||
m.emitLocked(it)
|
||||
return it, nil
|
||||
}
|
||||
@@ -188,6 +264,7 @@ func (m *MemStore) Snapshot(_ context.Context) (Snapshot, error) {
|
||||
func (m *MemStore) ApplyRemote(_ context.Context, s Snapshot) error {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
changed := false
|
||||
for _, ri := range s.Items {
|
||||
li, ok := m.items[ri.ID]
|
||||
if !ok || ri.UpdatedAt > li.UpdatedAt {
|
||||
@@ -196,9 +273,13 @@ func (m *MemStore) ApplyRemote(_ context.Context, s Snapshot) error {
|
||||
ri.Owner = li.Owner
|
||||
}
|
||||
m.items[ri.ID] = ri
|
||||
changed = true
|
||||
m.emitLocked(ri)
|
||||
}
|
||||
}
|
||||
if changed {
|
||||
_ = m.saveLocked() // ← NEU
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
@@ -12,6 +12,7 @@ import (
|
||||
"hash/crc32"
|
||||
"io"
|
||||
"log"
|
||||
"math/rand/v2"
|
||||
"net"
|
||||
"net/http"
|
||||
"os"
|
||||
@@ -33,6 +34,15 @@ type Config struct {
|
||||
DiscoveryAddress string // "239.8.8.8:9898"
|
||||
PeerTTL time.Duration // wie lange darf ein Peer inaktiv sein (Default siehe unten)
|
||||
PruneInterval time.Duration // wie oft wird gepruned
|
||||
|
||||
SyncInterval time.Duration
|
||||
Fanout int
|
||||
|
||||
// NEU:
|
||||
HelloInterval time.Duration // wie oft pingen
|
||||
HelloFanout int // wie viele Peers pro Tick
|
||||
|
||||
BlobTimeout time.Duration
|
||||
}
|
||||
|
||||
type Peer struct {
|
||||
@@ -190,21 +200,38 @@ func (n *Node) helloHandler(w http.ResponseWriter, r *http.Request) {
|
||||
http.Error(w, "bad signature", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
var p Peer
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
var req struct {
|
||||
URL string `json:"url"`
|
||||
}
|
||||
if err := json.Unmarshal(body, &req); err != nil || strings.TrimSpace(req.URL) == "" {
|
||||
http.Error(w, "bad json", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
p.LastSeen = time.Now()
|
||||
|
||||
// Peer anlegen (falls neu) und LastSeen setzen
|
||||
n.mu.Lock()
|
||||
if existing, ok := n.peers[p.URL]; ok {
|
||||
existing.LastSeen = p.LastSeen
|
||||
} else if p.URL != n.self.URL {
|
||||
cp := p
|
||||
n.peers[p.URL] = &cp
|
||||
if req.URL != n.self.URL {
|
||||
if p, ok := n.peers[req.URL]; ok {
|
||||
p.LastSeen = time.Now()
|
||||
} else {
|
||||
cp := Peer{URL: req.URL, LastSeen: time.Now()} // weitere Felder wie bekannt
|
||||
n.peers[req.URL] = &cp
|
||||
}
|
||||
}
|
||||
n.mu.Unlock()
|
||||
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}
|
||||
|
||||
func (n *Node) touchPeer(url string) {
|
||||
if strings.TrimSpace(url) == "" {
|
||||
return
|
||||
}
|
||||
n.mu.Lock()
|
||||
if p, ok := n.peers[url]; ok {
|
||||
p.LastSeen = time.Now()
|
||||
}
|
||||
n.mu.Unlock()
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}
|
||||
|
||||
func (n *Node) peersHandler(w http.ResponseWriter, r *http.Request) {
|
||||
@@ -283,6 +310,12 @@ func (n *Node) Serve() error {
|
||||
n.loopAntiEntropy()
|
||||
}()
|
||||
|
||||
n.wg.Add(1)
|
||||
go func() {
|
||||
defer n.wg.Done()
|
||||
n.loopHello()
|
||||
}()
|
||||
|
||||
// http server
|
||||
errc := make(chan error, 1)
|
||||
go func() {
|
||||
@@ -467,28 +500,100 @@ func (n *Node) loopBeaconRecv() {
|
||||
/*** Outgoing ***/
|
||||
|
||||
func (n *Node) sendHello(url string) error {
|
||||
p := n.self
|
||||
b, _ := json.Marshal(p)
|
||||
b, _ := json.Marshal(struct {
|
||||
URL string `json:"url"`
|
||||
}{URL: n.self.URL})
|
||||
req, _ := http.NewRequest(http.MethodPost, strings.TrimRight(url, "/")+"/mesh/hello", bytes.NewReader(b))
|
||||
req.Header.Set("X-Mesh-Sig", n.sign(b))
|
||||
|
||||
resp, err := n.client.Do(req)
|
||||
if err == nil {
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
resp.Body.Close()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
resp.Body.Close()
|
||||
if resp.StatusCode == http.StatusOK {
|
||||
n.touchPeer(url) // eigenen Blick auf den Peer direkt als „gesehen“ markieren
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("hello %s: %s", url, resp.Status)
|
||||
}
|
||||
|
||||
func (n *Node) loopHello() {
|
||||
interval := n.cfg.HelloInterval
|
||||
if interval <= 0 {
|
||||
interval = 20 * time.Second
|
||||
}
|
||||
fanout := n.cfg.HelloFanout
|
||||
if fanout <= 0 {
|
||||
fanout = 8
|
||||
}
|
||||
|
||||
t := time.NewTicker(interval)
|
||||
defer t.Stop()
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-n.stop:
|
||||
return
|
||||
case <-t.C:
|
||||
}
|
||||
|
||||
// Liste der *bekannten* Peers (nicht nur Seeds)
|
||||
n.mu.RLock()
|
||||
targets := make([]string, 0, len(n.peers))
|
||||
for url := range n.peers {
|
||||
if url != n.self.URL {
|
||||
targets = append(targets, url)
|
||||
}
|
||||
}
|
||||
n.mu.RUnlock()
|
||||
if len(targets) == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
// zufällig mischen und auf Fanout begrenzen
|
||||
rand.Shuffle(len(targets), func(i, j int) { targets[i], targets[j] = targets[j], targets[i] })
|
||||
if fanout < len(targets) {
|
||||
targets = targets[:fanout]
|
||||
}
|
||||
|
||||
// leicht parallel pingen
|
||||
var wg sync.WaitGroup
|
||||
for _, u := range targets {
|
||||
u := u
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
_ = n.sendHello(u)
|
||||
}()
|
||||
}
|
||||
wg.Wait()
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *Node) sendSync(url string, s Snapshot) error {
|
||||
b, _ := json.Marshal(s)
|
||||
req, _ := http.NewRequest(http.MethodPost, strings.TrimRight(url, "/")+"/mesh/sync", bytes.NewReader(b))
|
||||
req.Header.Set("X-Mesh-Sig", n.sign(b))
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)
|
||||
defer cancel()
|
||||
req = req.WithContext(ctx)
|
||||
|
||||
resp, err := n.client.Do(req)
|
||||
if err == nil {
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
resp.Body.Close()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return err
|
||||
defer resp.Body.Close()
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return fmt.Errorf("sync %s: %s", url, resp.Status)
|
||||
}
|
||||
|
||||
n.touchPeer(url)
|
||||
return nil
|
||||
}
|
||||
|
||||
// PeerList liefert eine Kopie der bekannten Peers inkl. Self.
|
||||
@@ -608,14 +713,37 @@ func (n *Node) blobHandler(w http.ResponseWriter, r *http.Request) {
|
||||
_, _ = io.Copy(w, rc)
|
||||
}
|
||||
|
||||
// interner Helper: signierter Blob-Request an einen Peer
|
||||
func (n *Node) sendBlobRequest(url string, id string) (*http.Response, error) {
|
||||
// sendBlobRequest schickt eine signierte Anfrage an /mesh/blob und liefert die Response
|
||||
// zurück (Caller MUSS resp.Body schließen!). Bei HTTP 200 wird der Peer als gesehen markiert.
|
||||
func (n *Node) sendBlobRequest(url, id string) (*http.Response, error) {
|
||||
b, _ := json.Marshal(struct {
|
||||
ID string `json:"id"`
|
||||
}{ID: id})
|
||||
|
||||
req, _ := http.NewRequest(http.MethodPost, strings.TrimRight(url, "/")+"/mesh/blob", bytes.NewReader(b))
|
||||
req.Header.Set("X-Mesh-Sig", n.sign(b))
|
||||
return n.client.Do(req)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
// ❗ WICHTIG: kein kurzer Timeout. Optional: großer Timeout aus Config
|
||||
ctx := context.Background()
|
||||
if n.cfg.BlobTimeout > 0 {
|
||||
var cancel context.CancelFunc
|
||||
ctx, cancel = context.WithTimeout(context.Background(), n.cfg.BlobTimeout)
|
||||
defer cancel()
|
||||
}
|
||||
req = req.WithContext(ctx)
|
||||
|
||||
resp, err := n.client.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.StatusCode == http.StatusOK {
|
||||
n.touchPeer(url) // ausgehender Erfolg zählt als "gesehen"
|
||||
return resp, nil
|
||||
}
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
resp.Body.Close()
|
||||
return nil, fmt.Errorf("blob %s from %s: %s", id, url, resp.Status)
|
||||
}
|
||||
|
||||
// Öffentliche Methode: versuche Blob bei irgendeinem Peer zu holen
|
||||
|
||||
Reference in New Issue
Block a user