Initial Commit

Based on Go-MySQL-Driver

Author: julienschmidt

LICENSE

@@ -0,0 +1,83 @@
+// Copyright 2013 Julien Schmidt. All rights reserved.
+// http://www.julienschmidt.com
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this file,
+// You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package sphinxql
+
+import (
+	"io"
+)
+
+const (
+	defaultBufSize = 4096
+)
+
+type buffer struct {
+	buf    []byte
+	rd     io.Reader
+	idx    int
+	length int
+}
+
+func newBuffer(rd io.Reader) *buffer {
+	return &buffer{
+		buf: make([]byte, defaultBufSize),
+		rd:  rd,
+	}
+}
+
+// fill reads at least _need_ bytes in the buffer
+// existing data in the buffer gets lost
+func (b *buffer) fill(need int) (err error) {
+	b.idx = 0
+	b.length = 0
+
+	n := 0
+	for b.length < need {
+		n, err = b.rd.Read(b.buf[b.length:])
+		b.length += n
+
+		if err == nil {
+			continue
+		}
+		return // err
+	}
+
+	return
+}
+
+// read len(p) bytes
+func (b *buffer) read(p []byte) (err error) {
+	need := len(p)
+
+	if b.length < need {
+		if b.length > 0 {
+			copy(p[0:b.length], b.buf[b.idx:])
+			need -= b.length
+			p = p[b.length:]
+
+			b.idx = 0
+			b.length = 0
+		}
+
+		if need >= len(b.buf) {
+			var n int
+			has := 0
+			for err == nil && need > has {
+				n, err = b.rd.Read(p[has:])
+				has += n
+			}
+			return
+		}
+
+		err = b.fill(need) // err deferred
+	}
+
+	copy(p, b.buf[b.idx:])
+	b.idx += need
+	b.length -= need
+	return
+}

buffer.go

@@ -0,0 +1,160 @@
+// Copyright 2013 Julien Schmidt. All rights reserved.
+// http://www.julienschmidt.com
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this file,
+// You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package sphinxql
+
+import (
+	"database/sql/driver"
+	"errors"
+	"net"
+	"strings"
+)
+
+type sphinxqlConn struct {
+	cfg          *config
+	flags        ClientFlag
+	charset      byte
+	cipher       []byte
+	netConn      net.Conn
+	buf          *buffer
+	protocol     uint8
+	sequence     uint8
+	affectedRows uint64
+	insertId     uint64
+}
+
+type config struct {
+	user   string
+	passwd string
+	net    string
+	addr   string
+	dbname string
+	params map[string]string
+}
+
+// Handles parameters set in DSN
+func (mc *sphinxqlConn) handleParams() (err error) {
+	for param, val := range mc.cfg.params {
+		switch param {
+		// Charset
+		case "charset":
+			charsets := strings.Split(val, ",")
+			for _, charset := range charsets {
+				err = mc.exec("SET NAMES " + charset)
+				if err != nil {
+					return
+				}
+			}
+
+		// Timeout - already handled on connecting
+		case "timeout":
+			continue
+
+		// System Vars
+		default:
+			err = mc.exec("SET " + param + "=" + val + "")
+			if err != nil {
+				return
+			}
+		}
+	}
+
+	return
+}
+
+func (mc *sphinxqlConn) Begin() (driver.Tx, error) {
+	err := mc.exec("START TRANSACTION")
+	if err == nil {
+		return &sphinxqlTx{mc}, err
+	}
+
+	return nil, err
+}
+
+func (mc *sphinxqlConn) Close() (err error) {
+	mc.writeCommandPacket(COM_QUIT)
+	mc.cfg = nil
+	mc.buf = nil
+	mc.netConn.Close()
+	mc.netConn = nil
+	return
+}
+
+func (mc *sphinxqlConn) Prepare(query string) (driver.Stmt, error) {
+	return &sphinxqlStmt{
+		mc:    mc,
+		query: query,
+	}, nil
+}
+
+func (mc *sphinxqlConn) Exec(query string, args []driver.Value) (_ driver.Result, err error) {
+	if len(args) == 0 {
+		mc.affectedRows = 0
+		mc.insertId = 0
+
+		err = mc.exec(query)
+		if err == nil {
+			return &sphinxqlResult{
+				affectedRows: int64(mc.affectedRows),
+				insertId:     int64(mc.insertId),
+			}, err
+		} else {
+			return nil, err
+		}
+
+	}
+	return nil, errors.New("args not supported")
+
+}
+
+// Internal function to execute commands
+func (mc *sphinxqlConn) exec(query string) (err error) {
+	// Send command
+	err = mc.writeCommandPacketStr(COM_QUERY, query)
+	if err != nil {
+		return
+	}
+
+	// Read Result
+	var resLen int
+	resLen, err = mc.readResultSetHeaderPacket()
+	if err == nil && resLen > 0 {
+		err = mc.readUntilEOF()
+		if err != nil {
+			return
+		}
+
+		err = mc.readUntilEOF()
+	}
+
+	return
+}
+
+func (mc *sphinxqlConn) Query(query string, args []driver.Value) (_ driver.Rows, err error) {
+	if len(args) == 0 {
+		var rows *sphinxqlRows
+		// Send command
+		err = mc.writeCommandPacketStr(COM_QUERY, query)
+		if err == nil {
+			// Read Result
+			var resLen int
+			resLen, err = mc.readResultSetHeaderPacket()
+			if err == nil {
+				rows = &sphinxqlRows{mc, nil, false}
+
+				if resLen > 0 {
+					// Columns
+					rows.columns, err = mc.readColumns(resLen)
+				}
+				return rows, err
+			}
+		}
+		return nil, err
+	}
+
+	return nil, errors.New("args not supported")
+}

connection.go

@@ -0,0 +1,126 @@
+// Copyright 2013 Julien Schmidt. All rights reserved.
+// http://www.julienschmidt.com
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this file,
+// You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package sphinxql
+
+const (
+	MIN_PROTOCOL_VERSION byte = 10
+	//MAX_PACKET_SIZE      = 1<<24 - 1
+	TIME_FORMAT = "2006-01-02 15:04:05"
+)
+
+// MySQL constants documentation:
+// http://dev.mysql.com/doc/internals/en/client-server-protocol.html
+
+type ClientFlag uint32
+
+const (
+	CLIENT_LONG_PASSWORD ClientFlag = 1 << iota
+	CLIENT_FOUND_ROWS
+	CLIENT_LONG_FLAG
+	CLIENT_CONNECT_WITH_DB
+	CLIENT_NO_SCHEMA
+	CLIENT_COMPRESS
+	CLIENT_ODBC
+	CLIENT_LOCAL_FILES
+	CLIENT_IGNORE_SPACE
+	CLIENT_PROTOCOL_41
+	CLIENT_INTERACTIVE
+	CLIENT_SSL
+	CLIENT_IGNORE_SIGPIPE
+	CLIENT_TRANSACTIONS
+	CLIENT_RESERVED
+	CLIENT_SECURE_CONN
+	CLIENT_MULTI_STATEMENTS
+	CLIENT_MULTI_RESULTS
+)
+
+type commandType byte
+
+const (
+	COM_QUIT commandType = iota + 1
+	COM_INIT_DB
+	COM_QUERY
+	COM_FIELD_LIST
+	COM_CREATE_DB
+	COM_DROP_DB
+	COM_REFRESH
+	COM_SHUTDOWN
+	COM_STATISTICS
+	COM_PROCESS_INFO
+	COM_CONNECT
+	COM_PROCESS_KILL
+	COM_DEBUG
+	COM_PING
+	COM_TIME
+	COM_DELAYED_INSERT
+	COM_CHANGE_USER
+	COM_BINLOG_DUMP
+	COM_TABLE_DUMP
+	COM_CONNECT_OUT
+	COM_REGISTER_SLAVE
+	COM_STMT_PREPARE
+	COM_STMT_EXECUTE
+	COM_STMT_SEND_LONG_DATA
+	COM_STMT_CLOSE
+	COM_STMT_RESET
+	COM_SET_OPTION
+	COM_STMT_FETCH
+)
+
+const (
+	FIELD_TYPE_DECIMAL byte = iota
+	FIELD_TYPE_TINY
+	FIELD_TYPE_SHORT
+	FIELD_TYPE_LONG
+	FIELD_TYPE_FLOAT
+	FIELD_TYPE_DOUBLE
+	FIELD_TYPE_NULL
+	FIELD_TYPE_TIMESTAMP
+	FIELD_TYPE_LONGLONG
+	FIELD_TYPE_INT24
+	FIELD_TYPE_DATE
+	FIELD_TYPE_TIME
+	FIELD_TYPE_DATETIME
+	FIELD_TYPE_YEAR
+	FIELD_TYPE_NEWDATE
+	FIELD_TYPE_VARCHAR
+	FIELD_TYPE_BIT
+)
+const (
+	FIELD_TYPE_NEWDECIMAL byte = iota + 0xf6
+	FIELD_TYPE_ENUM
+	FIELD_TYPE_SET
+	FIELD_TYPE_TINY_BLOB
+	FIELD_TYPE_MEDIUM_BLOB
+	FIELD_TYPE_LONG_BLOB
+	FIELD_TYPE_BLOB
+	FIELD_TYPE_VAR_STRING
+	FIELD_TYPE_STRING
+	FIELD_TYPE_GEOMETRY
+)
+
+type FieldFlag uint16
+
+const (
+	FLAG_NOT_NULL FieldFlag = 1 << iota
+	FLAG_PRI_KEY
+	FLAG_UNIQUE_KEY
+	FLAG_MULTIPLE_KEY
+	FLAG_BLOB
+	FLAG_UNSIGNED
+	FLAG_ZEROFILL
+	FLAG_BINARY
+	FLAG_ENUM
+	FLAG_AUTO_INCREMENT
+	FLAG_TIMESTAMP
+	FLAG_SET
+	FLAG_UNKNOWN_1
+	FLAG_UNKNOWN_2
+	FLAG_UNKNOWN_3
+	FLAG_UNKNOWN_4
+)