123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957 |
- 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)
- }
- }
|