kratos-utils/common/sql.go

package common

import (
	"fmt"
	"strings"
)

// SqlStrReplaceSingleQuotes 防止sql注入 单引号
func SqlStrReplaceSingleQuotes(str string) string {
	if strings.Contains(str, "'") {
		str = strings.Replace(str, "'", "\\'", -1)
	}
	return str
}

// SqlStrReplaceDoubleQuotes 防止sql注入 双引号
func SqlStrReplaceDoubleQuotes(str string) string {
	if strings.Contains(str, "\"") {
		str = strings.Replace(str, "\"", "\\\"", -1)
	}
	return str
}

func GetSqlInArrayStrToString(arr []string) string {
	return "'" + strings.Join(arr, "','") + "'"
}

func GetSqlInArrayInt64ToString(arr []int64) string {
	str := ""
	for _, v := range arr {
		str = str + fmt.Sprintf("%d,", v)
	}
	str = str[:len(str)-1]
	return str
}

func ArrayInt64ToString(ids []int64) []string {
	idsStr := make([]string, 0)
	for _, id := range ids {
		idsStr = append(idsStr, fmt.Sprint(id))
	}
	return idsStr
}