Skip to content

Commit 4a9e2b1

Browse files
committed
Make NewConn and NewCustomizedConn into methods
1 parent d2d72ea commit 4a9e2b1

File tree

2 files changed

+19
-6
lines changed

2 files changed

+19
-6
lines changed

cmd/go-mysqlserver/main.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ func main() {
2626

2727
// Create a connection with user root and an empty password.
2828
// You can use your own handler to handle command here.
29-
conn, err := server.NewConn(c, "root", "", server.EmptyHandler{})
29+
srv := server.NewDefaultServer()
30+
conn, err := srv.NewConn(c, "root", "", server.EmptyHandler{})
3031
if err != nil {
3132
log.Fatal(err)
3233
}

server/conn.go

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,26 +39,38 @@ type Conn struct {
3939
var baseConnID uint32 = 10000
4040

4141
// NewConn: create connection with default server settings
42+
// Deprecated: Use Server.NewConn instead.
4243
func NewConn(conn net.Conn, user string, password string, h Handler) (*Conn, error) {
44+
s := NewDefaultServer()
45+
return s.NewConn(conn, user, password, h)
46+
}
47+
48+
// NewCustomizedConn: create connection with customized server settings
49+
// Deprecated: Use Server.NewConn instead.
50+
func NewCustomizedConn(conn net.Conn, serverConf *Server, p CredentialProvider, h Handler) (*Conn, error) {
51+
return serverConf.NewCustomizedConn(conn, p, h)
52+
}
53+
54+
// NewConn: create connection with default server settings
55+
func (s *Server) NewConn(conn net.Conn, user string, password string, h Handler) (*Conn, error) {
4356
p := NewInMemoryProvider()
4457
p.AddUser(user, password)
45-
server := NewDefaultServer()
4658

47-
return NewCustomizedConn(conn, server, p, h)
59+
return s.NewCustomizedConn(conn, p, h)
4860
}
4961

5062
// NewCustomizedConn: create connection with customized server settings
51-
func NewCustomizedConn(conn net.Conn, serverConf *Server, p CredentialProvider, h Handler) (*Conn, error) {
63+
func (s *Server) NewCustomizedConn(conn net.Conn, p CredentialProvider, h Handler) (*Conn, error) {
5264
var packetConn *packet.Conn
53-
if serverConf.tlsConfig != nil {
65+
if s.tlsConfig != nil {
5466
packetConn = packet.NewTLSConn(conn)
5567
} else {
5668
packetConn = packet.NewConn(conn)
5769
}
5870

5971
c := &Conn{
6072
Conn: packetConn,
61-
serverConf: serverConf,
73+
serverConf: s,
6274
credentialProvider: p,
6375
h: h,
6476
connectionID: atomic.AddUint32(&baseConnID, 1),

0 commit comments

Comments
 (0)