Template
1
0
Fork 0
mirror of https://codeberg.org/forgejo/forgejo synced 2024-12-05 02:54:46 +01:00
forgejo/vendor/github.com/pingcap/tidb/mysql/bit.go

107 lines
2.8 KiB
Go
Raw Normal View History

// Copyright 2015 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
package mysql
import (
"fmt"
"strconv"
"strings"
"github.com/juju/errors"
)
// Bit is for mysql bit type.
type Bit struct {
// Value holds the value for bit type.
Value uint64
// Width is the display with for bit value.
// e.g, with is 8, 0 is for 0b00000000.
Width int
}
// String implements fmt.Stringer interface.
func (b Bit) String() string {
format := fmt.Sprintf("0b%%0%db", b.Width)
return fmt.Sprintf(format, b.Value)
}
// ToNumber changes bit type to float64 for numeric operation.
// MySQL treats bit as double type.
func (b Bit) ToNumber() float64 {
return float64(b.Value)
}
// ToString returns the binary string for bit type.
func (b Bit) ToString() string {
byteSize := (b.Width + 7) / 8
buf := make([]byte, byteSize)
for i := byteSize - 1; i >= 0; i-- {
buf[byteSize-i-1] = byte(b.Value >> uint(i*8))
}
return string(buf)
}
// Min and Max bit width.
const (
MinBitWidth = 1
MaxBitWidth = 64
// UnspecifiedBitWidth is the unspecified with if you want to calculate bit width dynamically.
UnspecifiedBitWidth = -1
)
// ParseBit parses bit string.
// The string format can be b'val', B'val' or 0bval, val must be 0 or 1.
// Width is the display width for bit representation. -1 means calculating
// width dynamically, using following algorithm: (len("011101") + 7) & ^7,
// e.g, if bit string is 0b01, the above will return 8 for its bit width.
func ParseBit(s string, width int) (Bit, error) {
if len(s) == 0 {
return Bit{}, errors.Errorf("invalid empty string for parsing bit type")
}
if s[0] == 'b' || s[0] == 'B' {
// format is b'val' or B'val'
s = strings.Trim(s[1:], "'")
} else if strings.HasPrefix(s, "0b") {
s = s[2:]
} else {
// here means format is not b'val', B'val' or 0bval.
return Bit{}, errors.Errorf("invalid bit type format %s", s)
}
if width == UnspecifiedBitWidth {
width = (len(s) + 7) & ^7
}
if width == 0 {
width = MinBitWidth
}
if width < MinBitWidth || width > MaxBitWidth {
return Bit{}, errors.Errorf("invalid display width for bit type, must in [1, 64], but %d", width)
}
n, err := strconv.ParseUint(s, 2, 64)
if err != nil {
return Bit{}, errors.Trace(err)
}
if n > (uint64(1)<<uint64(width))-1 {
return Bit{}, errors.Errorf("bit %s is too long for width %d", s, width)
}
return Bit{Value: n, Width: width}, nil
}