|
|
package weed_server
import ( "github.com/chrislusf/seaweedfs/weed/glog" "pack.ag/tftp")
func (vs *VolumeServer) ServeTFTP(r tftp.ReadRequest) {
filename := r.Name()
volumeId, n, err := vs.parseFileId(filename) if err != nil { glog.Errorf("parse file id %s: %v", filename, err) return }
hasVolume := vs.store.HasVolume(volumeId) _, hasEcVolume := vs.store.FindEcVolume(volumeId)
if hasVolume { if _, err = vs.store.ReadVolumeNeedle(volumeId, n, nil); err != nil { glog.Errorf("ReadVolumeNeedle %s: %v", filename, err) return } } if hasEcVolume { if _, err = vs.store.ReadEcShardNeedle(volumeId, n); err != nil { glog.Errorf("ReadEcShardNeedle %s: %v", filename, err) return } }
if _, err = r.Write(n.Data); err != nil { glog.Errorf("UDP Write data %s: %v", filename, err) return }
}
func (vs *VolumeServer) ReceiveTFTP(w tftp.WriteRequest) {
filename := w.Name() println("+ ", filename)
// Get the file size
size, err := w.Size()
// Note: The size value is sent by the client, the client could send more data than
// it indicated in the size option. To be safe we'd want to allocate a buffer
// with the size we're expecting and use w.Read(buf) rather than ioutil.ReadAll.
if filename[0] == '-' { err = vs.handleTcpDelete(filename[1:]) if err != nil { glog.Errorf("handleTcpDelete %s: %v", filename, err) return } }
volumeId, n, err := vs.parseFileId(filename) if err != nil { glog.Errorf("parse file id %s: %v", filename, err) return }
volume := vs.store.GetVolume(volumeId) if volume == nil { glog.Errorf("volume %d not found", volumeId) return }
err = volume.StreamWrite(n, w, uint32(size)) if err != nil { glog.Errorf("StreamWrite %s: %v", filename, err) return }
println("- ", filename)
}
|
