2013-12-02 17:37:36 +08:00
|
|
|
package weed_server
|
|
|
|
|
|
|
|
import (
|
|
|
|
"math/rand"
|
2013-12-10 05:27:09 +08:00
|
|
|
"net/http"
|
2014-04-26 13:09:42 +08:00
|
|
|
"strconv"
|
2013-12-02 17:37:36 +08:00
|
|
|
"time"
|
2014-10-27 02:34:55 +08:00
|
|
|
|
|
|
|
"github.com/chrislusf/weed-fs/go/glog"
|
|
|
|
"github.com/chrislusf/weed-fs/go/storage"
|
2013-12-02 17:37:36 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
type VolumeServer struct {
|
|
|
|
masterNode string
|
|
|
|
pulseSeconds int
|
|
|
|
dataCenter string
|
|
|
|
rack string
|
|
|
|
whiteList []string
|
|
|
|
store *storage.Store
|
2014-05-15 16:08:00 +08:00
|
|
|
|
|
|
|
FixJpgOrientation bool
|
2013-12-02 17:37:36 +08:00
|
|
|
}
|
|
|
|
|
2014-04-26 13:09:42 +08:00
|
|
|
func NewVolumeServer(r *http.ServeMux, ip string, port int, publicIp string, folders []string, maxCounts []int,
|
2013-12-02 17:37:36 +08:00
|
|
|
masterNode string, pulseSeconds int,
|
|
|
|
dataCenter string, rack string,
|
2014-05-15 16:08:00 +08:00
|
|
|
whiteList []string, fixJpgOrientation bool) *VolumeServer {
|
2014-04-26 13:09:42 +08:00
|
|
|
publicUrl := publicIp + ":" + strconv.Itoa(port)
|
2013-12-02 17:37:36 +08:00
|
|
|
vs := &VolumeServer{
|
2014-05-15 16:08:00 +08:00
|
|
|
masterNode: masterNode,
|
|
|
|
pulseSeconds: pulseSeconds,
|
|
|
|
dataCenter: dataCenter,
|
|
|
|
rack: rack,
|
|
|
|
whiteList: whiteList,
|
|
|
|
FixJpgOrientation: fixJpgOrientation,
|
2013-12-02 17:37:36 +08:00
|
|
|
}
|
|
|
|
vs.store = storage.NewStore(port, ip, publicUrl, folders, maxCounts)
|
|
|
|
|
|
|
|
r.HandleFunc("/status", secure(vs.whiteList, vs.statusHandler))
|
|
|
|
r.HandleFunc("/admin/assign_volume", secure(vs.whiteList, vs.assignVolumeHandler))
|
|
|
|
r.HandleFunc("/admin/vacuum_volume_check", secure(vs.whiteList, vs.vacuumVolumeCheckHandler))
|
|
|
|
r.HandleFunc("/admin/vacuum_volume_compact", secure(vs.whiteList, vs.vacuumVolumeCompactHandler))
|
|
|
|
r.HandleFunc("/admin/vacuum_volume_commit", secure(vs.whiteList, vs.vacuumVolumeCommitHandler))
|
|
|
|
r.HandleFunc("/admin/freeze_volume", secure(vs.whiteList, vs.freezeVolumeHandler))
|
2014-03-11 02:43:54 +08:00
|
|
|
r.HandleFunc("/admin/delete_collection", secure(vs.whiteList, vs.deleteCollectionHandler))
|
2014-03-26 04:46:59 +08:00
|
|
|
r.HandleFunc("/stats/counter", secure(vs.whiteList, statsCounterHandler))
|
|
|
|
r.HandleFunc("/stats/memory", secure(vs.whiteList, statsMemoryHandler))
|
2014-03-27 04:22:27 +08:00
|
|
|
r.HandleFunc("/stats/disk", secure(vs.whiteList, vs.statsDiskHandler))
|
2014-04-14 16:00:09 +08:00
|
|
|
r.HandleFunc("/delete", secure(vs.whiteList, vs.batchDeleteHandler))
|
2013-12-02 17:37:36 +08:00
|
|
|
r.HandleFunc("/", vs.storeHandler)
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
connected := true
|
2014-02-15 09:10:49 +08:00
|
|
|
vs.store.SetBootstrapMaster(vs.masterNode)
|
2013-12-02 17:37:36 +08:00
|
|
|
vs.store.SetDataCenter(vs.dataCenter)
|
|
|
|
vs.store.SetRack(vs.rack)
|
|
|
|
for {
|
2014-04-21 14:28:05 +08:00
|
|
|
master, err := vs.store.Join()
|
2013-12-02 17:37:36 +08:00
|
|
|
if err == nil {
|
|
|
|
if !connected {
|
|
|
|
connected = true
|
2014-04-21 14:28:05 +08:00
|
|
|
glog.V(0).Infoln("Volume Server Connected with master at", master)
|
2013-12-02 17:37:36 +08:00
|
|
|
}
|
|
|
|
} else {
|
2014-04-17 14:43:27 +08:00
|
|
|
glog.V(4).Infoln("Volume Server Failed to talk with master:", err.Error())
|
2013-12-02 17:37:36 +08:00
|
|
|
if connected {
|
|
|
|
connected = false
|
|
|
|
}
|
|
|
|
}
|
2014-02-15 09:10:49 +08:00
|
|
|
if connected {
|
|
|
|
time.Sleep(time.Duration(float32(vs.pulseSeconds*1e3)*(1+rand.Float32())) * time.Millisecond)
|
|
|
|
} else {
|
2014-03-03 14:16:54 +08:00
|
|
|
time.Sleep(time.Duration(float32(vs.pulseSeconds*1e3)*0.25) * time.Millisecond)
|
2014-02-15 09:10:49 +08:00
|
|
|
}
|
2013-12-02 17:37:36 +08:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
|
|
|
return vs
|
|
|
|
}
|
2014-05-13 15:03:10 +08:00
|
|
|
|
|
|
|
func (vs *VolumeServer) Shutdown() {
|
|
|
|
glog.V(0).Infoln("Shutting down volume server...")
|
|
|
|
vs.store.Close()
|
|
|
|
glog.V(0).Infoln("Shut down successfully!")
|
|
|
|
}
|