slowquery_test.go

package querydigest

import (
	"bytes"
	"os"
	"testing"

	"github.com/google/go-cmp/cmp"
)

func TestSlowQueryScanner_Next(t *testing.T) {

	// test fixtures created by using https://github.com/isucon/isucon9-qualify application.
	cases := []struct {
		name         string
		fixturesPath string
		expect       SlowQueryInfo
	}{
		{
			name:         "header",
			fixturesPath: "header",
			expect: SlowQueryInfo{
				RawQuery: bytes.NewBufferString("select @@version_comment limit 1;").Bytes(),
				QueryTime: QueryTime{
					QueryTime:    0.000126,
					LockTime:     0,
					RowsSent:     1,
					RowsExamined: 0,
				},
			},
		},
		{
			name:         "insert",
			fixturesPath: "insert",
			expect: SlowQueryInfo{
				RawQuery: bytes.NewBufferString("INSERT INTO categories (`id`,`parent_id`,`category_name`) VALUES" +
					"(1,0,\"ソファー\")," +
					"(2,1,\"一人掛けソファー\")," +
					"(3,1,\"二人掛けソファー\")," +
					"(4,1,\"コーナーソファー\");").Bytes(),
				QueryTime: QueryTime{
					QueryTime:    0.012964,
					LockTime:     0.001197,
					RowsSent:     0,
					RowsExamined: 0,
				},
			},
		},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			f, err := os.Open("./testdata/mysql-slow." + c.fixturesPath + ".log")
			if err != nil {
				t.Fatal(err)
			}
			defer f.Close()

			scanner := NewSlowQueryScanner(f)

			scanner.Next()

			if scanner.Err() != nil {
				t.Fatal(scanner.Err())
			}

			info := scanner.SlowQueryInfo()

			if info == nil {
				t.Fatal("info is nil")
			}

			if diff := cmp.Diff(*info, c.expect); diff != "" {
				t.Errorf("diff: %s", diff)
			}
		})
	}
}

func Test_parseHeader(t *testing.T) {

	src := `# Query_time: 0.004370  Lock_time: 0.001289 Rows_sent: 2  Rows_examined: 2`

	queryTime, lockTime, rowsSent, rowsExamined := parseHeader(src)

	if queryTime != "0.004370" {
		t.Errorf("expect: `%s` but `%s`", "0.004370", queryTime)
	}
	if lockTime != "0.001289" {
		t.Errorf("expect: `%s` but `%s`", "0.001289", lockTime)
	}
	if rowsSent != "2" {
		t.Errorf("expect: `%s` but `%s`", "2", rowsSent)
	}
	if rowsExamined != "2" {
		t.Errorf("expect: `%s` but `%s`", "2", rowsExamined)
	}

}

func BenchmarkSlowQueryScanner_SlowQueryInfo(b *testing.B) {
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		f, err := os.Open("./benchdata/mysql-slow.log")
		if err != nil {
			b.Fatal(err)
		}
		sc := NewSlowQueryScanner(f)

		for sc.Next() {
		}

		if err := sc.Err(); err != nil {
			b.Fatal(err)
		}

		f.Close()
	}
}