seaweedfs/weed/storage/volume_super_block.go

package storage

import (
	"fmt"
	"os"

	"github.com/joeslay/seaweedfs/weed/storage/memory_map"

	"github.com/golang/protobuf/proto"
	"github.com/joeslay/seaweedfs/weed/glog"
	"github.com/joeslay/seaweedfs/weed/pb/master_pb"
	"github.com/joeslay/seaweedfs/weed/storage/needle"
	"github.com/joeslay/seaweedfs/weed/util"
)

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
	CompactionRevision 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.CompactionRevision)

	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 {

	mMap, exists := memory_map.FileMemoryMap[v.dataFile.Name()]
	if exists {
		if mMap.End_of_file == -1 {
			v.SuperBlock.version = needle.CurrentVersion
			mMap.WriteMemory(0, uint64(len(v.SuperBlock.Bytes())), v.SuperBlock.Bytes())
		}
		return nil
	} else {
		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) {

	header := make([]byte, _SuperBlockSize)
	mMap, exists := memory_map.FileMemoryMap[dataFile.Name()]
	if exists {
		mem_buffer, e := mMap.ReadMemory(0, _SuperBlockSize)
		if err != nil {
			err = fmt.Errorf("cannot read volume %s super block: %v", dataFile.Name(), e)
			return
		}
		copy(header, mem_buffer.Buffer)
		mem_buffer.ReleaseMemory()
	} else {
		if _, err = dataFile.Seek(0, 0); err != nil {
			err = fmt.Errorf("cannot seek to the beginning of %s: %v", dataFile.Name(), err)
			return
		}
		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.CompactionRevision = 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
}