Increase version after save for generic

Author: minhduc140583

adapter/adapter.go

@@ -87,6 +87,17 @@ func (a *Adapter[T]) Create(ctx context.Context, model T) (int64, error) {
 	if err != nil {
 		return -1, err
 	}
+	rowsAffected, err := res.RowsAffected()
+	if err != nil {
+		return rowsAffected, err
+	}
+	if rowsAffected > 0 && a.versionIndex >= 0 {
+		vo := reflect.ValueOf(model)
+		if vo.Kind() == reflect.Ptr {
+			vo = reflect.Indirect(vo)
+		}
+		setVersion(vo, a.versionIndex)
+	}
 	return res.RowsAffected()
 }
 func (a *Adapter[T]) Update(ctx context.Context, model T) (int64, error) {
@@ -96,6 +107,18 @@ func (a *Adapter[T]) Update(ctx context.Context, model T) (int64, error) {
 	if err != nil {
 		return -1, err
 	}
+	rowsAffected, err := res.RowsAffected()
+	if err != nil {
+		return rowsAffected, err
+	}
+	if rowsAffected > 0 && a.versionIndex >= 0 {
+		vo := reflect.ValueOf(model)
+		if vo.Kind() == reflect.Ptr {
+			vo = reflect.Indirect(vo)
+		}
+		currentVersion := vo.Field(a.versionIndex).Interface()
+		increaseVersion(vo, a.versionIndex, currentVersion)
+	}
 	return res.RowsAffected()
 }
 func (a *Adapter[T]) Save(ctx context.Context, model T) (int64, error) {
@@ -108,6 +131,18 @@ func (a *Adapter[T]) Save(ctx context.Context, model T) (int64, error) {
 	if err != nil {
 		return -1, err
 	}
+	rowsAffected, err := res.RowsAffected()
+	if err != nil {
+		return rowsAffected, err
+	}
+	if rowsAffected > 0 && a.versionIndex >= 0 {
+		vo := reflect.ValueOf(model)
+		if vo.Kind() == reflect.Ptr {
+			vo = reflect.Indirect(vo)
+		}
+		currentVersion := vo.Field(a.versionIndex).Interface()
+		increaseVersion(vo, a.versionIndex, currentVersion)
+	}
 	return res.RowsAffected()
 }
 func (a *Adapter[T]) Patch(ctx context.Context, model map[string]interface{}) (int64, error) {
@@ -136,3 +171,38 @@ func handleDuplicate(db *sql.DB, err error) (int64, error) {
 	}
 	return 0, err
 }
+func increaseVersion(vo reflect.Value, versionIndex int, curVer interface{}) bool {
+	versionType := vo.Field(versionIndex).Type().String()
+	switch versionType {
+	case "int32":
+		nextVer := curVer.(int32) + 1
+		vo.Field(versionIndex).Set(reflect.ValueOf(nextVer))
+		return true
+	case "int":
+		nextVer := curVer.(int) + 1
+		vo.Field(versionIndex).Set(reflect.ValueOf(nextVer))
+		return true
+	case "int64":
+		nextVer := curVer.(int64) + 1
+		vo.Field(versionIndex).Set(reflect.ValueOf(nextVer))
+		return true
+	default:
+		return false
+	}
+}
+func setVersion(vo reflect.Value, versionIndex int) bool {
+	versionType := vo.Field(versionIndex).Type().String()
+	switch versionType {
+	case "int32":
+		vo.Field(versionIndex).Set(reflect.ValueOf(int32(1)))
+		return true
+	case "int":
+		vo.Field(versionIndex).Set(reflect.ValueOf(1))
+		return true
+	case "int64":
+		vo.Field(versionIndex).Set(reflect.ValueOf(int64(1)))
+		return true
+	default:
+		return false
+	}
+}

adapter/generic_adapter.go

@@ -49,7 +49,7 @@ func NewSqlGenericAdapterWithVersionAndArray[T any, K any](db *sql.DB, tableName
 		if kType.Kind() == reflect.Map {
 			idMap = true
 		} else if kType.Kind() != reflect.Struct {
-			return nil, errors.New("For composite keys, K must be a struct or a map")
+			return nil, errors.New("for composite keys, K must be a struct or a map")
 		}
 	}
 

adapter/search.go

@@ -11,7 +11,6 @@ import (
 
 type SearchAdapter[T any, K any, F any] struct {
 	*GenericAdapter[T, K]
-	DB         *sql.DB
 	BuildQuery func(F) (string, []interface{})
 	Mp         func(ctx context.Context, model interface{}) (interface{}, error)
 	Map        map[string]int
@@ -48,13 +47,13 @@ func NewSearchAdapterWithArray[T any, K any, F any](db *sql.DB, table string, bu
 	if err != nil {
 		return nil, err
 	}
-	builder := &SearchAdapter[T, K, F]{GenericAdapter: adapter, DB: db, Map: fieldsIndex, BuildQuery: buildQuery, Mp: mp, ToArray: toArray}
+	builder := &SearchAdapter[T, K, F]{GenericAdapter: adapter, Map: fieldsIndex, BuildQuery: buildQuery, Mp: mp, ToArray: toArray}
 	return builder, nil
 }
 
 func (b *SearchAdapter[T, K, F]) Search(ctx context.Context, filter F, limit int64, offset int64) ([]T, int64, error) {
 	var objs []T
-	sql, params := b.BuildQuery(filter)
-	total, er2 := q.BuildFromQuery(ctx, b.DB, b.Map, &objs, sql, params, limit, offset, b.ToArray, b.Mp)
+	query, args := b.BuildQuery(filter)
+	total, er2 := q.BuildFromQuery(ctx, b.DB, b.Map, &objs, query, args, limit, offset, b.ToArray, b.Mp)
 	return objs, total, er2
 }