mirror of
https://github.com/zr-hebo/sniffer-agent.git
synced 2025-08-09 16:00:01 +08:00
add vendor packages避免用户网络太慢无法下载编译
This commit is contained in:
16
vendor/github.com/zr-hebo/util-db/.gitignore
generated
vendored
Normal file
16
vendor/github.com/zr-hebo/util-db/.gitignore
generated
vendored
Normal file
@@ -0,0 +1,16 @@
|
||||
# Binaries for programs and plugins
|
||||
*.exe
|
||||
*.dll
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# Test binary, build with `go test -c`
|
||||
*.test
|
||||
|
||||
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||
*.out
|
||||
|
||||
# Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
|
||||
.glide/
|
||||
.idea/
|
||||
*.log
|
201
vendor/github.com/zr-hebo/util-db/LICENSE
generated
vendored
Normal file
201
vendor/github.com/zr-hebo/util-db/LICENSE
generated
vendored
Normal file
@@ -0,0 +1,201 @@
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright [yyyy] [name of copyright owner]
|
||||
|
||||
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.
|
2
vendor/github.com/zr-hebo/util-db/README.md
generated
vendored
Normal file
2
vendor/github.com/zr-hebo/util-db/README.md
generated
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
# db_util
|
||||
golang implement database operation util
|
93
vendor/github.com/zr-hebo/util-db/conn_info.go
generated
vendored
Normal file
93
vendor/github.com/zr-hebo/util-db/conn_info.go
generated
vendored
Normal file
@@ -0,0 +1,93 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"runtime/debug"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// GetMySQLConnInfoIgnoreErr 获取MySQL连接的字符串显示,忽略错误信息
|
||||
func GetMySQLConnInfoIgnoreErr(conn *sql.DB) (connInfo string) {
|
||||
connInfo, _ = GetMySQLConnInfo(conn)
|
||||
return
|
||||
}
|
||||
|
||||
// GetMySQLConnInfo 获取MySQL连接的字符串显示
|
||||
func GetMySQLConnInfo(conn *sql.DB) (connInfo string, err error) {
|
||||
defer func() {
|
||||
if panicRecover := recover(); panicRecover != nil {
|
||||
err = fmt.Errorf(
|
||||
"get mysql connection info failed for %v", panicRecover)
|
||||
debug.PrintStack()
|
||||
}
|
||||
}()
|
||||
|
||||
cv := reflect.ValueOf(conn).Elem()
|
||||
// fmt.Printf("%#v\n", cv)
|
||||
|
||||
dsnv := cv.FieldByName("dsn")
|
||||
dsnStr := fmt.Sprint(dsnv)
|
||||
dsnInfo, err := resolveDsn(dsnStr)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
fcsv := cv.FieldByName("dep")
|
||||
// fmt.Printf("%#v\n", fcsv)
|
||||
|
||||
lports := make([]string, 0)
|
||||
for _, key := range fcsv.MapKeys() {
|
||||
// fcv := fcsv.Index(0).Elem()
|
||||
depSet := fcsv.MapIndex(key).MapKeys()
|
||||
if len(depSet) < 1 {
|
||||
continue
|
||||
}
|
||||
|
||||
fcv := depSet[0].Elem().Elem()
|
||||
// fmt.Printf("%#v\n", fcv)
|
||||
// fmt.Printf("%#v\n", fcv.Elem())
|
||||
|
||||
mc := fcv.FieldByName("ci").Elem().Elem()
|
||||
// fmt.Printf("%#v", mc)
|
||||
|
||||
nc := mc.FieldByName("netConn").Elem().Elem()
|
||||
// fmt.Printf("%#v", nc)
|
||||
|
||||
ic := nc.FieldByName("conn")
|
||||
// fmt.Printf("%#v", conn)
|
||||
|
||||
fd := ic.FieldByName("fd").Elem()
|
||||
// fmt.Printf("%#v", fd)
|
||||
|
||||
la := fd.FieldByName("laddr").Elem().Elem()
|
||||
// fmt.Printf("%#v", la)
|
||||
|
||||
lport := la.FieldByName("Port")
|
||||
|
||||
lports = append(lports, fmt.Sprint(lport))
|
||||
// fmt.Printf("%#v", lport)
|
||||
}
|
||||
|
||||
connInfo = fmt.Sprintf(
|
||||
"127.0.0.1:%s <==> %s:%s", strings.Join(lports, ", "),
|
||||
dsnInfo["host"], dsnInfo["port"])
|
||||
return
|
||||
}
|
||||
|
||||
func resolveDsn(dsn string) (info map[string]string, err error) {
|
||||
info = make(map[string]string)
|
||||
|
||||
pattern := regexp.MustCompile(`.*:.*@tcp\((?P<host>[\w.]+):(?P<port>\d+)\)/`)
|
||||
|
||||
match := pattern.FindStringSubmatch(dsn)
|
||||
for idx, name := range pattern.SubexpNames() {
|
||||
if idx > 0 && idx <= len(match) {
|
||||
info[name] = match[idx]
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
16
vendor/github.com/zr-hebo/util-db/conn_operation.go
generated
vendored
Normal file
16
vendor/github.com/zr-hebo/util-db/conn_operation.go
generated
vendored
Normal file
@@ -0,0 +1,16 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
)
|
||||
|
||||
// CloseConnection 关闭数据库连接
|
||||
func CloseConnection(conn *sql.DB) (err error) {
|
||||
err = conn.Close()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = GetMySQLConnInfo(conn)
|
||||
return
|
||||
}
|
359
vendor/github.com/zr-hebo/util-db/query_db.go
generated
vendored
Normal file
359
vendor/github.com/zr-hebo/util-db/query_db.go
generated
vendored
Normal file
@@ -0,0 +1,359 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
dbTypeMysql = "mysql"
|
||||
)
|
||||
|
||||
// Host 主机
|
||||
type Host struct {
|
||||
IP string `json:"ip"`
|
||||
Domain string `json:"domain"`
|
||||
Port int `json:"port"`
|
||||
}
|
||||
|
||||
// UnanimityHost id标示的主机
|
||||
type UnanimityHost struct {
|
||||
Host string `json:"host"`
|
||||
Port int `json:"port"`
|
||||
}
|
||||
|
||||
func (uh *UnanimityHost) String() string {
|
||||
return fmt.Sprintf("%s:%d", uh.Host, uh.Port)
|
||||
}
|
||||
|
||||
// UnanimityHostWithDomains 带域名的id标示的主机
|
||||
type UnanimityHostWithDomains struct {
|
||||
UnanimityHost
|
||||
IP string `json:"ip"`
|
||||
Domains []string `json:"domains"`
|
||||
}
|
||||
|
||||
// MysqlDB Mysql主机实例
|
||||
type MysqlDB struct {
|
||||
Host
|
||||
UserName string
|
||||
Passwd string
|
||||
DatabaseType string
|
||||
DBName string
|
||||
ConnectTimeout int
|
||||
}
|
||||
|
||||
// NewMysqlDB 创建MySQL数据库
|
||||
func NewMysqlDB() (md *MysqlDB) {
|
||||
md = new(MysqlDB)
|
||||
md.DatabaseType = dbTypeMysql
|
||||
return
|
||||
}
|
||||
|
||||
// NewMysqlDBWithAllParam 带参数创建MySQL数据库
|
||||
func NewMysqlDBWithAllParam(
|
||||
ip string, port int, userName, passwd, dbName string) (
|
||||
pmd *MysqlDB) {
|
||||
pmd = NewMysqlDB()
|
||||
pmd.IP = ip
|
||||
pmd.Port = port
|
||||
pmd.UserName = userName
|
||||
pmd.Passwd = passwd
|
||||
pmd.DBName = dbName
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// GetConnection 获取数据库连接
|
||||
func (md *MysqlDB) getConnection() (*sql.DB, error) {
|
||||
connStr := md.fillConnStr()
|
||||
|
||||
stmtDB, err := sql.Open(md.DatabaseType, connStr)
|
||||
if err != nil {
|
||||
if stmtDB != nil {
|
||||
stmtDB.Close()
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
stmtDB.SetMaxOpenConns(0)
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
|
||||
defer cancel()
|
||||
if err := stmtDB.PingContext(ctx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return stmtDB, nil
|
||||
}
|
||||
|
||||
// GetConnection 获取数据库连接
|
||||
func (md *MysqlDB) getRealConnection(ctx context.Context) (*sql.Conn, error) {
|
||||
connStr := md.fillConnStr()
|
||||
|
||||
stmtDB, err := sql.Open(md.DatabaseType, connStr)
|
||||
if err != nil {
|
||||
if stmtDB != nil {
|
||||
stmtDB.Close()
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
conn, err := stmtDB.Conn(ctx)
|
||||
if err != nil {
|
||||
if conn != nil {
|
||||
conn.Close()
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return conn, nil
|
||||
}
|
||||
|
||||
type Field struct {
|
||||
Name string
|
||||
Type string
|
||||
}
|
||||
|
||||
// FieldType Common type include "STRING", "FLOAT", "INT", "BOOL"
|
||||
func (f *Field) FieldType() string {
|
||||
return f.Type
|
||||
}
|
||||
|
||||
type QueryRow struct {
|
||||
Fields []Field
|
||||
Record map[string]interface{}
|
||||
}
|
||||
|
||||
type QueryRows struct {
|
||||
Fields []Field
|
||||
Records []map[string]interface{}
|
||||
}
|
||||
|
||||
func newQueryRow() *QueryRow {
|
||||
queryRow := new(QueryRow)
|
||||
queryRow.Fields = make([]Field, 0)
|
||||
queryRow.Record = make(map[string]interface{})
|
||||
return queryRow
|
||||
}
|
||||
|
||||
func newQueryRows() *QueryRows {
|
||||
queryRows := new(QueryRows)
|
||||
queryRows.Fields = make([]Field, 0)
|
||||
queryRows.Records = make([]map[string]interface{}, 0)
|
||||
return queryRows
|
||||
}
|
||||
|
||||
// QueryRows 执行MySQL Query语句,返回多条数据
|
||||
func (md *MysqlDB) QueryRows(stmt string) (queryRows *QueryRows, err error) {
|
||||
defer func() {
|
||||
if err != nil {
|
||||
err = fmt.Errorf("query rows on %s:%d failed <-- %s", md.IP, md.Port, err.Error())
|
||||
}
|
||||
}()
|
||||
|
||||
connStr := md.fillConnStr()
|
||||
|
||||
db, err := sql.Open(md.DatabaseType, connStr)
|
||||
if db != nil {
|
||||
defer db.Close()
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
rawRows, err := db.Query(stmt)
|
||||
if rawRows != nil {
|
||||
defer rawRows.Close()
|
||||
}
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
colTypes, err := rawRows.ColumnTypes()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
fields := make([]Field, 0, len(colTypes))
|
||||
for _, colType := range colTypes {
|
||||
fields = append(fields, Field{Name: colType.Name(), Type: getDataType(colType.DatabaseTypeName())})
|
||||
}
|
||||
|
||||
queryRows = newQueryRows()
|
||||
queryRows.Fields = fields
|
||||
for rawRows.Next() {
|
||||
receiver := createReceiver(fields)
|
||||
err = rawRows.Scan(receiver...)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
queryRows.Records = append(queryRows.Records, getRecordFromReceiver(receiver, fields))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func createReceiver(fields []Field) (receiver []interface{}) {
|
||||
receiver = make([]interface{}, 0, len(fields))
|
||||
for _, field := range fields {
|
||||
switch field.Type {
|
||||
case "string":
|
||||
{
|
||||
var val sql.NullString
|
||||
receiver = append(receiver, &val)
|
||||
}
|
||||
case "int64":
|
||||
{
|
||||
var val sql.NullInt64
|
||||
receiver = append(receiver, &val)
|
||||
}
|
||||
case "float64":
|
||||
{
|
||||
var val sql.NullFloat64
|
||||
receiver = append(receiver, &val)
|
||||
}
|
||||
case "bool":
|
||||
{
|
||||
var val sql.NullBool
|
||||
receiver = append(receiver, &val)
|
||||
}
|
||||
default:
|
||||
var val sql.NullString
|
||||
receiver = append(receiver, &val)
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func getRecordFromReceiver(receiver []interface{}, fields []Field) (record map[string]interface{}) {
|
||||
record = make(map[string]interface{})
|
||||
for idx := 0; idx < len(fields); idx++ {
|
||||
field := fields[idx]
|
||||
value := receiver[idx]
|
||||
switch field.Type {
|
||||
case "string":
|
||||
{
|
||||
nullVal := value.(*sql.NullString)
|
||||
record[field.Name] = nil
|
||||
if nullVal.Valid {
|
||||
record[field.Name] = nullVal.String
|
||||
}
|
||||
}
|
||||
case "int64":
|
||||
{
|
||||
nullVal := value.(*sql.NullInt64)
|
||||
record[field.Name] = nil
|
||||
if nullVal.Valid {
|
||||
record[field.Name] = nullVal.Int64
|
||||
}
|
||||
}
|
||||
case "float64":
|
||||
{
|
||||
nullVal := value.(*sql.NullFloat64)
|
||||
record[field.Name] = nil
|
||||
if nullVal.Valid {
|
||||
record[field.Name] = nullVal.Float64
|
||||
}
|
||||
}
|
||||
case "bool":
|
||||
{
|
||||
nullVal := value.(*sql.NullBool)
|
||||
record[field.Name] = nil
|
||||
if nullVal.Valid {
|
||||
record[field.Name] = nullVal.Bool
|
||||
}
|
||||
}
|
||||
default:
|
||||
nullVal := value.(*sql.NullString)
|
||||
record[field.Name] = nil
|
||||
if nullVal.Valid {
|
||||
record[field.Name] = nullVal.String
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func getDataType(dbColType string) (colType string) {
|
||||
var columnTypeDict = map[string]string{
|
||||
"VARCHAR": "string",
|
||||
"TEXT": "string",
|
||||
"NVARCHAR": "string",
|
||||
"DATETIME": "float64",
|
||||
"DECIMAL": "float64",
|
||||
"BOOL": "bool",
|
||||
"INT": "int64",
|
||||
"BIGINT": "int64",
|
||||
}
|
||||
|
||||
colType, ok := columnTypeDict[dbColType]
|
||||
if ok {
|
||||
return
|
||||
}
|
||||
|
||||
colType = "string"
|
||||
return
|
||||
}
|
||||
|
||||
// QueryRow 执行MySQL Query语句,返回1条或0条数据
|
||||
func (md *MysqlDB) QueryRow(stmt string) (row *QueryRow, err error) {
|
||||
defer func() {
|
||||
if err != nil {
|
||||
err = fmt.Errorf("query row failed <-- %s", err.Error())
|
||||
}
|
||||
}()
|
||||
|
||||
queryRows, err := md.QueryRows(stmt)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if len(queryRows.Records) < 1 {
|
||||
return
|
||||
}
|
||||
|
||||
row = newQueryRow()
|
||||
row.Fields = queryRows.Fields
|
||||
row.Record = queryRows.Records[0]
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// ExecChange 执行MySQL DML Query语句
|
||||
func (md *MysqlDB) ExecChange(stmt string, args ...interface{}) (
|
||||
result sql.Result, err error) {
|
||||
defer func() {
|
||||
if err != nil {
|
||||
err = fmt.Errorf("execute dml failed <-- %s", err.Error())
|
||||
}
|
||||
}()
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
||||
defer cancel()
|
||||
|
||||
conn, err := md.getRealConnection(ctx)
|
||||
if conn != nil {
|
||||
defer conn.Close()
|
||||
}
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
result, err = conn.ExecContext(ctx, stmt, args...)
|
||||
return
|
||||
}
|
||||
|
||||
func (md *MysqlDB) fillConnStr() string {
|
||||
dbServerInfoStr := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s",
|
||||
md.UserName, md.Passwd, md.IP, md.Port, md.DBName)
|
||||
if md.ConnectTimeout > 0 {
|
||||
dbServerInfoStr = fmt.Sprintf("%s?timeout=%ds&readTimeout=%ds&writeTimeout=%ds",
|
||||
dbServerInfoStr, md.ConnectTimeout, md.ConnectTimeout, md.ConnectTimeout)
|
||||
}
|
||||
|
||||
return dbServerInfoStr
|
||||
}
|
105
vendor/github.com/zr-hebo/util-db/query_pooled_db.go
generated
vendored
Normal file
105
vendor/github.com/zr-hebo/util-db/query_pooled_db.go
generated
vendored
Normal file
@@ -0,0 +1,105 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
|
||||
// PooledMysqlDB Mysql主机实例
|
||||
type PooledMysqlDB struct {
|
||||
MysqlDB
|
||||
conn *sql.DB
|
||||
lock *sync.Mutex
|
||||
}
|
||||
|
||||
// NewPooledMysqlDBWithParam 带参数创建MySQL数据库
|
||||
func NewPooledMysqlDBWithParam(
|
||||
ip string, port int, userName, passwd string) (
|
||||
pmd *PooledMysqlDB) {
|
||||
pmd = NewPooledMysqlDB()
|
||||
pmd.IP = ip
|
||||
pmd.Port = port
|
||||
pmd.UserName = userName
|
||||
pmd.Passwd = passwd
|
||||
pmd.DatabaseType = dbTypeMysql
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// NewPooledMysqlDBWithAllParam 带参数创建MySQL数据库
|
||||
func NewPooledMysqlDBWithAllParam(
|
||||
ip string, port int, userName, passwd, dbName string) (
|
||||
pmd *PooledMysqlDB) {
|
||||
pmd = NewPooledMysqlDB()
|
||||
pmd.IP = ip
|
||||
pmd.Port = port
|
||||
pmd.UserName = userName
|
||||
pmd.Passwd = passwd
|
||||
pmd.DBName = dbName
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// NewPooledMysqlDB 创建MySQL数据库
|
||||
func NewPooledMysqlDB() (pmd *PooledMysqlDB) {
|
||||
pmd = new(PooledMysqlDB)
|
||||
pmd.DatabaseType = dbTypeMysql
|
||||
pmd.lock = new(sync.Mutex)
|
||||
return
|
||||
}
|
||||
|
||||
// CloseConnection 获取数据库连接
|
||||
func (pmd *PooledMysqlDB) CloseConnection() (err error) {
|
||||
if pmd.conn == nil {
|
||||
return
|
||||
}
|
||||
|
||||
err = pmd.conn.Close()
|
||||
return
|
||||
}
|
||||
|
||||
// GetConnection 获取数据库连接
|
||||
func (pmd *PooledMysqlDB) GetConnection() (conn *sql.DB, err error) {
|
||||
pmd.lock.Lock()
|
||||
defer func() {
|
||||
pmd.lock.Unlock()
|
||||
}()
|
||||
|
||||
if pmd.conn != nil {
|
||||
conn = pmd.conn
|
||||
return
|
||||
}
|
||||
|
||||
conn, err = pmd.MysqlDB.getConnection()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
conn.SetConnMaxLifetime(time.Second * 60 * 30)
|
||||
conn.SetMaxOpenConns(0)
|
||||
if err := conn.Ping(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
pmd.conn = conn
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// ExecChange 执行MySQL Query语句
|
||||
func (pmd *PooledMysqlDB) ExecChange(stmt string, args ...interface{}) (
|
||||
result sql.Result, err error) {
|
||||
return pmd.MysqlDB.ExecChange(stmt, args...)
|
||||
}
|
||||
|
||||
// QueryRows 执行MySQL Query语句
|
||||
func (pmd *PooledMysqlDB) QueryRows(stmt string) (queryRows *QueryRows, err error) {
|
||||
return pmd.MysqlDB.QueryRows(stmt)
|
||||
|
||||
}
|
||||
|
||||
// QueryRow 执行MySQL Query语句
|
||||
func (pmd *PooledMysqlDB) QueryRow(stmt string) (row *QueryRow, err error) {
|
||||
return pmd.MysqlDB.QueryRow(stmt)
|
||||
}
|
54
vendor/github.com/zr-hebo/util-db/scanner.go
generated
vendored
Normal file
54
vendor/github.com/zr-hebo/util-db/scanner.go
generated
vendored
Normal file
@@ -0,0 +1,54 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// CheckPair CheckPair
|
||||
type CheckPair struct {
|
||||
Host
|
||||
UserErp string `json:"erp"`
|
||||
DBName string `json:"dbname"`
|
||||
}
|
||||
|
||||
func (cp *CheckPair) String() string {
|
||||
return fmt.Sprintf(
|
||||
"check if %s can visit %s@%s:%d", cp.UserErp, cp.DBName, cp.IP, cp.Port)
|
||||
}
|
||||
|
||||
// Scanner SQL rows读取器
|
||||
type Scanner interface {
|
||||
Scan(*sql.Rows, ...*string) error
|
||||
}
|
||||
|
||||
type showNullScanner struct {
|
||||
}
|
||||
|
||||
// NewShowNullScanner 显示NULL值的接收器,要求Scan的时候传入的参数是string类型
|
||||
func NewShowNullScanner() (s Scanner) {
|
||||
return new(showNullScanner)
|
||||
}
|
||||
|
||||
func (ns *showNullScanner) Scan(
|
||||
rows *sql.Rows, receivers ...*string) (err error) {
|
||||
nullReceivers := make([]interface{}, 0, len(receivers))
|
||||
for range receivers {
|
||||
nullReceivers = append(nullReceivers, &sql.NullString{})
|
||||
}
|
||||
|
||||
if err = rows.Scan(nullReceivers...); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
for i, rv := range nullReceivers {
|
||||
nullReceiver := rv.(*sql.NullString)
|
||||
if nullReceiver.Valid {
|
||||
*receivers[i] = nullReceiver.String
|
||||
} else {
|
||||
*receivers[i] = "NULL"
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
Reference in New Issue
Block a user