req.go

package client

import (
	"github.com/go-mysql-org/go-mysql/utils"
)

func (c *Conn) writeCommand(command byte) error {
	c.ResetSequence()

	return c.WritePacket([]byte{
		0x01, // 1 bytes long
		0x00,
		0x00,
		0x00, // sequence
		command,
	})
}

func (c *Conn) writeCommandBuf(command byte, arg []byte) error {
	c.ResetSequence()

	length := len(arg) + 1
	data := utils.ByteSliceGet(length + 4)
	data.B[4] = command

	copy(data.B[5:], arg)

	err := c.WritePacket(data.B)

	utils.ByteSlicePut(data)

	return err
}

func (c *Conn) writeCommandStr(command byte, arg string) error {
	return c.writeCommandBuf(command, utils.StringToByteSlice(arg))
}

func (c *Conn) writeCommandUint32(command byte, arg uint32) error {
	c.ResetSequence()

	buf := utils.ByteSliceGet(9)

	buf.B[0] = 0x05 // 5 bytes long
	buf.B[1] = 0x00
	buf.B[2] = 0x00
	buf.B[3] = 0x00 // sequence

	buf.B[4] = command

	buf.B[5] = byte(arg)
	buf.B[6] = byte(arg >> 8)
	buf.B[7] = byte(arg >> 16)
	buf.B[8] = byte(arg >> 24)

	err := c.WritePacket(buf.B)
	utils.ByteSlicePut(buf)
	return err
}

func (c *Conn) writeCommandStrStr(command byte, arg1 string, arg2 string) error {
	c.ResetSequence()

	data := make([]byte, 4, 6+len(arg1)+len(arg2))

	data = append(data, command)
	data = append(data, arg1...)
	data = append(data, 0)
	data = append(data, arg2...)

	return c.WritePacket(data)
}