LHHDZ
3 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
33 changed files with 588 additions and 98 deletions
-
2.github/workflows/container_dev.yml
-
2.github/workflows/container_latest.yml
-
2.github/workflows/container_release1.yml
-
2.github/workflows/container_release2.yml
-
2.github/workflows/container_release3.yml
-
2.github/workflows/container_release4.yml
-
2.github/workflows/container_release5.yml
-
25go.mod
-
44go.sum
-
4k8s/helm_charts2/Chart.yaml
-
16other/metrics/grafana_seaweedfs.json
-
4weed/command/command.go
-
3weed/command/filer.go
-
25weed/command/filer_sync.go
-
6weed/command/server.go
-
382weed/command/update.go
-
9weed/command/update_full.go
-
3weed/command/volume.go
-
7weed/filer/filechunk_manifest.go
-
8weed/filer/filechunks.go
-
6weed/s3api/stats.go
-
4weed/server/common.go
-
4weed/server/filer_grpc_server_sub_meta.go
-
3weed/server/filer_server.go
-
7weed/server/filer_server_handlers_read_dir.go
-
12weed/server/filer_ui/filer.html
-
6weed/server/volume_server.go
-
22weed/server/volume_server_handlers.go
-
23weed/stats/metrics.go
-
7weed/storage/needle/needle_read_page.go
-
2weed/storage/needle_map/compact_map.go
-
2weed/util/constants.go
-
2weed/wdclient/vid_map.go
@ -1,5 +1,5 @@ |
|||
apiVersion: v1 |
|||
description: SeaweedFS |
|||
name: seaweedfs |
|||
appVersion: "3.10" |
|||
version: "3.10" |
|||
appVersion: "3.11" |
|||
version: "3.11" |
|||
@ -0,0 +1,382 @@ |
|||
package command |
|||
|
|||
import ( |
|||
"archive/tar" |
|||
"archive/zip" |
|||
"bytes" |
|||
"compress/gzip" |
|||
"context" |
|||
"crypto/md5" |
|||
"encoding/hex" |
|||
"encoding/json" |
|||
"fmt" |
|||
"io" |
|||
"io/ioutil" |
|||
"net/http" |
|||
"os" |
|||
"path/filepath" |
|||
"runtime" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/chrislusf/seaweedfs/weed/glog" |
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
"golang.org/x/net/context/ctxhttp" |
|||
) |
|||
|
|||
//copied from https://github.com/restic/restic/tree/master/internal/selfupdate
|
|||
|
|||
// Release collects data about a single release on GitHub.
|
|||
type Release struct { |
|||
Name string `json:"name"` |
|||
TagName string `json:"tag_name"` |
|||
Draft bool `json:"draft"` |
|||
PreRelease bool `json:"prerelease"` |
|||
PublishedAt time.Time `json:"published_at"` |
|||
Assets []Asset `json:"assets"` |
|||
|
|||
Version string `json:"-"` // set manually in the code
|
|||
} |
|||
|
|||
// Asset is a file uploaded and attached to a release.
|
|||
type Asset struct { |
|||
ID int `json:"id"` |
|||
Name string `json:"name"` |
|||
URL string `json:"url"` |
|||
} |
|||
|
|||
const githubAPITimeout = 30 * time.Second |
|||
|
|||
// githubError is returned by the GitHub API, e.g. for rate-limiting.
|
|||
type githubError struct { |
|||
Message string |
|||
} |
|||
|
|||
//default version is not full version
|
|||
var isFullVersion = false |
|||
|
|||
var ( |
|||
updateOpt UpdateOptions |
|||
) |
|||
|
|||
type UpdateOptions struct { |
|||
dir *string |
|||
name *string |
|||
Version *string |
|||
} |
|||
|
|||
func init() { |
|||
path, _ := os.Executable() |
|||
_, name := filepath.Split(path) |
|||
updateOpt.dir = cmdUpdate.Flag.String("dir", filepath.Dir(path), "directory to save new weed.") |
|||
updateOpt.name = cmdUpdate.Flag.String("name", name, "name of new weed. On windows, name shouldn't be same to the orignial name.") |
|||
updateOpt.Version = cmdUpdate.Flag.String("version", "0", "specific version of weed you want to download. If not specified, get the latest version.") |
|||
cmdUpdate.Run = runUpdate |
|||
} |
|||
|
|||
var cmdUpdate = &Command{ |
|||
UsageLine: "update [-dir=/path/to/dir] [-name=name] [-version=x.xx]", |
|||
Short: "get latest or specific version from https://github.com/chrislusf/seaweedfs", |
|||
Long: `get latest or specific version from https://github.com/chrislusf/seaweedfs`, |
|||
} |
|||
|
|||
func runUpdate(cmd *Command, args []string) bool { |
|||
path, _ := os.Executable() |
|||
_, name := filepath.Split(path) |
|||
|
|||
if *updateOpt.dir != "" { |
|||
if err := util.TestFolderWritable(util.ResolvePath(*updateOpt.dir)); err != nil { |
|||
glog.Fatalf("Check Folder(-dir) Writable %s : %s", *updateOpt.dir, err) |
|||
return false |
|||
} |
|||
} else { |
|||
*updateOpt.dir = filepath.Dir(path) |
|||
} |
|||
|
|||
if *updateOpt.name == "" { |
|||
*updateOpt.name = name |
|||
} |
|||
|
|||
target := filepath.Join(*updateOpt.dir, *updateOpt.name) |
|||
|
|||
if runtime.GOOS == "windows" { |
|||
if target == path { |
|||
glog.Fatalf("On windows, name of the new weed shouldn't be same to the orignial name.") |
|||
return false |
|||
} |
|||
} |
|||
|
|||
glog.V(0).Infof("new weed will be saved to %s", target) |
|||
|
|||
_, err := downloadRelease(context.Background(), target, *updateOpt.Version) |
|||
if err != nil { |
|||
glog.Errorf("unable to download weed: %v", err) |
|||
return false |
|||
} |
|||
return true |
|||
} |
|||
|
|||
func downloadRelease(ctx context.Context, target string, ver string) (version string, err error) { |
|||
currentVersion := util.VERSION_NUMBER |
|||
rel, err := GitHubLatestRelease(ctx, ver, "chrislusf", "seaweedfs") |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
|
|||
if rel.Version == currentVersion { |
|||
if ver == "0" { |
|||
glog.V(0).Infof("weed is up to date") |
|||
} else { |
|||
glog.V(0).Infof("no need to download the same version of weed ") |
|||
} |
|||
return currentVersion, nil |
|||
} |
|||
|
|||
glog.V(0).Infof("download version: %s", rel.Version) |
|||
|
|||
largeDiskSuffix := "" |
|||
if util.VolumeSizeLimitGB == 8000 { |
|||
largeDiskSuffix = "_large_disk" |
|||
} |
|||
|
|||
fullSuffix := "" |
|||
if isFullVersion { |
|||
fullSuffix = "_full" |
|||
} |
|||
|
|||
ext := "tar.gz" |
|||
if runtime.GOOS == "windows" { |
|||
ext = "zip" |
|||
} |
|||
|
|||
suffix := fmt.Sprintf("%s_%s%s%s.%s", runtime.GOOS, runtime.GOARCH, fullSuffix, largeDiskSuffix, ext) |
|||
md5Filename := fmt.Sprintf("%s.md5", suffix) |
|||
_, md5Val, err := getGithubDataFile(ctx, rel.Assets, md5Filename) |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
|
|||
downloadFilename, buf, err := getGithubDataFile(ctx, rel.Assets, suffix) |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
|
|||
md5Ctx := md5.New() |
|||
md5Ctx.Write(buf) |
|||
binaryMd5 := md5Ctx.Sum(nil) |
|||
if hex.EncodeToString(binaryMd5) != string(md5Val[0:32]) { |
|||
glog.Errorf("md5:'%s' '%s'", hex.EncodeToString(binaryMd5), string(md5Val[0:32])) |
|||
err = fmt.Errorf("binary md5sum doesn't match") |
|||
return "", err |
|||
} |
|||
|
|||
err = extractToFile(buf, downloadFilename, target) |
|||
if err != nil { |
|||
return "", err |
|||
} else { |
|||
glog.V(0).Infof("successfully updated weed to version %v\n", rel.Version) |
|||
} |
|||
|
|||
return rel.Version, nil |
|||
} |
|||
|
|||
// GitHubLatestRelease uses the GitHub API to get information about the specific
|
|||
// release of a repository.
|
|||
func GitHubLatestRelease(ctx context.Context, ver string, owner, repo string) (Release, error) { |
|||
ctx, cancel := context.WithTimeout(ctx, githubAPITimeout) |
|||
defer cancel() |
|||
|
|||
url := fmt.Sprintf("https://api.github.com/repos/%s/%s/releases", owner, repo) |
|||
req, err := http.NewRequest(http.MethodGet, url, nil) |
|||
if err != nil { |
|||
return Release{}, err |
|||
} |
|||
|
|||
// pin API version 3
|
|||
req.Header.Set("Accept", "application/vnd.github.v3+json") |
|||
|
|||
res, err := ctxhttp.Do(ctx, http.DefaultClient, req) |
|||
if err != nil { |
|||
return Release{}, err |
|||
} |
|||
|
|||
if res.StatusCode != http.StatusOK { |
|||
content := res.Header.Get("Content-Type") |
|||
if strings.Contains(content, "application/json") { |
|||
// try to decode error message
|
|||
var msg githubError |
|||
jerr := json.NewDecoder(res.Body).Decode(&msg) |
|||
if jerr == nil { |
|||
return Release{}, fmt.Errorf("unexpected status %v (%v) returned, message:\n %v", res.StatusCode, res.Status, msg.Message) |
|||
} |
|||
} |
|||
|
|||
_ = res.Body.Close() |
|||
return Release{}, fmt.Errorf("unexpected status %v (%v) returned", res.StatusCode, res.Status) |
|||
} |
|||
|
|||
buf, err := ioutil.ReadAll(res.Body) |
|||
if err != nil { |
|||
_ = res.Body.Close() |
|||
return Release{}, err |
|||
} |
|||
|
|||
err = res.Body.Close() |
|||
if err != nil { |
|||
return Release{}, err |
|||
} |
|||
|
|||
var release Release |
|||
var releaseList []Release |
|||
err = json.Unmarshal(buf, &releaseList) |
|||
if err != nil { |
|||
return Release{}, err |
|||
} |
|||
if ver == "0" { |
|||
release = releaseList[0] |
|||
glog.V(0).Infof("latest version is %v\n", release.TagName) |
|||
} else { |
|||
for _, r := range releaseList { |
|||
if r.TagName == ver { |
|||
release = r |
|||
break |
|||
} |
|||
} |
|||
} |
|||
|
|||
if release.TagName == "" { |
|||
return Release{}, fmt.Errorf("can not find the specific version") |
|||
} |
|||
|
|||
release.Version = release.TagName |
|||
return release, nil |
|||
} |
|||
|
|||
func getGithubData(ctx context.Context, url string) ([]byte, error) { |
|||
req, err := http.NewRequest(http.MethodGet, url, nil) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
// request binary data
|
|||
req.Header.Set("Accept", "application/octet-stream") |
|||
|
|||
res, err := ctxhttp.Do(ctx, http.DefaultClient, req) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
if res.StatusCode != http.StatusOK { |
|||
return nil, fmt.Errorf("unexpected status %v (%v) returned", res.StatusCode, res.Status) |
|||
} |
|||
|
|||
buf, err := ioutil.ReadAll(res.Body) |
|||
if err != nil { |
|||
_ = res.Body.Close() |
|||
return nil, err |
|||
} |
|||
|
|||
err = res.Body.Close() |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return buf, nil |
|||
} |
|||
|
|||
func getGithubDataFile(ctx context.Context, assets []Asset, suffix string) (filename string, data []byte, err error) { |
|||
var url string |
|||
for _, a := range assets { |
|||
if strings.HasSuffix(a.Name, suffix) { |
|||
url = a.URL |
|||
filename = a.Name |
|||
break |
|||
} |
|||
} |
|||
|
|||
if url == "" { |
|||
return "", nil, fmt.Errorf("unable to find file with suffix %v", suffix) |
|||
} |
|||
|
|||
glog.V(0).Infof("download %v\n", filename) |
|||
data, err = getGithubData(ctx, url) |
|||
if err != nil { |
|||
return "", nil, err |
|||
} |
|||
|
|||
return filename, data, nil |
|||
} |
|||
|
|||
func extractToFile(buf []byte, filename, target string) error { |
|||
var rd io.Reader = bytes.NewReader(buf) |
|||
|
|||
switch filepath.Ext(filename) { |
|||
case ".gz": |
|||
gr, err := gzip.NewReader(rd) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
defer gr.Close() |
|||
trd := tar.NewReader(gr) |
|||
hdr, terr := trd.Next() |
|||
if terr != nil { |
|||
glog.Errorf("uncompress file(%s) failed:%s", hdr.Name, terr) |
|||
return terr |
|||
} |
|||
rd = trd |
|||
case ".zip": |
|||
zrd, err := zip.NewReader(bytes.NewReader(buf), int64(len(buf))) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
if len(zrd.File) != 1 { |
|||
return fmt.Errorf("ZIP archive contains more than one file") |
|||
} |
|||
|
|||
file, err := zrd.File[0].Open() |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
defer func() { |
|||
_ = file.Close() |
|||
}() |
|||
|
|||
rd = file |
|||
} |
|||
|
|||
// Write everything to a temp file
|
|||
dir := filepath.Dir(target) |
|||
new, err := ioutil.TempFile(dir, "weed") |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
n, err := io.Copy(new, rd) |
|||
if err != nil { |
|||
_ = new.Close() |
|||
_ = os.Remove(new.Name()) |
|||
return err |
|||
} |
|||
if err = new.Sync(); err != nil { |
|||
return err |
|||
} |
|||
if err = new.Close(); err != nil { |
|||
return err |
|||
} |
|||
|
|||
mode := os.FileMode(0755) |
|||
// attempt to find the original mode
|
|||
if fi, err := os.Lstat(target); err == nil { |
|||
mode = fi.Mode() |
|||
} |
|||
|
|||
// Rename the temp file to the final location atomically.
|
|||
if err := os.Rename(new.Name(), target); err != nil { |
|||
return err |
|||
} |
|||
|
|||
glog.V(0).Infof("saved %d bytes in %v\n", n, target) |
|||
return os.Chmod(target, mode) |
|||
} |
|||
@ -0,0 +1,9 @@ |
|||
//go:build elastic && ydb && gocdk && hdfs
|
|||
// +build elastic,ydb,gocdk,hdfs
|
|||
|
|||
package command |
|||
|
|||
//set true if gtags are set
|
|||
func init() { |
|||
isFullVersion = true |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue