state.go

// Copyright © 2018 Chad Netzer <chad.netzer@gmail.com>
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package hardlinkable

import (
	"github.com/chadnetzer/hardlinkable/internal/inode"
	P "github.com/chadnetzer/hardlinkable/internal/pathpool"
)

const maxCmpBufSize = 8 * minCmpBufSize // Power of two multiplier
const minCmpBufSize = 4096
const digestBufSize = 4096

type status struct {
	Options   *Options
	Results   *Results
	Progress  progress
	cmpBuf1   []byte
	cmpBuf2   []byte
	digestBuf []byte
	pool      *P.StringPool
}

type linkableState struct {
	status
	fsDevs map[uint64]fsDev
}

func newLinkableState(opts *Options) *linkableState {
	return &linkableState{
		status: status{
			Options:   opts,
			Results:   newResults(opts),
			cmpBuf1:   make([]byte, minCmpBufSize, maxCmpBufSize),
			cmpBuf2:   make([]byte, minCmpBufSize, maxCmpBufSize),
			digestBuf: make([]byte, digestBufSize),
			pool:      P.NewPool(),
		},
		fsDevs: make(map[uint64]fsDev),
	}
}

func (ls *linkableState) dev(di inode.DevStatInfo, pathname string) fsDev {
	if fsdev, ok := ls.fsDevs[di.Dev]; ok {
		return fsdev
	}
	fsdev := newFSDev(ls.status, di.Dev, inode.MaxNlinkVal(pathname))
	ls.fsDevs[di.Dev] = fsdev
	return fsdev
}