package main
import (
"database/sql"
"errors"
"fmt"
"net"
"sync/atomic"
"time"
"github.com/go-sql-driver/MysqL"
)
func main() {}
//新建MysqL连接,connstr为MysqL连接串"root:123456@tcp(172.18.80.211:3306)/workstacks?charset=utf8"
//maxopen为最大连接数
func NewMysqL(connstr string,maxopen int) (*_MysqL,error) {
m := &_MysqL{connstr: connstr,maxopen: maxopen}
err := m.dial()
return m,err
}
var UnavailableError = errors.New("Status Unavailable")
type _MysqL struct {
connstr string
maxopen int
db *sql.DB
status uint32
}
//初始化连接串
func (s *_MysqL) init() error {
db,err := sql.Open("MysqL",s.connstr)
if err != nil {
return err
}
if s.maxopen <= 0 {
db.SetMaxOpenConns(20)
} else {
db.SetMaxOpenConns(s.maxopen)
}
s.db = db
return nil
}
//连接数据库,当网络不可达的时候会循环连接,知道连接成功,每次连接等待时间会递增,最大等待时间为120s
func (s *_MysqL) dial() error {
if s.db == nil {
err := s.init()
if err != nil {
return err
}
}
var rtime time.Duration = 10
redial:
err := s.db.Ping()
if err != nil {
if _,ok := err.(*MysqL.MysqLError); ok {
return err
}
if isNetError(err) {
<-time.After(rtime * time.Second)
if rtime < 120 {
rtime += 10
}
fmt.Println(err)
goto redial
}
return err
}
s.SetStatus(1)
return nil
}
//重置数据库连接信息
func (s *_MysqL) Reset(connstr string,maxopen int) error {
db,connstr)
if err != nil {
return err
}
if maxopen <= 0 {
db.SetMaxOpenConns(20)
} else {
db.SetMaxOpenConns(maxopen)
}
err = db.Ping()
if err != nil {
return err
}
s.connstr = connstr
s.maxopen = maxopen
tmpdb := s.db
s.db = db
tmpdb.Close()
return nil
}
//执行sql语句,insert,update,delete等
func (s *_MysqL) Exec(query string,args ...interface{}) (sql.Result,error) {
if s.Status() {
result,err := s.db.Exec(query,args...)
if err != nil {
if isNetError(err) {
s.SetStatus(0)
if err = s.dial(); err == nil {
return s.Exec(query,args...)
}
return nil,err
}
}
return result,err
}
return nil,UnavailableError
}
//执行SQL查询
func (s *_MysqL) Query(query string,args ...interface{}) (*sql.Rows,error) {
if s.Status() {
rows,err := s.db.Query(query,args...)
if err != nil {
if isNetError(err) {
s.SetStatus(0)
if err = s.dial(); err == nil {
return s.Query(query,err
}
}
return rows,UnavailableError
}
//查看当前状态是否可用,返回true表示可用
func (s *_MysqL) Status() bool {
return atomic.LoadUint32(&s.status) != 0
}
//设置状态,0表示状态不可用,1表示状态可用
func (s *_MysqL) SetStatus(status uint32) {
atomic.StoreUint32(&s.status,status)
}
//关闭数据库连接
func (s *_MysqL) Close() {
if s.db != nil {
s.db.Close()
}
}
func isNetError(err error) bool {
_,ok := err.(*net.OpError)
return ok
}
原文地址:https://www.jb51.cc/go/188955.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。