加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
dialect.go 41.78 KB
一键复制 编辑 原始数据 按行查看 历史
springrain 提交于 2024-04-18 22:07 . 优化正则表达式声明
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 zorm
import (
"context"
"crypto/rand"
"database/sql"
"errors"
"fmt"
"math/big"
"reflect"
"regexp"
"strconv"
"strings"
"time"
)
// wrapPageSQL 包装分页的SQL语句
// wrapPageSQL SQL statement for wrapping paging
var wrapPageSQL = func(ctx context.Context, config *DataSourceConfig, sqlstr *string, page *Page) error {
if page.PageNo < 1 { // 默认第一页
page.PageNo = 1
}
var sqlbuilder strings.Builder
sqlbuilder.Grow(stringBuilderGrowLen)
sqlbuilder.WriteString(*sqlstr)
switch config.Dialect {
case "mysql", "sqlite", "dm", "gbase", "clickhouse", "tdengine", "db2": // MySQL,sqlite3,dm,南通,clickhouse,TDengine,db2 7.2+
sqlbuilder.WriteString(" LIMIT ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize * (page.PageNo - 1)))
sqlbuilder.WriteByte(',')
sqlbuilder.WriteString(strconv.Itoa(page.PageSize))
case "postgresql", "kingbase", "shentong": // postgresql,kingbase,神通数据库
sqlbuilder.WriteString(" LIMIT ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize))
sqlbuilder.WriteString(" OFFSET ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize * (page.PageNo - 1)))
case "mssql": // sqlserver 2012+
locOrderBy := findOrderByIndex(sqlstr)
if len(locOrderBy) < 1 { // 如果没有 order by,增加默认的排序
sqlbuilder.WriteString(" ORDER BY (SELECT NULL) ")
}
sqlbuilder.WriteString(" OFFSET ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize * (page.PageNo - 1)))
sqlbuilder.WriteString(" ROWS FETCH NEXT ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize))
sqlbuilder.WriteString(" ROWS ONLY ")
case "oracle": // oracle 12c+
locOrderBy := findOrderByIndex(sqlstr)
if len(locOrderBy) < 1 { // 如果没有 order by,增加默认的排序
sqlbuilder.WriteString(" ORDER BY NULL ")
}
sqlbuilder.WriteString(" OFFSET ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize * (page.PageNo - 1)))
sqlbuilder.WriteString(" ROWS FETCH NEXT ")
sqlbuilder.WriteString(strconv.Itoa(page.PageSize))
sqlbuilder.WriteString(" ROWS ONLY ")
default:
return errors.New("->wrapPageSQL-->不支持的数据库类型:" + config.Dialect)
}
*sqlstr = sqlbuilder.String()
// return reBuildSQL(dialect, sqlstr)
return nil
}
// wrapInsertSQL 包装保存Struct语句.返回语句,是否自增,错误信息
// 数组传递,如果外部方法有调用append的逻辑,append会破坏指针引用,所以传递指针
// wrapInsertSQL Pack and save 'Struct' statement. Return SQL statement, whether it is incremented, error message
// Array transfer, if the external method has logic to call append, append will destroy the pointer reference, so the pointer is passed
var wrapInsertSQL = func(ctx context.Context, config *DataSourceConfig, typeOf *reflect.Type, entity IEntityStruct, columns *[]reflect.StructField, values *[]interface{}) (*string, int, string, error) {
sqlstr := ""
inserColumnName, valuesql, autoIncrement, pktype, err := wrapInsertValueSQL(ctx, typeOf, entity, columns, values)
if err != nil {
return &sqlstr, autoIncrement, pktype, err
}
var sqlBuilder strings.Builder
// sqlBuilder.Grow(len(entity.GetTableName()) + len(inserColumnName) + len(entity.GetTableName()) + len(valuesql) + 19)
sqlBuilder.Grow(stringBuilderGrowLen)
// sqlstr := "INSERT INTO " + insersql + " VALUES" + valuesql
sqlBuilder.WriteString("INSERT INTO ")
sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteString(*inserColumnName)
sqlBuilder.WriteString(" VALUES")
sqlBuilder.WriteString(*valuesql)
sqlstr = sqlBuilder.String()
return &sqlstr, autoIncrement, pktype, err
}
// wrapInsertValueSQL 包装保存Struct语句.返回语句,没有rebuild,返回原始的InsertSQL,ValueSQL,是否自增,主键类型,错误信息
// 数组传递,如果外部方法有调用append的逻辑,传递指针,因为append会破坏指针引用
// Pack and save Struct statement. Return SQL statement, no rebuild, return original SQL, whether it is self-increment, error message
// Array transfer, if the external method has logic to call append, append will destroy the pointer reference, so the pointer is passed
func wrapInsertValueSQL(ctx context.Context, typeOf *reflect.Type, entity IEntityStruct, columns *[]reflect.StructField, values *[]interface{}) (*string, *string, int, string, error) {
var inserColumnName, valuesql string
// 自增类型 0(不自增),1(普通自增),2(序列自增)
// Self-increment type: 0(Not increase),1(Ordinary increment),2(Sequence increment)
autoIncrement := 0
// 主键类型
// Primary key type
pktype := ""
// SQL语句的构造器
// SQL statement constructor
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
// sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteByte('(')
// SQL语句中,VALUES(?,?,...)语句的构造器
// In the SQL statement, the constructor of the VALUES(?,?,...) statement
var valueSQLBuilder strings.Builder
valueSQLBuilder.Grow(stringBuilderGrowLen)
valueSQLBuilder.WriteString(" (")
// 主键的名称
// The name of the primary key.
pkFieldName, err := entityPKFieldName(entity, typeOf)
if err != nil {
return &inserColumnName, &valuesql, autoIncrement, pktype, err
}
sequence := entity.GetPkSequence()
if sequence != "" {
// 序列自增 Sequence increment
autoIncrement = 2
}
// 获取所有字段和tag对照的Map
tagMap, err := getStructFieldTagMap(typeOf)
if err != nil {
return &inserColumnName, &valuesql, autoIncrement, pktype, err
}
for i := 0; i < len(*columns); i++ {
field := (*columns)[i]
if field.Name == pkFieldName { // 如果是主键 | If it is the primary key
// 获取主键类型 | Get the primary key type.
pkKind := field.Type.Kind()
switch pkKind {
case reflect.String:
pktype = "string"
case reflect.Int, reflect.Int32, reflect.Int16, reflect.Int8:
pktype = "int"
case reflect.Int64:
pktype = "int64"
default:
return &inserColumnName, &valuesql, autoIncrement, pktype, errors.New("->wrapInsertValueSQL不支持的主键类型")
}
if autoIncrement == 2 { // 如果是序列自增 | If it is a sequence increment
// 去掉这一列,后续不再处理
// Remove this column and will not process it later.
*columns = append((*columns)[:i], (*columns)[i+1:]...)
*values = append((*values)[:i], (*values)[i+1:]...)
i = i - 1
if i > 0 { // i+1<len(*columns)会有风险:id是最后的字段,而且还是自增,被忽略了,但是前面的已经处理,是 逗号, 结尾的,就会bug,实际概率极低
sqlBuilder.WriteByte(',')
valueSQLBuilder.WriteByte(',')
}
colName := getFieldTagName(&field, tagMap)
sqlBuilder.WriteString(colName)
valueSQLBuilder.WriteString(sequence)
continue
}
// 主键的值
// The value of the primary key
pkValue := (*values)[i]
valueIsZero := reflect.ValueOf(pkValue).IsZero()
if valueIsZero && (pktype == "string") { // 主键是字符串类型,并且值为"",赋值id
// 生成主键字符串
// Generate primary key string
id := FuncGenerateStringID(ctx)
(*values)[i] = id
// 给对象主键赋值
// Assign a value to the primary key of the object
v := reflect.ValueOf(entity).Elem()
v.FieldByName(field.Name).Set(reflect.ValueOf(id))
} else if valueIsZero && (pktype == "int" || pktype == "int64") {
// 标记是自增主键
// Mark is auto-incrementing primary key
autoIncrement = 1
// 去掉这一列,后续不再处理
// Remove this column and will not process it later.
*columns = append((*columns)[:i], (*columns)[i+1:]...)
*values = append((*values)[:i], (*values)[i+1:]...)
i = i - 1
continue
}
}
if i > 0 { // i+1<len(*columns)会有风险:id是最后的字段,而且还是自增,被忽略了,但是前面的已经处理,是 逗号, 结尾的,就会bug,实际概率极低
sqlBuilder.WriteByte(',')
valueSQLBuilder.WriteByte(',')
}
colName := getFieldTagName(&field, tagMap)
sqlBuilder.WriteString(colName)
valueSQLBuilder.WriteByte('?')
}
sqlBuilder.WriteByte(')')
valueSQLBuilder.WriteByte(')')
inserColumnName = sqlBuilder.String()
valuesql = valueSQLBuilder.String()
return &inserColumnName, &valuesql, autoIncrement, pktype, nil
}
// wrapInsertSliceSQL 包装批量保存StructSlice语句.返回语句,是否自增,错误信息
// 数组传递,如果外部方法有调用append的逻辑,append会破坏指针引用,所以传递指针
// wrapInsertSliceSQL Package and save Struct Slice statements in batches. Return SQL statement, whether it is incremented, error message
// Array transfer, if the external method has logic to call append, append will destroy the pointer reference, so the pointer is passed
var wrapInsertSliceSQL = func(ctx context.Context, config *DataSourceConfig, typeOf *reflect.Type, entityStructSlice []IEntityStruct, columns *[]reflect.StructField, values *[]interface{}) (*string, int, error) {
sliceLen := len(entityStructSlice)
sqlstr := ""
if entityStructSlice == nil || sliceLen < 1 {
return &sqlstr, 0, errors.New("->wrapInsertSliceSQL对象数组不能为空")
}
// 第一个对象,获取第一个Struct对象,用于获取数据库字段,也获取了值
// The first object, get the first Struct object, used to get the database field, and also get the value
entity := entityStructSlice[0]
// 先生成一条语句
// Generate a statement first
inserColumnName, valuesql, autoIncrement, _, firstErr := wrapInsertValueSQL(ctx, typeOf, entity, columns, values)
if firstErr != nil {
return &sqlstr, autoIncrement, firstErr
}
var sqlBuilder strings.Builder
// sqlBuilder.Grow(len(entity.GetTableName()) + len(inserColumnName) + len(entity.GetTableName()) + len(valuesql) + 19)
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString("INSERT INTO ")
sqlBuilder.WriteString(entity.GetTableName())
// sqlstr := "INSERT INTO "
if config.Dialect == "tdengine" && !config.TDengineInsertsColumnName { // 如果是tdengine,拼接类似 INSERT INTO table1 values('2','3') table2 values('4','5'),目前要求字段和类型必须一致,如果不一致,改动略多
} else {
// sqlstr = sqlstr + insertsql + " VALUES" + valuesql
sqlBuilder.WriteString(*inserColumnName)
}
sqlBuilder.WriteString(" VALUES")
sqlBuilder.WriteString(*valuesql)
sqlstr = sqlBuilder.String()
// 如果只有一个Struct对象
// If there is only one Struct object
if sliceLen == 1 {
return &sqlstr, autoIncrement, firstErr
}
// 主键的名称
// The name of the primary key
pkFieldName, e := entityPKFieldName(entity, typeOf)
if e != nil {
return &sqlstr, autoIncrement, e
}
/*
defaultValueMap := entity.GetDefaultValue()
ctxValueMap := ctx.Value(contextDefaultValueKey)
if ctxValueMap != nil {
defaultValueMap = ctxValueMap.(map[string]interface{})
} else {
defaultValueMap = entity.GetDefaultValue()
}
*/
for i := 1; i < sliceLen; i++ {
// 拼接字符串
// Splicing string
if config.Dialect == "tdengine" { // 如果是tdengine,拼接类似 INSERT INTO table1 values('2','3') table2 values('4','5'),目前要求字段和类型必须一致,如果不一致,改动略多
sqlBuilder.WriteByte(' ')
sqlBuilder.WriteString(entityStructSlice[i].GetTableName())
if config.TDengineInsertsColumnName {
sqlBuilder.WriteString(*inserColumnName)
}
sqlBuilder.WriteString(" VALUES")
sqlBuilder.WriteString(*valuesql)
} else { // 标准语法 类似 INSERT INTO table1(id,name) values('2','3'),('4','5')
sqlBuilder.WriteByte(',')
sqlBuilder.WriteString(*valuesql)
}
entityStruct := entityStructSlice[i]
//默认值的map,每一个对象的默认值
defaultValueMap := entityStruct.GetDefaultValue()
for j := 0; j < len(*columns); j++ {
// 获取实体类的反射,指针下的struct
// Get the reflection of the entity class, the struct under the pointer
valueOf := reflect.ValueOf(entityStruct).Elem()
field := (*columns)[j]
// 字段的值
// The value of the primary key
fieldValue := valueOf.FieldByName(field.Name)
if field.Name == pkFieldName { // 如果是主键 | If it is the primary key
pkKind := field.Type.Kind()
// pkValue := valueOf.FieldByName(field.Name).Interface()
// 只处理字符串类型的主键,其他类型,columns中并不包含
// Only handle primary keys of string type, other types, not included in columns
if (pkKind == reflect.String) && fieldValue.IsZero() {
// 主键是字符串类型,并且值为"",赋值'id'
// 生成主键字符串
// The primary key is a string type, and the value is "", assigned the value'id'
// Generate primary key string
id := FuncGenerateStringID(ctx)
*values = append(*values, id)
// 给对象主键赋值
// Assign a value to the primary key of the object
fieldValue.Set(reflect.ValueOf(id))
continue
}
}
//默认值
isDefaultValue := false
var defaultValue interface{}
if defaultValueMap != nil { //如果只更新不是零值的字段,零值时不能更新为默认值.Map取值性能一般高于反射
defaultValue, isDefaultValue = defaultValueMap[field.Name]
}
// 给字段赋值
// Assign a value to the field.
if isDefaultValue && fieldValue.IsZero() {
*values = append(*values, defaultValue)
} else {
*values = append(*values, fieldValue.Interface())
}
}
}
sqlstr = sqlBuilder.String()
return &sqlstr, autoIncrement, nil
}
// wrapInsertEntityMapSliceSQL 包装批量保存EntityMapSlice语句.返回语句,值,错误信息
var wrapInsertEntityMapSliceSQL = func(ctx context.Context, config *DataSourceConfig, entityMapSlice []IEntityMap) (*string, *[]interface{}, error) {
sliceLen := len(entityMapSlice)
sqlstr := ""
if entityMapSlice == nil || sliceLen < 1 {
return &sqlstr, nil, errors.New("->wrapInsertSliceSQL对象数组不能为空")
}
// 第一个对象,获取第一个Struct对象,用于获取数据库字段,也获取了值
entity := entityMapSlice[0]
// 检查是否是指针对象
_, err := checkEntityKind(entity)
if err != nil {
return &sqlstr, nil, err
}
dbFieldMapKey := entity.GetDBFieldMapKey()
// SQL语句
inserColumnName, valuesql, values, _, err := wrapInsertValueEntityMapSQL(entity)
if err != nil {
return &sqlstr, values, err
}
var sqlBuilder strings.Builder
// sqlBuilder.Grow(len(entity.GetTableName()) + len(inserColumnName) + len(valuesql) + 19)
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString("INSERT INTO ")
sqlBuilder.WriteString(entity.GetTableName())
// sqlstr = sqlstr + insertsql + " VALUES" + valuesql
sqlBuilder.WriteString(*inserColumnName)
sqlBuilder.WriteString(" VALUES")
sqlBuilder.WriteString(*valuesql)
for i := 1; i < sliceLen; i++ {
// 拼接字符串
// Splicing string
if config.Dialect == "tdengine" { // 如果是tdengine,拼接类似 INSERT INTO table1 values('2','3') table2 values('4','5'),目前要求字段和类型必须一致,如果不一致,改动略多
sqlBuilder.WriteByte(' ')
sqlBuilder.WriteString(entityMapSlice[i].GetTableName())
if config.TDengineInsertsColumnName {
sqlBuilder.WriteString(*inserColumnName)
}
sqlBuilder.WriteString(" VALUES")
sqlBuilder.WriteString(*valuesql)
} else { // 标准语法 类似 INSERT INTO table1(id,name) values('2','3'), values('4','5')
sqlBuilder.WriteByte(',')
sqlBuilder.WriteString(*valuesql)
}
entityMap := entityMapSlice[i]
for j := 0; j < len(dbFieldMapKey); j++ {
key := dbFieldMapKey[j]
value := entityMap.GetDBFieldMap()[key]
*values = append(*values, value)
}
}
sqlstr = sqlBuilder.String()
return &sqlstr, values, err
}
// wrapUpdateSQL 包装更新Struct语句
// 数组传递,如果外部方法有调用append的逻辑,append会破坏指针引用,所以传递指针
// wrapUpdateSQL Package update Struct statement
// Array transfer, if the external method has logic to call append, append will destroy the pointer reference, so the pointer is passed
var wrapUpdateSQL = func(ctx context.Context, typeOf *reflect.Type, entity IEntityStruct, columns *[]reflect.StructField, values *[]interface{}) (string, error) {
// SQL语句的构造器
// SQL statement constructor
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString("UPDATE ")
sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteString(" SET ")
// 主键的值
// The value of the primary key
var pkValue interface{}
// 主键的名称
// The name of the primary key
pkFieldName, err := entityPKFieldName(entity, typeOf)
if err != nil {
return "", err
}
// 获取所有字段和tag对照的Map
tagMap, err := getStructFieldTagMap(typeOf)
if err != nil {
return "", err
}
for i := 0; i < len(*columns); i++ {
field := (*columns)[i]
if field.Name == pkFieldName {
// 如果是主键
// If it is the primary key.
pkValue = (*values)[i]
// 去掉这一列,最后处理主键
// Remove this column, and finally process the primary key
*columns = append((*columns)[:i], (*columns)[i+1:]...)
*values = append((*values)[:i], (*values)[i+1:]...)
i = i - 1
continue
}
/*
// 如果是默认值字段,删除掉,不更新
// If it is the default value field, delete it and do not update
if onlyUpdateNotZero && ((*values)[i] == nil || reflect.ValueOf((*values)[i]).IsZero()) {
// 去掉这一列,不再处理
// Remove this column and no longer process
*columns = append((*columns)[:i], (*columns)[i+1:]...)
*values = append((*values)[:i], (*values)[i+1:]...)
i = i - 1
continue
}
*/
if i > 0 {
sqlBuilder.WriteByte(',')
}
colName := getFieldTagName(&field, tagMap)
sqlBuilder.WriteString(colName)
sqlBuilder.WriteString("=?")
}
// 主键的值是最后一个
// The value of the primary key is the last
*values = append(*values, pkValue)
// sqlstr = sqlstr + " WHERE " + entity.GetPKColumnName() + "=?"
sqlBuilder.WriteString(" WHERE ")
sqlBuilder.WriteString(entity.GetPKColumnName())
sqlBuilder.WriteString("=?")
sqlstr := sqlBuilder.String()
return sqlstr, nil
}
// wrapDeleteSQL 包装删除Struct语句
// wrapDeleteSQL Package delete Struct statement
var wrapDeleteSQL = func(ctx context.Context, entity IEntityStruct) (string, error) {
// SQL语句的构造器
// SQL statement constructor
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString("DELETE FROM ")
sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteString(" WHERE ")
sqlBuilder.WriteString(entity.GetPKColumnName())
sqlBuilder.WriteString("=?")
sqlstr := sqlBuilder.String()
return sqlstr, nil
}
// wrapInsertEntityMapSQL 包装保存Map语句,Map因为没有字段属性,无法完成Id的类型判断和赋值,需要确保Map的值是完整的
// wrapInsertEntityMapSQL Pack and save the Map statement. Because Map does not have field attributes,
// it cannot complete the type judgment and assignment of Id. It is necessary to ensure that the value of Map is complete
var wrapInsertEntityMapSQL = func(ctx context.Context, config *DataSourceConfig, entity IEntityMap) (string, *[]interface{}, bool, error) {
sqlstr := ""
inserColumnName, valuesql, values, autoIncrement, err := wrapInsertValueEntityMapSQL(entity)
if err != nil {
return sqlstr, nil, autoIncrement, err
}
// 拼接SQL语句,带上列名,因为Map取值是无序的
// sqlstr := "INSERT INTO " + insertsql + " VALUES" + valuesql
var sqlBuilder strings.Builder
// sqlBuilder.Grow(len(inserColumnName) + len(entity.GetTableName()) + len(valuesql) + 19)
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString("INSERT INTO ")
sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteString(*inserColumnName)
sqlBuilder.WriteString(" VALUES")
sqlBuilder.WriteString(*valuesql)
sqlstr = sqlBuilder.String()
return sqlstr, values, autoIncrement, nil
}
// wrapInsertValueEntityMapSQL 包装保存Map语句,Map因为没有字段属性,无法完成Id的类型判断和赋值,需要确保Map的值是完整的
// wrapInsertValueEntityMapSQL Pack and save the Map statement. Because Map does not have field attributes,
// it cannot complete the type judgment and assignment of Id. It is necessary to ensure that the value of Map is complete
func wrapInsertValueEntityMapSQL(entity IEntityMap) (*string, *string, *[]interface{}, bool, error) {
var inserColumnName, valuesql string
// 是否自增,默认false
autoIncrement := false
dbFieldMap := entity.GetDBFieldMap()
if len(dbFieldMap) < 1 {
return &inserColumnName, &inserColumnName, nil, autoIncrement, errors.New("->wrapInsertEntityMapSQL-->GetDBFieldMap返回值不能为空")
}
// SQL对应的参数
// SQL corresponding parameters
values := []interface{}{}
// SQL语句的构造器
// SQL statement constructor
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
// sqlBuilder.WriteString("INSERT INTO ")
// sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteByte('(')
// SQL语句中,VALUES(?,?,...)语句的构造器
// In the SQL statement, the constructor of the VALUES(?,?,...) statement.
var valueSQLBuilder strings.Builder
valueSQLBuilder.Grow(stringBuilderGrowLen)
valueSQLBuilder.WriteString(" (")
// 是否Set了主键
// Whether the primary key is set.
_, hasPK := dbFieldMap[entity.GetPKColumnName()]
if entity.GetPKColumnName() != "" && !hasPK { // 如果有主键字段,却没值,认为是自增或者序列 | If the primary key is not set, it is considered to be auto-increment or sequence
autoIncrement = true
if entity.GetEntityMapPkSequence() != "" { // 如果是序列 | If it is a sequence.
sqlBuilder.WriteString(entity.GetPKColumnName())
valueSQLBuilder.WriteString(entity.GetEntityMapPkSequence())
if len(dbFieldMap) > 1 { // 如果不只有序列
sqlBuilder.WriteByte(',')
valueSQLBuilder.WriteByte(',')
}
}
}
dbFieldMapKey := entity.GetDBFieldMapKey()
for dbFieldMapIndex := 0; dbFieldMapIndex < len(dbFieldMapKey); dbFieldMapIndex++ {
if dbFieldMapIndex > 0 {
sqlBuilder.WriteByte(',')
valueSQLBuilder.WriteByte(',')
}
k := dbFieldMapKey[dbFieldMapIndex]
v := dbFieldMap[k]
// 拼接字符串
// Concatenated string
sqlBuilder.WriteString(k)
valueSQLBuilder.WriteByte('?')
values = append(values, v)
}
sqlBuilder.WriteByte(')')
valueSQLBuilder.WriteByte(')')
inserColumnName = sqlBuilder.String()
valuesql = valueSQLBuilder.String()
return &inserColumnName, &valuesql, &values, autoIncrement, nil
}
// wrapUpdateEntityMapSQL 包装Map更新语句,Map因为没有字段属性,无法完成Id的类型判断和赋值,需要确保Map的值是完整的
// wrapUpdateEntityMapSQL Wrap the Map update statement. Because Map does not have field attributes,
// it cannot complete the type judgment and assignment of Id. It is necessary to ensure that the value of Map is complete
var wrapUpdateEntityMapSQL = func(ctx context.Context, entity IEntityMap) (*string, *[]interface{}, error) {
dbFieldMap := entity.GetDBFieldMap()
sqlstr := ""
if len(dbFieldMap) < 1 {
return &sqlstr, nil, errors.New("->wrapUpdateEntityMapSQL-->GetDBFieldMap返回值不能为空")
}
// SQL语句的构造器
// SQL statement constructor
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString("UPDATE ")
sqlBuilder.WriteString(entity.GetTableName())
sqlBuilder.WriteString(" SET ")
// SQL对应的参数
// SQL corresponding parameters
values := []interface{}{}
// 主键名称
// Primary key name
var pkValue interface{}
dbFieldMapIndex := 0
for k, v := range dbFieldMap {
if k == entity.GetPKColumnName() { // 如果是主键 | If it is the primary key
pkValue = v
continue
}
if dbFieldMapIndex > 0 {
sqlBuilder.WriteByte(',')
}
// 拼接字符串 | Splicing string.
sqlBuilder.WriteString(k)
sqlBuilder.WriteString("=?")
values = append(values, v)
dbFieldMapIndex++
}
// 主键的值是最后一个
// The value of the primary key is the last
values = append(values, pkValue)
sqlBuilder.WriteString(" WHERE ")
sqlBuilder.WriteString(entity.GetPKColumnName())
sqlBuilder.WriteString("=?")
sqlstr = sqlBuilder.String()
return &sqlstr, &values, nil
}
// wrapQuerySQL 封装查询语句
// wrapQuerySQL Encapsulated query statement
func wrapQuerySQL(ctx context.Context, config *DataSourceConfig, finder *Finder, page *Page) (string, error) {
// 获取到没有page的sql的语句
// Get the SQL statement without page.
sqlstr, err := finder.GetSQL()
if err != nil {
return "", err
}
if page != nil {
err = wrapPageSQL(ctx, config, &sqlstr, page)
}
if err != nil {
return "", err
}
return sqlstr, err
}
// 查询'order by'在sql中出现的开始位置和结束位置
// Query the start position and end position of'order by' in SQL
var (
orderByExpr = `(?i)\s(order)\s+by\s`
orderByRegexp, _ = regexp.Compile(orderByExpr)
)
// findOrderByIndex 查询order by在sql中出现的开始位置和结束位置
// findOrderByIndex Query the start position and end position of'order by' in SQL
func findOrderByIndex(strsql *string) []int {
loc := orderByRegexp.FindStringIndex(*strsql)
return loc
}
// 查询'group by'在sql中出现的开始位置和结束位置
// Query the start position and end position of'group by' in sql。
var (
groupByExpr = `(?i)\s(group)\s+by\s`
groupByRegexp, _ = regexp.Compile(groupByExpr)
)
// findGroupByIndex 查询group by在sql中出现的开始位置和结束位置
// findGroupByIndex Query the start position and end position of'group by' in sql
func findGroupByIndex(strsql *string) []int {
loc := groupByRegexp.FindStringIndex(*strsql)
return loc
}
// 查询 from 在sql中出现的开始位置和结束位置
// Query the start position and end position of 'from' in sql
// var fromExpr = "(?i)(^\\s*select)(.+?\\(.+?\\))*.*?(from)"
// 感谢奔跑(@zeqjone)提供的正则,排除不在括号内的from,已经满足绝大部分场景,
// select id1,(select (id2) from t1 where id=2) _s FROM table select的子查询 _s中的 id2还有括号,才会出现问题,建议使用CountFinder处理分页语句
// countFinder := zorm.NewFinder().Append("select count(*) from (")
// countFinder.AppendFinder(finder)
// countFinder.Append(") tempcountfinder")
// finder.CountFinder = countFinder
var (
//fromExpr = "(?i)(^\\s*select)(\\(.*?\\)|[^()]+)*?( from )"
// 处理sql中from格式化换行的正则问题
fromExpr = `(?i)(^\s*select)(\(.*?\)|[^()]+)*?(\s+from\s)`
fromRegexp, _ = regexp.Compile(fromExpr)
)
// findFromIndexa 查询from在sql中出现的开始位置和结束位置
// findSelectFromIndex Query the start position and end position of 'from' in sql
func findSelectFromIndex(strsql *string) []int {
// 匹配出来的是完整的字符串,用最后的FROM即可
loc := fromRegexp.FindStringIndex(*strsql)
if len(loc) < 2 {
return loc
}
// 最后的FROM前推5位字符串,没有空格
loc[0] = loc[1] - 5
return loc
}
/*
var fromExpr = `\(([\s\S]+?)\)`
var fromRegexp, _ = regexp.Compile(fromExpr)
//查询 from 在sql中出现的开始位置
//Query the start position of 'from' in sql
func findSelectFromIndex(strsql string) int {
sql := strings.ToLower(strsql)
m := fromRegexp.FindAllString(sql, -1)
for i := 0; i < len(m); i++ {
str := m[i]
strnofrom := strings.ReplaceAll(str, " from ", " zorm ")
sql = strings.ReplaceAll(sql, str, strnofrom)
}
fromIndex := strings.LastIndex(sql, " from ")
if fromIndex < 0 {
return fromIndex
}
//补上一个空格
fromIndex = fromIndex + 1
return fromIndex
}
*/
/*
// 从更新语句中获取表名
//update\\s(.+)set\\s.*
var (
updateExper = "(?i)^\\s*update\\s+(\\w+)\\s+set\\s"
updateRegexp, _ = regexp.Compile(updateExper)
)
// findUpdateTableName 获取语句中表名
// 第一个是符合的整体数据,第二个是表名
func findUpdateTableName(strsql *string) []string {
matchs := updateRegexp.FindStringSubmatch(*strsql)
return matchs
}
// 从删除语句中获取表名
// delete\\sfrom\\s(.+)where\\s(.*)
var (
deleteExper = "(?i)^\\s*delete\\s+from\\s+(\\w+)\\s+where\\s"
deleteRegexp, _ = regexp.Compile(deleteExper)
)
// findDeleteTableName 获取语句中表名
// 第一个是符合的整体数据,第二个是表名
func findDeleteTableName(strsql *string) []string {
matchs := deleteRegexp.FindStringSubmatch(*strsql)
return matchs
}
*/
// FuncGenerateStringID 默认生成字符串ID的函数.方便自定义扩展
// FuncGenerateStringID Function to generate string ID by default. Convenient for custom extension
var FuncGenerateStringID = func(ctx context.Context) string {
// 使用 crypto/rand 真随机9位数
randNum, randErr := rand.Int(rand.Reader, big.NewInt(1000000000))
if randErr != nil {
return ""
}
// 获取9位数,前置补0,确保9位数
rand9 := fmt.Sprintf("%09d", randNum)
// 获取纳秒 按照 年月日时分秒毫秒微秒纳秒 拼接为长度23位的字符串
pk := time.Now().Format("2006.01.02.15.04.05.000000000")
pk = strings.ReplaceAll(pk, ".", "")
// 23位字符串+9位随机数=32位字符串,这样的好处就是可以使用ID进行排序
pk = pk + rand9
return pk
}
// FuncWrapFieldTagName 用于包裹字段名, eg. `describe`
var FuncWrapFieldTagName func(field *reflect.StructField, colName string) string = nil
/*
var FuncWrapFieldTagName = func(field *reflect.StructField, colName string) string {
// return fmt.Sprintf("`%s`", colName)
return colName
}
*/
// getFieldTagName 获取模型中定义的数据库的 column tag
func getFieldTagName(field *reflect.StructField, structFieldTagMap *map[string]string) string {
// colName := field.Tag.Get(tagColumnName)
colName := (*structFieldTagMap)[field.Name]
if FuncWrapFieldTagName != nil {
colName = FuncWrapFieldTagName(field, colName)
}
/*
if dialect == "kingbase" {
// kingbase R3 驱动大小写敏感,通常是大写。数据库全的列名部换成双引号括住的大写字符,避免与数据库内置关键词冲突时报错
colName = strings.ReplaceAll(colName, "\"", "")
colName = fmt.Sprintf(`"%s"`, strings.ToUpper(colName))
}
*/
return colName
}
// wrapSQLHint 在sql语句中增加hint
func wrapSQLHint(ctx context.Context, sqlstr *string) error {
// 获取hint
contextValue := ctx.Value(contextSQLHintValueKey)
if contextValue == nil { // 如果没有设置hint
return nil
}
hint, ok := contextValue.(string)
if !ok {
return errors.New("->wrapSQLHint-->contextValue转换string失败")
}
if hint == "" {
return nil
}
sqlByte := []byte(*sqlstr)
// 获取第一个单词
_, start, end, err := firstOneWord(0, &sqlByte)
if err != nil {
return err
}
if start == -1 || end == -1 { // 未取到字符串
return nil
}
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString((*sqlstr)[:end])
sqlBuilder.WriteByte(' ')
sqlBuilder.WriteString(hint)
sqlBuilder.WriteString((*sqlstr)[end:])
*sqlstr = sqlBuilder.String()
return nil
}
// reBuildSQL 包装基础的SQL语句,根据数据库类型,调整SQL变量符号,例如?,? $1,$2这样的
// reBuildSQL Pack basic SQL statements, adjust the SQL variable symbols according to the database type, such as?,? $1,$2
var reBuildSQL = func(ctx context.Context, config *DataSourceConfig, sqlstr *string, args *[]interface{}) (*string, *[]interface{}, error) {
argsNum := len(*args)
if argsNum < 1 { // 没有参数,不需要处理,也不判断参数数量了,数据库会报错提示
return sqlstr, args, nil
}
// 重新记录参数值
// Re-record the parameter value
newValues := make([]interface{}, 0)
// 记录sql参数值的下标,例如 $1 @p1 ,从1开始
sqlParamIndex := 1
// 新的sql
// new sql
var newSQLStr strings.Builder
// newSQLStr.Grow(len(*sqlstr))
newSQLStr.Grow(stringBuilderGrowLen)
i := -1
for _, vbyte := range []byte(*sqlstr) {
if vbyte != '?' { // 如果不是?问号
newSQLStr.WriteByte(vbyte)
continue
}
i = i + 1
if i >= argsNum { // 占位符数量比参数值多,不使用 strings.Count函数,避免多次操作字符串
return nil, nil, fmt.Errorf("sql语句中参数和值数量不一致,-->zormErrorExecSQL:%s,-->zormErrorSQLValues:%s", *sqlstr, sqlErrorValues2String(*args))
}
v := (*args)[i]
var valueOf reflect.Value
var kind reflect.Kind
var typeOf reflect.Type
// 参数值长度,默认是1,其他取值数组长度
valueLen := 1
// 值是否有效,例如指针类型的值为nil
isValid := false
if v != nil {
isValid = true
// 反射获取参数的值
valueOf = reflect.ValueOf(v)
// 获取类型
kind = valueOf.Kind()
// 如果参数是个指针类型
// If the parameter is a pointer type
if kind == reflect.Ptr { // 如果是指针 | If it is a pointer
valueOf = valueOf.Elem()
kind = valueOf.Kind()
//(Type=*int,Value=nil)/(Type=nil,Value=nil) https://golang.org/doc/faq#nil_error
isValid = valueOf.IsValid()
}
}
// 如果是有效值,获取类型
if isValid {
typeOf = valueOf.Type()
}
if !isValid { // 如果是无效值,设值为nil
newValues = append(newValues, nil)
} else if kind != reflect.Array && kind != reflect.Slice { // 如果不是数组或者slice
// 记录新值
// Record new value.
newValues = append(newValues, v)
//} else if _, ok := v.([]byte); ok { //字节数组
// newValues = append(newValues, v)
} else if kind == reflect.Slice && typeOf.Elem().Kind() == reflect.Uint8 { //[]byte字节数组或派生类型
// 记录新值
// Record new value
newValues = append(newValues, v)
} else {
// 获取数组类型参数值的长度
// Get the length of the array type parameter value
valueLen = valueOf.Len()
// 数组类型的参数长度小于1,认为是有异常的参数
// The parameter length of the array type is less than 1, which is considered to be an abnormal parameter
if valueLen < 1 {
// return nil, nil, errors.New("->reBuildSQL()语句:" + *sqlstr + ",第" + strconv.Itoa(i+1) + "个参数,类型是Array或者Slice,值的长度为0,请检查sql参数有效性")
valueLen = 1
newValues = append(newValues, nil)
} else if valueLen == 1 { // 如果数组里只有一个参数,认为是单个参数
v = valueOf.Index(0).Interface()
newValues = append(newValues, v)
}
}
switch config.Dialect {
case "mysql", "sqlite", "dm", "gbase", "clickhouse", "db2":
wrapParamSQL("?", valueLen, &sqlParamIndex, &newSQLStr, &valueOf, &newValues, false, false)
case "postgresql", "kingbase": // postgresql,kingbase
wrapParamSQL("$", valueLen, &sqlParamIndex, &newSQLStr, &valueOf, &newValues, true, false)
case "mssql": // mssql
wrapParamSQL("@p", valueLen, &sqlParamIndex, &newSQLStr, &valueOf, &newValues, true, false)
case "oracle", "shentong": // oracle,神通
wrapParamSQL(":", valueLen, &sqlParamIndex, &newSQLStr, &valueOf, &newValues, true, false)
case "tdengine": // tdengine,重新处理 字符类型的参数 '?'
wrapParamSQL("?", valueLen, &sqlParamIndex, &newSQLStr, &valueOf, &newValues, false, true)
default: // 其他情况,还是使用 ? | In other cases, or use ?
newSQLStr.WriteByte('?')
}
}
//?号占位符的数量和参数不一致,不使用 strings.Count函数,避免多次操作字符串
if (i + 1) != argsNum {
return nil, nil, fmt.Errorf("sql语句中参数和值数量不一致,-->zormErrorExecSQL:%s,-->zormErrorSQLValues:%s", *sqlstr, sqlErrorValues2String(*args))
}
sqlstring := newSQLStr.String()
return &sqlstring, &newValues, nil
}
// reUpdateFinderSQL 根据数据类型更新 手动编写的 UpdateFinder的语句,用于处理数据库兼容,例如 clickhouse的 UPDATE 和 DELETE
var reBuildUpdateSQL = func(ctx context.Context, config *DataSourceConfig, sqlstr *string) error {
if config.Dialect != "clickhouse" { // 目前只处理clickhouse
return nil
}
// 处理clickhouse的特殊更新语法
sqlByte := []byte(*sqlstr)
// 获取第一个单词
firstWord, start, end, err := firstOneWord(0, &sqlByte)
if err != nil {
return err
}
if start == -1 || end == -1 { // 未取到字符串
return nil
}
// SQL语句的构造器
// SQL statement constructor
var sqlBuilder strings.Builder
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString((*sqlstr)[:start])
sqlBuilder.WriteString("ALTER TABLE ")
firstWord = strings.ToUpper(firstWord)
tableName := ""
if firstWord == "UPDATE" { // 更新 update tableName set
tableName, _, end, err = firstOneWord(end, &sqlByte)
if err != nil {
return err
}
// 拿到 set
_, start, end, err = firstOneWord(end, &sqlByte)
} else if firstWord == "DELETE" { // 删除 delete from tableName
// 拿到from
_, _, end, err = firstOneWord(end, &sqlByte)
if err != nil {
return err
}
// 拿到 tableName
tableName, start, end, err = firstOneWord(end, &sqlByte)
} else { // 只处理UPDATE 和 DELETE 语法
return nil
}
if err != nil {
return err
}
if start == -1 || end == -1 { // 获取的位置异常
return errors.New("->reBuildUpdateSQL中clickhouse语法异常,请检查sql语句是否标准,-->zormErrorExecSQL:" + *sqlstr)
}
sqlBuilder.WriteString(tableName)
sqlBuilder.WriteByte(' ')
sqlBuilder.WriteString(firstWord)
// sqlBuilder.WriteByte(' ')
sqlBuilder.WriteString((*sqlstr)[end:])
*sqlstr = sqlBuilder.String()
return nil
}
// wrapAutoIncrementInsertSQL 包装自增的自增主键的插入sql
var wrapAutoIncrementInsertSQL = func(ctx context.Context, config *DataSourceConfig, pkColumnName string, sqlstr *string, values *[]interface{}) (*int64, *int64) {
// oracle 12c+ 支持IDENTITY属性的自增列,因为分页也要求12c+的语法,所以数据库就IDENTITY创建自增吧
// 处理序列产生的自增主键,例如oracle,postgresql等
var lastInsertID, zormSQLOutReturningID *int64
var sqlBuilder strings.Builder
// sqlBuilder.Grow(len(*sqlstr) + len(pkColumnName) + 40)
sqlBuilder.Grow(stringBuilderGrowLen)
sqlBuilder.WriteString(*sqlstr)
switch config.Dialect {
case "postgresql", "kingbase":
var p int64 = 0
lastInsertID = &p
// sqlstr = sqlstr + " RETURNING " + pkColumnName
sqlBuilder.WriteString(" RETURNING ")
sqlBuilder.WriteString(pkColumnName)
case "oracle", "shentong":
var p int64 = 0
zormSQLOutReturningID = &p
// sqlstr = sqlstr + " RETURNING " + pkColumnName + " INTO :zormSQLOutReturningID "
sqlBuilder.WriteString(" RETURNING ")
sqlBuilder.WriteString(pkColumnName)
sqlBuilder.WriteString(" INTO :zormSQLOutReturningID ")
v := sql.Named("zormSQLOutReturningID", sql.Out{Dest: zormSQLOutReturningID})
*values = append(*values, v)
}
*sqlstr = sqlBuilder.String()
return lastInsertID, zormSQLOutReturningID
}
// getConfigFromConnection 从dbConnection中获取数据库方言,如果没有,从FuncReadWriteStrategy获取dbDao,获取dbdao.config.Dialect
func getConfigFromConnection(ctx context.Context, dbConnection *dataBaseConnection, rwType int) (*DataSourceConfig, error) {
var config *DataSourceConfig
// dbConnection为nil,使用defaultDao
// dbConnection is nil, use default Dao
if dbConnection == nil {
dbdao, err := FuncReadWriteStrategy(ctx, rwType)
if err != nil {
return nil, err
}
config = dbdao.config
} else {
config = dbConnection.config
}
return config, nil
}
// wrapParamSQL 包装SQL语句
// symbols(占位符) valueLen(参数长度) sqlParamIndexPtr(参数的下标指针,数组会改变值) newSQLStr(SQL字符串Builder) valueOf(参数值的反射对象) hasParamIndex(是否拼接参数下标 $1 $2) isTDengine(TDengine数据库需要单独处理字符串类型)
func wrapParamSQL(symbols string, valueLen int, sqlParamIndexPtr *int, newSQLStr *strings.Builder, valueOf *reflect.Value, newValues *[]interface{}, hasParamIndex bool, isTDengine bool) {
sqlParamIndex := *sqlParamIndexPtr
if valueLen == 1 {
if isTDengine && valueOf.Kind() == reflect.String { // 处理tdengine的字符串类型
symbols = "'?'"
}
newSQLStr.WriteString(symbols)
if hasParamIndex {
newSQLStr.WriteString(strconv.Itoa(sqlParamIndex))
}
} else if valueLen > 1 { // 如果值是数组
for j := 0; j < valueLen; j++ {
valuej := (*valueOf).Index(j)
if isTDengine && valuej.Kind() == reflect.String { // 处理tdengine的字符串类型
symbols = "'?'"
}
if j == 0 { // 第一个
newSQLStr.WriteString(symbols)
} else {
newSQLStr.WriteByte(',')
newSQLStr.WriteString(symbols)
}
if hasParamIndex {
newSQLStr.WriteString(strconv.Itoa(sqlParamIndex + j))
}
sliceValue := valuej.Interface()
*newValues = append(*newValues, sliceValue)
}
}
*sqlParamIndexPtr = *sqlParamIndexPtr + valueLen
}
// firstOneWord 从指定下标,获取第一个单词,不包含前后空格,并返回开始下标和结束下标,如果找不到合法的字符串,返回-1
func firstOneWord(index int, strByte *[]byte) (string, int, int, error) {
start := -1
end := -1
byteLen := len(*strByte)
if index < 0 {
return "", start, end, errors.New("->firstOneWord索引小于0")
}
if index > byteLen { // 如果索引大于长度
return "", start, end, errors.New("->firstOneWord索引大于字符串长度")
}
var newStr strings.Builder
newStr.Grow(10)
for ; index < byteLen; index++ {
v := (*strByte)[index]
if v == '(' || v == ')' { // 不处理括号
continue
}
if start == -1 && v != ' ' { // 不是空格
start = index
}
if start == -1 && v == ' ' { // 空格
continue
}
if start >= 0 && v != ' ' { // 需要的字符
newStr.WriteByte(v)
} else { // 遇到空格结束记录
end = index
break
}
}
if start >= 0 && end == -1 { // 记录到结尾,不是空格结束
end = byteLen
}
return newStr.String(), start, end, nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化