gistfile1.go

type ObjectHandler struct {
    ...
    diskLimit        int64
    diskInUse        map[string]*int64
    diskInUseLock    *sync.Mutex
}


func (server *ObjectHandler) AcquireDisk(disk string) bool {
    if val, ok := server.diskInUse[disk]; !ok {
        server.diskInUseLock.Lock()
        if _, ok := server.diskInUse[disk]; !ok {
            server.diskInUse[disk] = new(int64)
        }
        server.diskInUseLock.Unlock()
    } else if *val > server.diskLimit {
        return false
    }
    atomic.AddInt64(server.diskInUse[disk], 1)
    return true
}

func (server *ObjectHandler) ReleaseDisk(disk string) {
    atomic.AddInt64(server.diskInUse[disk], -1)
}