2021-08-27 06:18:34 +08:00
|
|
|
package volume_info
|
2019-12-29 04:28:58 +08:00
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"fmt"
|
2021-10-14 12:27:58 +08:00
|
|
|
"os"
|
2019-12-29 04:28:58 +08:00
|
|
|
|
|
|
|
"github.com/golang/protobuf/jsonpb"
|
|
|
|
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/pb/volume_server_pb"
|
2021-10-14 12:27:58 +08:00
|
|
|
_ "github.com/chrislusf/seaweedfs/weed/storage/backend/s3_backend"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
2019-12-29 04:28:58 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
// MaybeLoadVolumeInfo load the file data as *volume_server_pb.VolumeInfo, the returned volumeInfo will not be nil
|
2021-03-10 04:09:32 +08:00
|
|
|
func MaybeLoadVolumeInfo(fileName string) (volumeInfo *volume_server_pb.VolumeInfo, hasRemoteFile bool, hasVolumeInfoFile bool, err error) {
|
2019-12-29 04:28:58 +08:00
|
|
|
|
2021-03-10 04:09:32 +08:00
|
|
|
volumeInfo = &volume_server_pb.VolumeInfo{}
|
2019-12-29 04:28:58 +08:00
|
|
|
|
|
|
|
glog.V(1).Infof("maybeLoadVolumeInfo checks %s", fileName)
|
|
|
|
if exists, canRead, _, _, _ := util.CheckFile(fileName); !exists || !canRead {
|
|
|
|
if !exists {
|
2021-03-10 04:09:32 +08:00
|
|
|
return
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
2021-03-10 04:09:32 +08:00
|
|
|
hasVolumeInfoFile = true
|
2019-12-29 04:28:58 +08:00
|
|
|
if !canRead {
|
|
|
|
glog.Warningf("can not read %s", fileName)
|
2021-03-10 04:09:32 +08:00
|
|
|
err = fmt.Errorf("can not read %s", fileName)
|
|
|
|
return
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
2021-03-10 04:09:32 +08:00
|
|
|
return
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
2021-03-10 04:09:32 +08:00
|
|
|
hasVolumeInfoFile = true
|
|
|
|
|
2019-12-29 04:28:58 +08:00
|
|
|
glog.V(1).Infof("maybeLoadVolumeInfo reads %s", fileName)
|
2021-10-14 12:27:58 +08:00
|
|
|
tierData, readErr := os.ReadFile(fileName)
|
2019-12-29 04:28:58 +08:00
|
|
|
if readErr != nil {
|
|
|
|
glog.Warningf("fail to read %s : %v", fileName, readErr)
|
2021-03-10 04:09:32 +08:00
|
|
|
err = fmt.Errorf("fail to read %s : %v", fileName, readErr)
|
|
|
|
return
|
|
|
|
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
glog.V(1).Infof("maybeLoadVolumeInfo Unmarshal volume info %v", fileName)
|
2021-03-10 04:09:32 +08:00
|
|
|
if err = jsonpb.Unmarshal(bytes.NewReader(tierData), volumeInfo); err != nil {
|
2019-12-29 04:28:58 +08:00
|
|
|
glog.Warningf("unmarshal error: %v", err)
|
2021-03-10 04:09:32 +08:00
|
|
|
err = fmt.Errorf("unmarshal error: %v", err)
|
|
|
|
return
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if len(volumeInfo.GetFiles()) == 0 {
|
2021-03-10 04:09:32 +08:00
|
|
|
return
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
2021-03-10 04:09:32 +08:00
|
|
|
hasRemoteFile = true
|
|
|
|
|
|
|
|
return
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func SaveVolumeInfo(fileName string, volumeInfo *volume_server_pb.VolumeInfo) error {
|
|
|
|
|
2019-12-29 13:40:12 +08:00
|
|
|
if exists, _, canWrite, _, _ := util.CheckFile(fileName); exists && !canWrite {
|
|
|
|
return fmt.Errorf("%s not writable", fileName)
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
m := jsonpb.Marshaler{
|
|
|
|
EmitDefaults: true,
|
|
|
|
Indent: " ",
|
|
|
|
}
|
|
|
|
|
|
|
|
text, marshalErr := m.MarshalToString(volumeInfo)
|
|
|
|
if marshalErr != nil {
|
|
|
|
return fmt.Errorf("marshal to %s: %v", fileName, marshalErr)
|
|
|
|
}
|
|
|
|
|
2022-02-05 03:14:04 +08:00
|
|
|
writeErr := util.WriteFile(fileName, []byte(text), 0755)
|
2019-12-29 04:28:58 +08:00
|
|
|
if writeErr != nil {
|
2019-12-29 13:40:12 +08:00
|
|
|
return fmt.Errorf("fail to write %s : %v", fileName, writeErr)
|
2019-12-29 04:28:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|