package main

import (
	"fmt"
	"git.sxidc.com/go-tools/utils/strutils"
	"git.sxidc.com/service-supports/ds-sdk/sdk"
	"git.sxidc.com/service-supports/ds-sdk/sql"
	"git.sxidc.com/service-supports/ds-sdk/sql/sql_tpl"
	"github.com/iancoleman/strcase"
	"math/rand"
	"strings"
	"sync"
	"testing"
	"time"
)

type IDField struct {
	ID string
}

type TimeFields struct {
	CreatedTime     *time.Time
	LastUpdatedTime time.Time
}

type GraduatedTimeTestStruct struct {
	Field *string `sqlmapping:"-" sqlresult:"column:graduated_time;parseTime:2006-01-02 15:04:05"`
}

type Class struct {
	IDField
	Name          string `sqlmapping:"updateClear;aes:@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L;" sqlresult:"aes:@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L;"`
	StudentNum    int    `sqlmapping:"column:student_num;notUpdate;" sqlresult:"column:student_num_alias"`
	GraduatedTime *time.Time
	StudentIDs    []string `sqlmapping:"column:student_ids;joinWith:'\n'" sqlresult:"column:student_ids;splitWith:'\n'"`
	TimeFields
	Ignored string `sqlmapping:"-" sqlresult:"-"`
	*GraduatedTimeTestStruct
}

const (
	token          = "IpTTwAQweh/BP51fz5CzWKQFaXHvZe6ewvk6yOcAOkU="
	address        = "localhost"
	httpPort       = "10000"
	grpcPort       = "10001"
	namespace      = "ns-sdk-demo"
	dataSource     = "ds-sdk-demo"
	deleteSql      = "delete-sdk-demo"
	goRoutineCount = 100
	tableName      = "test.classes"
)

var (
	sqlSpec = sdk.SqlSpec{
		Clauses: "- DELETE FROM {{ .table_name }} WHERE id = '{{ .id }}'",
	}
)

const (
	sqlResultTimeMicroFormat = "2006-01-02T15:04:05.000000+08:00"
	sqlResultTimeMilliFormat = "2006-01-02T15:04:05.000+08:00"
	sqlResultTimeSecFormat   = "2006-01-02T15:04:05+08:00"
)

func chooseTimeLayout(timeStr string) string {
	if strings.HasSuffix(timeStr, ".000000+08:00") {
		return sqlResultTimeMicroFormat
	} else if strings.HasSuffix(timeStr, ".000+08:00") {
		return sqlResultTimeMilliFormat
	} else {
		return sqlResultTimeSecFormat
	}
}

func TestBasic(t *testing.T) {
	classID := strutils.SimpleUUID()
	className := strutils.SimpleUUID()
	studentNum := rand.Int31n(100)
	studentIDs := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}
	now := time.Now()

	insertExecuteParams, err := sql_tpl.InsertExecuteParams{
		TableName: tableName,
		TableRow: sql_tpl.NewTableRow().Add("id", classID).
			Add("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Add("student_num", studentNum).
			Add("student_ids", strings.Join(studentIDs, "\n")).
			Add("graduated_time", now).
			Add("created_time", now).
			Add("last_updated_time", now),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	deleteExecuteParams := map[string]any{
		"table_name": tableName,
		"id":         classID,
	}

	err = sdk.InitInstance(token, address, httpPort, grpcPort, namespace, dataSource)
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		err := sdk.DestroyInstance()
		if err != nil {
			t.Fatal(err)
		}
	}()

	err = sdk.GetInstance().CreateSQL(deleteSql, sqlSpec.ToMap())
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		err = sdk.GetInstance().DeleteSQL(deleteSql)
		if err != nil {
			t.Fatal(err)
		}
	}()

	_, err = sdk.GetInstance().ExecuteRawSql(sql_tpl.InsertTpl, insertExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	_, err = sdk.GetInstance().ExecuteSql(deleteSql, deleteExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	wg := sync.WaitGroup{}
	wg.Add(goRoutineCount)

	start := time.Now()

	for i := 0; i < goRoutineCount; i++ {
		go func() {
			defer wg.Done()

			err = sdk.GetInstance().Transaction(func(tx *sdk.Transaction) error {
				_, err := tx.ExecuteRawSql(sql_tpl.InsertTpl, insertExecuteParams)
				if err != nil {
					return err
				}

				_, err = tx.ExecuteSql(deleteSql, deleteExecuteParams)
				if err != nil {
					return err
				}

				return nil
			})
			if err != nil {
				panic(err)
			}
		}()
	}

	wg.Wait()

	end := time.Now()

	fmt.Println(end.Sub(start).Milliseconds())
}

func TestRawSqlTemplate(t *testing.T) {
	classID := strutils.SimpleUUID()
	className := strutils.SimpleUUID()
	studentNum := rand.Int31n(100)
	studentIDs := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}
	newClassName := strutils.SimpleUUID()
	newStudentNum := rand.Int31n(100)
	newStudentIDs := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}

	now := time.Now()

	insertExecuteParams, err := sql_tpl.InsertExecuteParams{
		TableName: tableName,
		TableRow: sql_tpl.NewTableRow().Add("id", classID).
			Add("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Add("student_num", studentNum).
			Add("student_ids", strings.Join(studentIDs, "\n")).
			Add("graduated_time", now).
			Add("created_time", now).
			Add("last_updated_time", now),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	deleteExecuteParams, err := sql_tpl.DeleteExecuteParams{
		TableName:  tableName,
		Conditions: sql_tpl.NewConditions().Equal("id", classID),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	updateExecuteParams, err := sql_tpl.UpdateExecuteParams{
		TableName: tableName,
		TableRow: sql_tpl.NewTableRow().
			Add("name", newClassName, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Add("student_num", newStudentNum).
			Add("student_ids", strings.Join(newStudentIDs, "\n")),
		Conditions: sql_tpl.NewConditions().Equal("id", classID),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	queryExecuteParams, err := sql_tpl.QueryExecuteParams{
		TableName:     tableName,
		SelectColumns: []string{"id", "name", "student_num as student_num_alias", "student_ids", "graduated_time", "created_time", "last_updated_time"},
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
		PageNo:   1,
		PageSize: 1,
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	newQueryExecuteParams, err := sql_tpl.QueryExecuteParams{
		TableName:     tableName,
		SelectColumns: []string{"id", "name", "student_num as student_num_alias", "student_ids", "graduated_time", "created_time", "last_updated_time"},
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", newClassName, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", newStudentNum),
		PageNo:   0,
		PageSize: 0,
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	countExecuteParams, err := sql_tpl.CountExecuteParams{
		TableName: tableName,
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	newCountExecuteParams, err := sql_tpl.CountExecuteParams{
		TableName: tableName,
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", newClassName, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", newStudentNum),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	err = sdk.InitInstance(token, address, httpPort, grpcPort, namespace, dataSource)
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		err := sdk.DestroyInstance()
		if err != nil {
			t.Fatal(err)
		}
	}()

	_, err = sdk.GetInstance().ExecuteRawSql(sql_tpl.InsertTpl, insertExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	queryResults, err := sdk.GetInstance().ExecuteRawSql(sql_tpl.QueryTpl, queryExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	countResults, err := sdk.GetInstance().ExecuteRawSql(sql_tpl.CountTpl, countExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	classes, err := sql.ParseSqlResult[[]Class](queryResults)
	if err != nil {
		t.Fatal(err)
	}

	if float64(len(classes)) != countResults[0]["count"].(float64) {
		t.Fatal("总数不正确")
	}

	graduatedTimeLayout := chooseTimeLayout(queryResults[0]["graduated_time"].(string))
	createdTimeLayout := chooseTimeLayout(queryResults[0]["created_time"].(string))
	lastUpdatedTimeLayout := chooseTimeLayout(queryResults[0]["last_updated_time"].(string))

	if classes[0].ID != classID ||
		classes[0].Name != className ||
		classes[0].StudentNum != int(studentNum) ||
		strings.Join(classes[0].StudentIDs, "\n") != strings.Join(studentIDs, "\n") ||
		classes[0].GraduatedTime.Format(graduatedTimeLayout) != now.Format(graduatedTimeLayout) ||
		classes[0].CreatedTime.Format(createdTimeLayout) != now.Format(createdTimeLayout) ||
		classes[0].LastUpdatedTime.Format(lastUpdatedTimeLayout) != now.Format(lastUpdatedTimeLayout) {
		t.Fatal("查询数据不正确")
	}

	_, err = sdk.GetInstance().ExecuteRawSql(sql_tpl.UpdateTpl, updateExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	queryResults, err = sdk.GetInstance().ExecuteRawSql(sql_tpl.QueryTpl, newQueryExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	countResults, err = sdk.GetInstance().ExecuteRawSql(sql_tpl.CountTpl, newCountExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	classes, err = sql.ParseSqlResult[[]Class](queryResults)
	if err != nil {
		t.Fatal(err)
	}

	if float64(len(classes)) != countResults[0]["count"].(float64) {
		t.Fatal("总数不正确")
	}

	graduatedTimeLayout = chooseTimeLayout(queryResults[0]["graduated_time"].(string))
	createdTimeLayout = chooseTimeLayout(queryResults[0]["created_time"].(string))
	lastUpdatedTimeLayout = chooseTimeLayout(queryResults[0]["last_updated_time"].(string))

	if classes[0].ID != classID ||
		classes[0].Name != newClassName ||
		classes[0].StudentNum != int(newStudentNum) ||
		strings.Join(classes[0].StudentIDs, "\n") != strings.Join(newStudentIDs, "\n") ||
		classes[0].GraduatedTime.Format(graduatedTimeLayout) != now.Format(graduatedTimeLayout) ||
		classes[0].CreatedTime.Format(createdTimeLayout) != now.Format(createdTimeLayout) ||
		classes[0].LastUpdatedTime.Format(lastUpdatedTimeLayout) != now.Format(lastUpdatedTimeLayout) {
		t.Fatal("查询数据不正确")
	}

	_, err = sdk.GetInstance().ExecuteRawSql(sql_tpl.DeleteTpl, deleteExecuteParams)
	if err != nil {
		t.Fatal(err)
	}
}

func TestSqlMapping(t *testing.T) {
	sqlMapping, err := sql.ParseSqlMappingTag(&Class{})
	if err != nil {
		t.Fatal(err)
	}

	checkSqlMapping(t, sqlMapping)
}

func checkSqlMapping(t *testing.T, sqlMapping *sql.Mapping) {
	for fieldName, mappingElement := range sqlMapping.MappingElement {
		switch element := mappingElement.(type) {
		case *sql.Mapping:
			checkSqlMapping(t, element)
		case *sql.MappingColumn:
			if fieldName != "ID" && fieldName != "Name" &&
				fieldName != "StudentNum" && fieldName != "StudentIDs" &&
				fieldName != "GraduatedTime" && fieldName != "CreatedTime" &&
				fieldName != "LastUpdatedTime" {
				t.Fatal("字段名不正确")
			}

			if element.Name != "id" && element.Name != "name" &&
				element.Name != "student_num" && element.Name != "student_ids" &&
				element.Name != "graduated_time" && element.Name != "created_time" &&
				element.Name != "last_updated_time" {
				t.Fatal("列名不正确")
			}

			if element.Name != "student_ids" && element.Name != strcase.ToSnake(fieldName) {
				t.Fatal("列名不正确")
			}

			if element.Name == "id" {
				if !element.IsKey || element.CanUpdate || element.CanUpdateClear ||
					strutils.IsStringNotEmpty(element.AESKey) {
					t.Fatal("id字段Tag不正确")
				}
			}

			if element.Name == "name" {
				if element.IsKey || !element.CanUpdate || !element.CanUpdateClear ||
					strutils.IsStringEmpty(element.AESKey) || element.AESKey != "@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L" {
					t.Fatal("name字段Tag不正确")
				}
			}

			if element.Name == "student_num" {
				if element.IsKey || element.CanUpdate || element.CanUpdateClear ||
					strutils.IsStringNotEmpty(element.AESKey) {
					t.Fatal("student_num字段Tag不正确")
				}
			}

			if element.Name == "student_ids" {
				if element.JoinWith != "\n" {
					t.Fatal("student_ids字段Tag不正确")
				}
			}
		default:
			t.Fatal("不支持的元素类型")
		}
	}
}

func TestSqlResult(t *testing.T) {
	sqlResult, err := sql.ParseSqlResultTag(&Class{})
	if err != nil {
		t.Fatal(err)
	}

	checkSqlResult(t, sqlResult)
}

func checkSqlResult(t *testing.T, sqlResult *sql.Result) {
	for fieldName, resultElement := range sqlResult.ResultElement {
		switch element := resultElement.(type) {
		case *sql.Result:
			checkSqlResult(t, element)
		case *sql.ResultColumn:
			if fieldName != "ID" && fieldName != "Name" &&
				fieldName != "StudentNum" && fieldName != "StudentIDs" &&
				fieldName != "GraduatedTime" && fieldName != "CreatedTime" &&
				fieldName != "LastUpdatedTime" && fieldName != "Field" {
				t.Fatal("字段名不正确")
			}

			if element.Name != "id" && element.Name != "name" &&
				element.Name != "student_num_alias" && element.Name != "student_ids" &&
				element.Name != "graduated_time" && element.Name != "created_time" &&
				element.Name != "last_updated_time" && element.Name != "graduated_time_test" {
				t.Fatal("列名不正确")
			}

			if element.Name != "student_num_alias" &&
				element.Name != "graduated_time" &&
				element.Name != "student_ids" &&
				element.Name != strcase.ToSnake(fieldName) {
				t.Fatal("列名不正确")
			}

			if element.Name == "id" {
				if strutils.IsStringNotEmpty(element.ParseTime) ||
					strutils.IsStringNotEmpty(element.AESKey) {
					t.Fatal("id字段Tag不正确")
				}
			}

			if element.Name == "name" {
				if strutils.IsStringNotEmpty(element.ParseTime) ||
					strutils.IsStringEmpty(element.AESKey) ||
					element.AESKey != "@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L" {
					t.Fatal("name字段Tag不正确")
				}
			}

			if element.Name == "student_num" {
				if strutils.IsStringNotEmpty(element.ParseTime) ||
					strutils.IsStringNotEmpty(element.AESKey) {
					t.Fatal("student_num字段Tag不正确")
				}
			}

			if element.Name == "student_ids" {
				if element.SplitWith != "\n" {
					t.Fatal("student_ids字段Tag不正确")
				}
			}

			if element.Name == "graduate_time" {
				if strutils.IsStringEmpty(element.ParseTime) ||
					strutils.IsStringNotEmpty(element.AESKey) {
					t.Fatal("graduate_time字段Tag不正确")
				}
			}
		}
	}
}

func TestSql(t *testing.T) {
	classID := strutils.SimpleUUID()
	className := strutils.SimpleUUID()
	studentNum := rand.Int31n(100)
	studentIDs := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}
	newClassName := strutils.SimpleUUID()
	newStudentNum := rand.Int31n(100)
	newStudentIDs := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}
	now := time.Now()
	newNow := time.Now()

	insertExecuteParams, err := sql_tpl.InsertExecuteParams{
		TableName: tableName,
		TableRow: sql_tpl.NewTableRow().Add("id", classID).
			Add("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Add("student_num", studentNum).
			Add("student_ids", strings.Join(studentIDs, "\n")).
			Add("graduated_time", now).
			Add("created_time", now).
			Add("last_updated_time", now),
	}.Map()
	if err != nil {
		t.Fatal(err)
	}

	deleteExecuteParams := map[string]any{
		"table_name": tableName,
		"id":         classID,
	}

	class := &Class{
		IDField:       IDField{ID: classID},
		Name:          className,
		StudentNum:    int(studentNum),
		StudentIDs:    studentIDs,
		GraduatedTime: &now,
		Ignored:       "",
	}

	newClass := &Class{
		IDField:       IDField{ID: classID},
		Name:          newClassName,
		StudentNum:    int(newStudentNum),
		StudentIDs:    newStudentIDs,
		GraduatedTime: &newNow,
		Ignored:       "",
	}

	err = sdk.InitInstance(token, address, httpPort, grpcPort, namespace, dataSource)
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		err := sdk.DestroyInstance()
		if err != nil {
			t.Fatal(err)
		}
	}()

	err = sdk.GetInstance().CreateSQL(deleteSql, sqlSpec.ToMap())
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		err = sdk.GetInstance().DeleteSQL(deleteSql)
		if err != nil {
			t.Fatal(err)
		}
	}()

	err = sql.InsertEntity(sdk.GetInstance(), tableName, class)
	if err != nil {
		t.Fatal(err)
	}

	err = sql.UpdateEntity(sdk.GetInstance(), tableName, newClass)
	if err != nil {
		t.Fatal(err)
	}

	err = sql.DeleteEntity(sdk.GetInstance(), tableName, class)
	if err != nil {
		t.Fatal(err)
	}

	err = sql.Insert(sdk.GetInstance(), &sql_tpl.InsertExecuteParams{
		TableName: tableName,
		TableRow: sql_tpl.NewTableRow().Add("id", classID).
			Add("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Add("student_num", studentNum).
			Add("student_ids", strings.Join(studentIDs, "\n")).
			Add("graduated_time", now).
			Add("created_time", now).
			Add("last_updated_time", now),
	})
	if err != nil {
		t.Fatal(err)
	}

	err = sql.Update(sdk.GetInstance(), &sql_tpl.UpdateExecuteParams{
		TableName: tableName,
		TableRow: sql_tpl.NewTableRow().Add("id", classID).
			Add("name", newClassName, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Add("student_ids", strings.Join(newStudentIDs, "\n")).
			Add("student_num", newStudentNum),
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID),
	})
	if err != nil {
		t.Fatal(err)
	}

	err = sql.Delete(sdk.GetInstance(), &sql_tpl.DeleteExecuteParams{
		TableName: tableName,
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID),
	})
	if err != nil {
		t.Fatal(err)
	}

	_, err = sql.ExecuteRawSql(sdk.GetInstance(), sql_tpl.InsertTpl, insertExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	_, err = sql.ExecuteSql(sdk.GetInstance(), deleteSql, deleteExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	err = sdk.GetInstance().Transaction(func(tx *sdk.Transaction) error {
		err = sql.InsertEntity(tx, tableName, class)
		if err != nil {
			t.Fatal(err)
		}

		err = sql.UpdateEntity(tx, tableName, newClass)
		if err != nil {
			t.Fatal(err)
		}

		err = sql.DeleteEntity(tx, tableName, class)
		if err != nil {
			t.Fatal(err)
		}

		err = sql.Insert(tx, &sql_tpl.InsertExecuteParams{
			TableName: tableName,
			TableRow: sql_tpl.NewTableRow().Add("id", classID).
				Add("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
				Add("student_num", studentNum).
				Add("student_ids", strings.Join(studentIDs, "\n")).
				Add("graduated_time", now).
				Add("created_time", now).
				Add("last_updated_time", now),
		})
		if err != nil {
			t.Fatal(err)
		}

		err = sql.Update(tx, &sql_tpl.UpdateExecuteParams{
			TableName: tableName,
			TableRow: sql_tpl.NewTableRow().Add("id", classID).
				Add("name", newClassName, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
				Add("student_num", newStudentNum).
				Add("student_ids", strings.Join(newStudentIDs, "\n")),
			Conditions: sql_tpl.NewConditions().
				Equal("id", classID),
		})
		if err != nil {
			t.Fatal(err)
		}

		err = sql.Delete(tx, &sql_tpl.DeleteExecuteParams{
			TableName: tableName,
			Conditions: sql_tpl.NewConditions().
				Equal("id", classID),
		})
		if err != nil {
			t.Fatal(err)
		}

		_, err = sql.ExecuteRawSql(tx, sql_tpl.InsertTpl, insertExecuteParams)
		if err != nil {
			t.Fatal(err)
		}

		_, err = sql.ExecuteSql(tx, deleteSql, deleteExecuteParams)
		if err != nil {
			t.Fatal(err)
		}

		return nil
	})
	if err != nil {
		t.Fatal(err)
	}

	err = sql.InsertEntity(sdk.GetInstance(), tableName, class)
	if err != nil {
		t.Fatal(err)
	}

	tableRows, totalCount, err := sql.Query(sdk.GetInstance(), &sql_tpl.QueryExecuteParams{
		TableName:     tableName,
		SelectColumns: []string{"id", "name", "student_ids"},
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
		PageNo:   0,
		PageSize: 0,
	})
	if err != nil {
		t.Fatal(err)
	}

	if totalCount != 1 || len(tableRows) != int(totalCount) {
		t.Fatal("总数不正确")
	}

	queryClasses, err := sql.ParseSqlResult[[]Class](tableRows)
	if err != nil {
		t.Fatal(err)
	}

	if queryClasses[0].ID != classID ||
		queryClasses[0].Name != className ||
		queryClasses[0].StudentNum != 0 ||
		strings.Join(queryClasses[0].StudentIDs, "\n") != strings.Join(studentIDs, "\n") ||
		!queryClasses[0].GraduatedTime.IsZero() ||
		(queryClasses[0].CreatedTime != nil && !queryClasses[0].CreatedTime.IsZero()) ||
		!queryClasses[0].LastUpdatedTime.IsZero() {
		t.Fatal("查询数据不正确")
	}

	tableRow, err := sql.QueryOne(sdk.GetInstance(), &sql_tpl.QueryOneExecuteParams{
		TableName:     tableName,
		SelectColumns: []string{"id", "name"},
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
	})
	if err != nil {
		t.Fatal(err)
	}

	queryClass, err := sql.ParseSqlResult[Class](tableRow)
	if err != nil {
		t.Fatal(err)
	}

	if queryClass.ID != classID ||
		queryClass.Name != className ||
		queryClass.StudentNum != 0 ||
		strings.Join(queryClasses[0].StudentIDs, "\n") != strings.Join(studentIDs, "\n") ||
		!queryClass.GraduatedTime.IsZero() ||
		(queryClass.CreatedTime != nil && !queryClass.CreatedTime.IsZero()) ||
		!queryClass.LastUpdatedTime.IsZero() {
		t.Fatal("查询数据不正确")
	}

	queryCount, err := sql.Count(sdk.GetInstance(), &sql_tpl.CountExecuteParams{
		TableName: tableName,
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
	})
	if err != nil {
		t.Fatal(err)
	}

	if queryCount != 1 {
		t.Fatal("数量不正确")
	}

	exist, err := sql.CheckExist(sdk.GetInstance(), &sql_tpl.CheckExistExecuteParams{
		TableName: tableName,
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
	})
	if err != nil {
		t.Fatal(err)
	}

	if !exist {
		t.Fatal("存在状态错误")
	}

	hasOnlyOne, err := sql.CheckHasOnlyOne(sdk.GetInstance(), &sql_tpl.CheckHasOnlyOneExecuteParams{
		TableName: tableName,
		Conditions: sql_tpl.NewConditions().
			Equal("id", classID).
			Equal("name", className, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
			Equal("student_num", studentNum),
	})
	if err != nil {
		t.Fatal(err)
	}

	if !hasOnlyOne {
		t.Fatal("唯一性错误")
	}

	err = sql.DeleteEntity(sdk.GetInstance(), tableName, class)
	if err != nil {
		t.Fatal(err)
	}
}

func TestInsertBatch(t *testing.T) {
	classID1 := strutils.SimpleUUID()
	className1 := strutils.SimpleUUID()
	studentNum1 := rand.Int31n(100)
	classID2 := strutils.SimpleUUID()
	className2 := strutils.SimpleUUID()
	studentNum2 := rand.Int31n(100)
	now := time.Now()

	insertBatchExecuteParams := &sql_tpl.InsertBatchExecuteParams{
		TableName: tableName,
		TableRowBatch: []sql_tpl.TableRow{
			*(sql_tpl.NewTableRow().Add("id", classID1).
				Add("name", className1, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
				Add("student_num", studentNum1).
				Add("student_ids", "").
				Add("created_time", now).
				Add("last_updated_time", now)),
			*(sql_tpl.NewTableRow().Add("id", classID2).
				Add("name", className2, sql_tpl.WithAESKey("@MKU^AHYCN$:j76J<TAHCVD#$XZSWQ@L")).
				Add("student_num", studentNum2).
				Add("student_ids", "").
				Add("created_time", now).
				Add("last_updated_time", now)),
		},
	}

	class1 := &Class{
		IDField:    IDField{ID: classID1},
		Name:       className1,
		StudentNum: int(studentNum1),
		StudentIDs: make([]string, 0),
	}

	class2 := &Class{
		IDField:    IDField{ID: classID2},
		Name:       className2,
		StudentNum: int(studentNum2),
		StudentIDs: make([]string, 0),
	}

	err := sdk.InitInstance(token, address, httpPort, grpcPort, namespace, dataSource)
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		err := sdk.DestroyInstance()
		if err != nil {
			t.Fatal(err)
		}
	}()

	err = sql.InsertBatch(sdk.GetInstance(), insertBatchExecuteParams)
	if err != nil {
		t.Fatal(err)
	}

	results, _, err := sql.Query(sdk.GetInstance(), &sql_tpl.QueryExecuteParams{
		TableName:     tableName,
		SelectColumns: []string{"id", "name", "student_num as student_num_alias"},
		Conditions:    sql_tpl.NewConditions().In("id", []string{classID1, classID2}),
		PageNo:        0,
		PageSize:      0,
	})
	if err != nil {
		t.Fatal(err)
	}

	classInfos, err := sql.ParseSqlResult[[]Class](results)
	if err != nil {
		t.Fatal(err)
	}

	for _, classInfo := range classInfos {
		if classInfo.ID != classID1 && classInfo.ID != classID2 {
			t.Fatal("id不正确")
		}

		if classInfo.ID == classID1 &&
			(classInfo.Name != className1 || classInfo.StudentNum != int(studentNum1)) {
			t.Fatal("数据不正确")
		}

		if classInfo.ID == classID2 &&
			(classInfo.Name != className2 || classInfo.StudentNum != int(studentNum2)) {
			t.Fatal("数据不正确")
		}
	}

	err = sql.Delete(sdk.GetInstance(), &sql_tpl.DeleteExecuteParams{
		TableName:  tableName,
		Conditions: sql_tpl.NewConditions().In("id", []string{classID1, classID2}),
	})
	if err != nil {
		t.Fatal(err)
	}

	err = sql.InsertEntity(sdk.GetInstance(), tableName, []*Class{class1, class2})
	if err != nil {
		t.Fatal(err)
	}

	results, _, err = sql.Query(sdk.GetInstance(), &sql_tpl.QueryExecuteParams{
		TableName:     tableName,
		SelectColumns: []string{"id", "name", "student_num as student_num_alias"},
		Conditions:    sql_tpl.NewConditions().In("id", []string{classID1, classID2}),
		PageNo:        0,
		PageSize:      0,
	})
	if err != nil {
		t.Fatal(err)
	}

	classInfos, err = sql.ParseSqlResult[[]Class](results)
	if err != nil {
		t.Fatal(err)
	}

	for _, classInfo := range classInfos {
		if classInfo.ID != classID1 && classInfo.ID != classID2 {
			t.Fatal("id不正确")
		}

		if classInfo.ID == classID1 &&
			(classInfo.Name != className1 || classInfo.StudentNum != int(studentNum1)) {
			t.Fatal("数据不正确")
		}

		if classInfo.ID == classID2 &&
			(classInfo.Name != className2 || classInfo.StudentNum != int(studentNum2)) {
			t.Fatal("数据不正确")
		}
	}

	err = sql.Delete(sdk.GetInstance(), &sql_tpl.DeleteExecuteParams{
		TableName:  tableName,
		Conditions: sql_tpl.NewConditions().In("id", []string{classID1, classID2}),
	})
	if err != nil {
		t.Fatal(err)
	}
}