новый

Makefile

@@ -23,19 +23,18 @@ build:
 	go build -race -o $(FILEAPP) $(FILEMAIN)
 	cd ./cmd && \
 	./VersionToFile.py
-	cp $(FILEAPP) $(GOPATH)/bin
 
 lint:
 	clear
 	go fmt ./...
-	golangci-lint run ./internal/v0/...
+	golangci-lint run ./internal/...
-	golangci-lint run ./pkg/v0/...
+	golangci-lint run ./pkg/...
-	gocyclo -over 10 ./internal/v0
+	gocyclo -over 10 ./internal
-	gocyclo -over 10 ./pkg/v0
+	gocyclo -over 10 ./pkg
-	gocritic check ./internal/v0/...
+	gocritic check ./internal/...
-	gocritic check ./pkg/v0/...
+	gocritic check ./pkg/...
-	staticcheck ./internal/v0/...
+	staticcheck ./internal/...
-	staticcheck ./pkg/v0/...
+	staticcheck ./pkg/...
 run.test:
 	clear
 	go fmt ./...

bin/templates/db/db_lawsuit_status_types.go

@@ -0,0 +1,299 @@
+package db_lawsuit_status_types
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	model "gitlab.aescorp.ru/dsp_dev/claim/common/object_model"
+	"gitlab.aescorp.ru/dsp_dev/claim/common/object_model/pkg/db/constants"
+	"gitlab.aescorp.ru/dsp_dev/claim/nikitin/micro"
+	"gitlab.aescorp.ru/dsp_dev/claim/nikitin/postgres_gorm"
+	"gorm.io/gorm"
+	"time"
+)
+
+// TableName - имя таблицы в БД Postgres
+const TableName string = "lawsuit_status_types"
+
+// Crud_DB - объект для CRUD операций через БД
+type Crud_DB struct {
+}
+
+// Read - находит запись в БД по ID
+func (crud Crud_DB) Read(l *model.LawsuitStatusType) error {
+	//var Otvet model.LawsuitStatusType
+	var err error
+
+	//log.Trace("start Read() ", TableName, " id: ", id)
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err = crud.Read_ctx(ctx, l)
+	return err
+}
+
+// Read_ctx - находит запись в БД по ID
+func (crud Crud_DB) Read_ctx(ctx context.Context, l *model.LawsuitStatusType) error {
+	//var Otvet model.LawsuitStatusType
+	var err error
+
+	id := l.ID
+
+	db := postgres_gorm.GetConnection()
+	db.WithContext(ctx)
+
+	tx := db.First(l, id)
+	err = tx.Error
+
+	return err
+}
+
+// Save - записывает новый или существующий объект в базу данных
+func (crud Crud_DB) Save(l *model.LawsuitStatusType) error {
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err := crud.Save_ctx(ctx, l)
+	return err
+}
+
+// Save_ctx - записывает новый или существующий объект в базу данных
+func (crud Crud_DB) Save_ctx(ctx context.Context, l *model.LawsuitStatusType) error {
+	is_create := !micro.BoolFromInt64(l.ID)
+	err := crud.create_update_ctx(ctx, l, is_create)
+	return err
+}
+
+// Update - записывает существующий объект в базу данных
+func (crud Crud_DB) Update(l *model.LawsuitStatusType) error {
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err := crud.Update_ctx(ctx, l)
+	return err
+}
+
+// Update_ctx - записывает существующий объект в базу данных
+func (crud Crud_DB) Update_ctx(ctx context.Context, l *model.LawsuitStatusType) error {
+	err := crud.create_update_ctx(ctx, l, false)
+	return err
+}
+
+// Create - записывает новый объект в базу данных
+func (crud Crud_DB) Create(l *model.LawsuitStatusType) error {
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err := crud.Create_ctx(ctx, l)
+	return err
+}
+
+// Create_ctx - записывает новый объект в базу данных
+func (crud Crud_DB) Create_ctx(ctx context.Context, l *model.LawsuitStatusType) error {
+	err := crud.create_update_ctx(ctx, l, true)
+	return err
+}
+
+// create_update - записывает объект в базу данных
+func (crud Crud_DB) create_update(l *model.LawsuitStatusType, is_create bool) error {
+	var err error
+
+	//log.Trace("start Save() ", TableName, " id: ", m.ID)
+
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err = crud.create_update_ctx(ctx, l, is_create)
+	return err
+}
+
+// create_update_ctx - записывает объект в базу данных
+func (crud Crud_DB) create_update_ctx(ctx context.Context, l *model.LawsuitStatusType, is_create bool) error {
+	var err error
+
+	//log.Trace("start Save() ", TableName, " id: ", m.ID)
+
+	// проверка ID
+	if is_create == true {
+		if l.ID != 0 {
+			TextError := fmt.Sprint("db.Save() ", TableName, " error: id !=0")
+			//log.Panic(sError)
+			err = errors.New(TextError)
+			return err
+		}
+	} else if l.ID == 0 {
+		TextError := fmt.Sprint("db.Save() ", TableName, " error: id =0")
+		err = errors.New(TextError)
+		//log.Panic(sError)
+		return err
+	}
+
+	//
+	db := postgres_gorm.GetConnection()
+	db.WithContext(ctx)
+
+	//заполним даты
+	Now := time.Now()
+	l.ModifiedAt = Now
+	if l.IsDeleted == true && l.DeletedAt.IsZero() == true {
+		l.DeletedAt = Now
+	} else if l.IsDeleted == false && l.DeletedAt.IsZero() == false {
+		l.DeletedAt = time.Time{}
+	}
+
+	//колонки с null
+	tx := db
+	MassOmit := make([]string, 0)
+	var ColumnName string
+
+	ColumnName = "DeletedAt"
+	if l.DeletedAt.IsZero() == true {
+		MassOmit = append(MassOmit, ColumnName)
+	}
+
+	ColumnName = "ExtID"
+	if l.ExtID == 0 {
+		MassOmit = append(MassOmit, ColumnName)
+	}
+
+	//игнор пустых колонок
+	tx = tx.Omit(MassOmit...)
+
+	//запись
+	if is_create == true {
+		tx = tx.Create(&l)
+	} else {
+		tx = tx.Save(&l)
+	}
+	err = tx.Error
+	if err != nil {
+		return err
+	}
+
+	//запишем NULL в пустые колонки
+	for f := 0; f < len(MassOmit); f++ {
+		ColumnName := MassOmit[f]
+		tx = db.First(&l).Update(ColumnName, gorm.Expr("NULL"))
+
+		err = tx.Error
+		if err != nil {
+			TextError := fmt.Sprint("db.Update() ", TableName, " id: ", l.ID, " error: ", err)
+			err = errors.New(TextError)
+			return err
+			//log.Panic(sError)
+		}
+	}
+
+	return err
+}
+
+// Delete - записывает is_deleted = true
+func (crud Crud_DB) Delete(l *model.LawsuitStatusType) error {
+	//var Otvet model.LawsuitStatusType
+	var err error
+
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err = crud.Delete_ctx(ctx, l)
+	return err
+}
+
+// Delete_ctx - записывает is_deleted = true
+func (crud Crud_DB) Delete_ctx(ctx context.Context, l *model.LawsuitStatusType) error {
+	//var Otvet model.LawsuitStatusType
+	var err error
+
+	l2 := model.LawsuitStatusType{}
+	l2.ID = l.ID
+	err = crud.Read_ctx(ctx, &l2)
+	if err != nil {
+		return err
+	}
+
+	l2.IsDeleted = true
+	l.IsDeleted = true
+
+	err = crud.Save_ctx(ctx, &l2)
+
+	return err
+}
+
+// Restore - записывает is_deleted = true
+func (crud Crud_DB) Restore(l *model.LawsuitStatusType) error {
+	//var Otvet model.LawsuitStatusType
+	var err error
+
+	ctxMain := context.Background()
+	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+	defer ctxCancelFunc()
+
+	err = crud.Restore_ctx(ctx, l)
+	return err
+}
+
+// Restore_ctx - записывает is_deleted = true
+func (crud Crud_DB) Restore_ctx(ctx context.Context, l *model.LawsuitStatusType) error {
+	//var Otvet model.LawsuitStatusType
+	var err error
+
+	l2 := model.LawsuitStatusType{}
+	l2.ID = l.ID
+	err = crud.Read_ctx(ctx, &l2)
+	if err != nil {
+		return err
+	}
+
+	l2.IsDeleted = false
+	l.IsDeleted = false
+
+	err = crud.Save_ctx(ctx, &l2)
+
+	return err
+}
+
+//// Find_ByExtID - находит запись в БД по ext_id и connection_id
+//func Find_ByExtID(ext_id int64, connection_id int64) (model.LawsuitStatusType, error) {
+//	var Otvet model.LawsuitStatusType
+//	var err error
+//
+//	if ext_id <= 0 {
+//		err = errors.New("Error: ext_id <=0")
+//		return Otvet, err
+//	}
+//
+//	//
+//	ctxMain := context.Background()
+//	ctx, ctxCancelFunc := context.WithTimeout(ctxMain, time.Second*time.Duration(constants.TIMEOUT_DB_SECONDS))
+//	defer ctxCancelFunc()
+//
+//	Otvet, err = Find_ByExtID_ctx(ctx, ext_id, connection_id)
+//
+//	return Otvet, err
+//}
+//
+//// Find_ByExtID_ctx - находит запись в БД по ext_id и connection_id
+//func Find_ByExtID_ctx(ctx context.Context, ext_id int64, connection_id int64) (model.LawsuitStatusType, error) {
+//	var Otvet model.LawsuitStatusType
+//	var err error
+//	//log.Trace("start Find_ByExtID() ", TableName, " ext_id: ", ext_id)
+//
+//	if ext_id <= 0 {
+//		err = errors.New("Error: ext_id <=0")
+//		return Otvet, err
+//	}
+//
+//	db := postgres_gorm.GetConnection()
+//	db.WithContext(ctx)
+//
+//	tx := db.Where("ext_id = ?", ext_id).Where("connection_id = ?", connection_id).First(&Otvet)
+//	err = tx.Error
+//
+//	return Otvet, err
+//}

bin/templates/db/db_lawsuit_status_types_test.go

@@ -0,0 +1,111 @@
+package db_lawsuit_status_types
+
+import (
+	model "gitlab.aescorp.ru/dsp_dev/claim/common/object_model"
+	"gitlab.aescorp.ru/dsp_dev/claim/nikitin/config"
+	"gitlab.aescorp.ru/dsp_dev/claim/nikitin/postgres_gorm"
+	"testing"
+)
+
+const Postgres_ID_Test = 1
+
+func TestRead(t *testing.T) {
+	config.LoadEnv()
+
+	postgres_gorm.Connect()
+	defer postgres_gorm.CloseConnection()
+
+	crud := Crud_DB{}
+	Otvet := model.LawsuitStatusType{}
+	Otvet.ID = Postgres_ID_Test
+	err := crud.Read(&Otvet)
+	if err != nil {
+		t.Error("TestRead() error: ", err)
+	}
+
+	if Otvet.Name == "" {
+		t.Error(TableName + "_test.TestRead() error name= '' ")
+	} else {
+		t.Log(TableName+"_test.TestRead() Otvet: ", Otvet.Name)
+	}
+}
+
+func TestSave(t *testing.T) {
+	config.LoadEnv()
+
+	postgres_gorm.Connect()
+	defer postgres_gorm.CloseConnection()
+
+	crud := Crud_DB{}
+	Otvet := model.LawsuitStatusType{}
+	Otvet.ID = Postgres_ID_Test
+	err := crud.Read(&Otvet)
+	if err != nil {
+		t.Error("TestSave() error: ", err)
+	}
+
+	if Otvet.Name == "" {
+		t.Error(TableName + "_test.TestSave() error name= '' ")
+	}
+
+	err = crud.Save(&Otvet)
+	if err != nil {
+		t.Error("TestSave() error: ", err)
+	}
+	t.Log(TableName+"_test.TestSave() Otvet: ", Otvet.Name)
+
+}
+
+func TestDelete(t *testing.T) {
+	config.LoadEnv()
+
+	postgres_gorm.Connect()
+	defer postgres_gorm.CloseConnection()
+
+	crud := Crud_DB{}
+	Otvet := model.LawsuitStatusType{}
+	Otvet.ID = Postgres_ID_Test
+	err := crud.Read(&Otvet)
+	if err != nil {
+		t.Error("TestDelete() error: ", err)
+	}
+
+	if Otvet.IsDeleted == false {
+		err = crud.Delete(&Otvet)
+		if err != nil {
+			t.Error("TestDelete() error: ", err)
+		}
+
+		err = crud.Restore(&Otvet)
+		if err != nil {
+			t.Error("TestDelete() error: ", err)
+		}
+	} else {
+		err = crud.Restore(&Otvet)
+		if err != nil {
+			t.Error("TestDelete() error: ", err)
+		}
+
+		err = crud.Delete(&Otvet)
+		if err != nil {
+			t.Error("TestDelete() error: ", err)
+		}
+
+	}
+
+}
+
+//func TestFind_ByExtID(t *testing.T) {
+//	config.LoadEnv()
+//	postgres_gorm.Connect()
+//	defer postgres_gorm.CloseConnection()
+//
+//	Otvet, err := Find_ByExtID(1, constants.CONNECTION_ID_TEST)
+//	if err != nil {
+//		t.Error("TestFind_ByExtID() error: ", err)
+//	}
+//
+//	if Otvet.ID == 0 {
+//		t.Error("TestFind_ByExtID() error: ID =0")
+//	}
+//}

internal/config/config.go

@@ -1,6 +1,7 @@
 package config
 
 import (
+	"log"
 	"os"
 	"strings"
 )
@@ -10,16 +11,17 @@ var Settings SettingsINI
 
 // SettingsINI - структура для хранения всех нужных переменных окружения
 type SettingsINI struct {
-	INCLUDE_TABLES          string
+	INCLUDE_TABLES            string
-	EXCLUDE_TABLES          string
+	EXCLUDE_TABLES            string
-	TEMPLATE_FILENAME_MODEL string
+	TEMPLATE_FOLDERNAME       string
-	TEMPLATE_FILENAME_DB    string
+	TEMPLATE_FOLDERNAME_MODEL string
-	TEMPLATE_FILENAME_GRPC  string
+	TEMPLATE_FOLDERNAME_DB    string
-	TEMPLATE_FILENAME_NRPC  string
+	TEMPLATE_FOLDERNAME_GRPC  string
-	NEED_CRUD               bool
+	TEMPLATE_FOLDERNAME_NRPC  string
-	NEED_GRPC               bool
+	NEED_CRUD                 bool
-	NEED_NRPC               bool
+	NEED_GRPC                 bool
-	SERVICE_NAME            string
+	NEED_NRPC                 bool
+	SERVICE_NAME              string
 }
 
 // FillSettings загружает переменные окружения в структуру из переменных окружения
@@ -27,10 +29,11 @@ func FillSettings() {
 	Settings = SettingsINI{}
 	Settings.INCLUDE_TABLES = os.Getenv("INCLUDE_TABLES")
 	Settings.EXCLUDE_TABLES = os.Getenv("EXCLUDE_TABLES")
-	Settings.TEMPLATE_FILENAME_MODEL = os.Getenv("TEMPLATE_FILENAME_MODEL")
+	Settings.TEMPLATE_FOLDERNAME = os.Getenv("TEMPLATE_FOLDERNAME")
-	Settings.TEMPLATE_FILENAME_DB = os.Getenv("TEMPLATE_FILENAME_DB")
+	Settings.TEMPLATE_FOLDERNAME_MODEL = os.Getenv("TEMPLATE_FOLDERNAME_MODEL")
-	Settings.TEMPLATE_FILENAME_GRPC = os.Getenv("TEMPLATE_FILENAME_GRPC")
+	Settings.TEMPLATE_FOLDERNAME_DB = os.Getenv("TEMPLATE_FOLDERNAME_DB")
-	Settings.TEMPLATE_FILENAME_NRPC = os.Getenv("TEMPLATE_FILENAME_NRPC")
+	Settings.TEMPLATE_FOLDERNAME_GRPC = os.Getenv("TEMPLATE_FOLDERNAME_GRPC")
+	Settings.TEMPLATE_FOLDERNAME_NRPC = os.Getenv("TEMPLATE_FOLDERNAME_NRPC")
 
 	sNEED_CRUD := os.Getenv("NEED_CRUD")
 	Settings.NEED_CRUD = BoolFromString(sNEED_CRUD)
@@ -43,6 +46,9 @@ func FillSettings() {
 
 	Settings.SERVICE_NAME = os.Getenv("SERVICE_NAME")
 
+	if Settings.TEMPLATE_FOLDERNAME == "" {
+		log.Panic("Need fill TEMPLATE_FOLDERNAME")
+	}
 	//
 }