package storage import ( "fmt" "os" "github.com/chrislusf/seaweedfs/weed/glog" "github.com/chrislusf/seaweedfs/weed/pb/master_pb" "github.com/chrislusf/seaweedfs/weed/storage/needle" "github.com/chrislusf/seaweedfs/weed/util" "github.com/golang/protobuf/proto" ) const ( _SuperBlockSize = 8 ) /* * Super block currently has 8 bytes allocated for each volume. * Byte 0: version, 1 or 2 * Byte 1: Replica Placement strategy, 000, 001, 002, 010, etc * Byte 2 and byte 3: Time to live. See TTL for definition * Byte 4 and byte 5: The number of times the volume has been compacted. * Rest bytes: Reserved */ type SuperBlock struct { version needle.Version ReplicaPlacement *ReplicaPlacement Ttl *needle.TTL CompactRevision uint16 Extra *master_pb.SuperBlockExtra extraSize uint16 } func (s *SuperBlock) BlockSize() int { switch s.version { case needle.Version2, needle.Version3: return _SuperBlockSize + int(s.extraSize) } return _SuperBlockSize } func (s *SuperBlock) Version() needle.Version { return s.version } func (s *SuperBlock) Bytes() []byte { header := make([]byte, _SuperBlockSize) header[0] = byte(s.version) header[1] = s.ReplicaPlacement.Byte() s.Ttl.ToBytes(header[2:4]) util.Uint16toBytes(header[4:6], s.CompactRevision) if s.Extra != nil { extraData, err := proto.Marshal(s.Extra) if err != nil { glog.Fatalf("cannot marshal super block extra %+v: %v", s.Extra, err) } extraSize := len(extraData) if extraSize > 256*256-2 { // reserve a couple of bits for future extension glog.Fatalf("super block extra size is %d bigger than %d", extraSize, 256*256-2) } s.extraSize = uint16(extraSize) util.Uint16toBytes(header[6:8], s.extraSize) header = append(header, extraData...) } return header } func (v *Volume) maybeWriteSuperBlock() error { stat, e := v.dataFile.Stat() if e != nil { glog.V(0).Infof("failed to stat datafile %s: %v", v.dataFile.Name(), e) return e } if stat.Size() == 0 { v.SuperBlock.version = needle.CurrentVersion _, e = v.dataFile.Write(v.SuperBlock.Bytes()) if e != nil && os.IsPermission(e) { //read-only, but zero length - recreate it! if v.dataFile, e = os.Create(v.dataFile.Name()); e == nil { if _, e = v.dataFile.Write(v.SuperBlock.Bytes()); e == nil { v.readOnly = false } } } } return e } func (v *Volume) readSuperBlock() (err error) { v.SuperBlock, err = ReadSuperBlock(v.dataFile) return err } // ReadSuperBlock reads from data file and load it into volume's super block func ReadSuperBlock(dataFile *os.File) (superBlock SuperBlock, err error) { if _, err = dataFile.Seek(0, 0); err != nil { err = fmt.Errorf("cannot seek to the beginning of %s: %v", dataFile.Name(), err) return } header := make([]byte, _SuperBlockSize) if _, e := dataFile.Read(header); e != nil { err = fmt.Errorf("cannot read volume %s super block: %v", dataFile.Name(), e) return } superBlock.version = needle.Version(header[0]) if superBlock.ReplicaPlacement, err = NewReplicaPlacementFromByte(header[1]); err != nil { err = fmt.Errorf("cannot read replica type: %s", err.Error()) return } superBlock.Ttl = needle.LoadTTLFromBytes(header[2:4]) superBlock.CompactRevision = util.BytesToUint16(header[4:6]) superBlock.extraSize = util.BytesToUint16(header[6:8]) if superBlock.extraSize > 0 { // read more extraData := make([]byte, int(superBlock.extraSize)) superBlock.Extra = &master_pb.SuperBlockExtra{} err = proto.Unmarshal(extraData, superBlock.Extra) if err != nil { err = fmt.Errorf("cannot read volume %s super block extra: %v", dataFile.Name(), err) return } } return }