Bolt源码解析（三）：kv的插入、查询、删除操作

1、Put

前面我们讲完了创建bucket，创建完bucket就需要kv的读写，我们分析写流程，写流程的函数入口是Bucket.Put()函数

func (b *Bucket) Put(key []byte, value []byte) error {
    if b.tx.db == nil {
        return ErrTxClosed
    } else if !b.Writable() {
        return ErrTxNotWritable
    } else if len(key) == 0 {
        return ErrKeyRequired
    } else if len(key) > MaxKeySize {
        return ErrKeyTooLarge
    } else if int64(len(value)) > MaxValueSize {
        return ErrValueTooLarge
    }

    // Move cursor to correct position.
    c := b.Cursor()
    k, _, flags := c.seek(key)//seek方法前面已经介绍

    // 如果key已存在，返回
    if bytes.Equal(key, k) && (flags&bucketLeafFlag) != 0 {
        return ErrIncompatibleValue
    }

    // Insert into node.
key = cloneBytes(key)
//Cursor.node()前面已经介绍
    c.node().put(key, key, value, 0, 0)

    return nil
}

Put方法比较简单，也比较容易理解。这里额外多介绍点，数据写入page，该page每次都是新申请的，数据都向新的page上写，这就是前面介绍的COW技术，简单示意图如下：

                                   

 

2、Get

func (b *Bucket) Get(key []byte) []byte {
    k, v, flags := b.Cursor().seek(key)

    // Return nil if this is a bucket.
    if (flags & bucketLeafFlag) != 0 {
        return nil
    }

    // If our target node isn't the same key as what's passed in then return nil.
    if !bytes.Equal(key, k) {
        return nil
    }
    return v
}

查询中的函数都是介绍过，这里不在做介绍 

3、Delete

func (b *Bucket) Delete(key []byte) error {
    if b.tx.db == nil {
        return ErrTxClosed
    } else if !b.Writable() {
        return ErrTxNotWritable
    }

    // Move cursor to correct position.
    c := b.Cursor()
    _, _, flags := c.seek(key)

    // Return an error if there is already existing bucket value.
    if (flags & bucketLeafFlag) != 0 {
        return ErrIncompatibleValue
    }

    // Delete the node if we have a matching key.
    c.node().del(key)

    return nil
}

func (n *node) del(key []byte) {
    // Find index of key.
    index := sort.Search(len(n.inodes), func(i int) bool { return bytes.Compare(n.inodes[i].key, key) != -1 })

    // Exit if the key isn't found.
    if index >= len(n.inodes) || !bytes.Equal(n.inodes[index].key, key) {
        return
    }

    // Delete inode from the node.
    n.inodes = append(n.inodes[:index], n.inodes[index+1:]...)

    // Mark the node as needing rebalancing.
    n.unbalanced = true
}

Del的过程和put相反，del是从node.inodes中把数据删掉。