Golang1.7.4標(biāo)準(zhǔn)庫sql使用

2019-11-14 10:26:32

字體：大中小

供稿：網(wǎng)友

package mainimport ( "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"http://maxopen為最大連接數(shù)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}//連接數(shù)據(jù)庫,當(dāng)網(wǎng)絡(luò)不可達(dá)的時(shí)候會(huì)循環(huán)連接,知道連接成功,每次連接等待時(shí)間會(huì)遞增,最大等待時(shí)間為120sfunc (s *_mysql) dial() error { if s.db == nil { err := s.init() if err != nil { return err } } var rtime time.Duration = 10redial: 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}//重置數(shù)據(jù)庫連接信息func (s *_mysql) Reset(connstr string, maxopen int) error { db, err := sql.Open("mysql", 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}//執(zhí)行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}//執(zhí)行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, args...) } return nil, err } } return rows, err } return nil, UnavailableError}//查看當(dāng)前狀態(tài)是否可用,返回true表示可用func (s *_mysql) Status() bool { return atomic.LoadUint32(&s.status) != 0}//設(shè)置狀態(tài),0表示狀態(tài)不可用,1表示狀態(tài)可用func (s *_mysql) SetStatus(status uint32) { atomic.StoreUint32(&s.status, status)}//關(guān)閉數(shù)據(jù)庫連接func (s *_mysql) Close() { if s.db != nil { s.db.Close() }}func isNetError(err error) bool { _, ok := err.(*net.OpError) return ok}

上一篇：ACM簡(jiǎn)單計(jì)算題-棋盤上的距離

下一篇：2016年藍(lán)橋杯省賽A組C語言（星系爆炸）計(jì)算日期