tidb parser 源码

2022-09-19
浏览 (346)

tidb parser 代码

文件路径：/util/parser/parser.go

// Copyright 2019 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,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package parser

import (
	"strconv"
	"unicode"

	"github.com/pingcap/errors"
)

var (
	// ErrPatternNotMatch represents an error that patterns doesn't match.
	ErrPatternNotMatch = errors.New("Pattern not match")
)

// Match matches the `pat` at least `times`, and returns the match, the rest and the error
func Match(buf string, pat func(byte) bool, times int) (match string, rest string, err error) {
	var i int
	for i < len(buf) && pat(buf[i]) {
		i++
	}
	if i < times {
		return "", buf, ErrPatternNotMatch
	}
	return buf[:i], buf[i:], nil
}

// MatchOne matches only one time with pat
func MatchOne(buf string, pat func(byte) bool) (string, error) {
	if len(buf) == 0 || !pat(buf[0]) {
		return buf, ErrPatternNotMatch
	}
	return buf[1:], nil
}

// AnyPunct matches an arbitrary punctuation
func AnyPunct(buf string) (string, error) {
	return MatchOne(buf, func(b byte) bool {
		return unicode.IsPunct(rune(b))
	})
}

// AnyChar matches an arbitrary character
func AnyChar(buf string) (string, error) {
	return MatchOne(buf, func(byte) bool {
		return true
	})
}

// Char matches a character: c
func Char(buf string, c byte) (string, error) {
	return MatchOne(buf, func(x byte) bool {
		return x == c
	})
}

// Space matches at least `times` spaces
func Space(buf string, times int) (string, error) {
	_, rest, err := Match(buf, func(c byte) bool {
		return unicode.IsSpace(rune(c))
	}, times)
	return rest, err
}

// Space0 matches at least 0 space.
func Space0(buf string) string {
	rest, err := Space(buf, 0)
	if err != nil {
		panic(err)
	}
	return rest
}

// Digit matches at least `times` digits
func Digit(buf string, times int) (match string, rest string, err error) {
	return Match(buf, func(c byte) bool {
		return unicode.IsDigit(rune(c))
	}, times)
}

// Number matches a series of digits and convert it to an int
func Number(str string) (int, string, error) {
	digits, rest, err := Digit(str, 1)
	if err != nil {
		return 0, str, err
	}
	num, err := strconv.Atoi(digits)
	return num, rest, err
}

相关信息

tidb 源码目录

相关文章

tidb ast 源码

0 赞

所属分类： 大数据
本文标签： golang 大数据
版权声明： 原创文章如转载，请注明本文链接: https://m.seaxiang.com/blog/bb9afa5f4fb64554a046b389ec63b6c8

热门推荐

1、直接访问google.com
2、 - 优质文章
3、 gate.io
4、 harmony 鸿蒙hdc使用指导
5、 harmony 鸿蒙初识ArkTS语言
6、 harmony 鸿蒙ArkUI组件（ArkTS）开发常见问题
7、 flink kafka connector scan.startup.mode 的几个选项
8、 golang
9、 openharmony
10、 Vue中input框自动聚焦

Loading...