go/crud_generator
1
0
Fork 0
mirror of https://github.com/ManyakRus/crud_generator.git synced 2025-07-16 02:54:19 +02:00
Code Issues Releases Activity

новый

Browse Source
This commit is contained in:
Nikitin Aleksandr
2023-10-23 18:09:55 +03:00
parent 85115db79f
commit 7534fbca7d
5 changed files with 438 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
Makefile
View File

@ -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 ./pkg/v0/...
gocyclo -over 10 ./internal/v0
gocyclo -over 10 ./pkg/v0
gocritic check ./internal/v0/...
gocritic check ./pkg/v0/...
staticcheck ./internal/v0/...
staticcheck ./pkg/v0/...
golangci-lint run ./internal/...
golangci-lint run ./pkg/...
gocyclo -over 10 ./internal
gocyclo -over 10 ./pkg
gocritic check ./internal/...
gocritic check ./pkg/...
staticcheck ./internal/...
staticcheck ./pkg/...
run.test:
clear
go fmt ./...

299
bin/templates/db/db_lawsuit_status_types.go Normal file
View File

@ -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
//}

111
bin/templates/db/db_lawsuit_status_types_test.go Normal file
View File

@ -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")
// }
//}

0
bin/templates/lawsuit_status_type.go → bin/templates/model/lawsuit_status_type.go
View File

34
internal/config/config.go
View File

@ -1,6 +1,7 @@
package config
import (
"log"
"os"
"strings"
)
@ -10,16 +11,17 @@ var Settings SettingsINI
// SettingsINI - структура для хранения всех нужных переменных окружения
type SettingsINI struct {
INCLUDE_TABLES string
EXCLUDE_TABLES string
TEMPLATE_FILENAME_MODEL string
TEMPLATE_FILENAME_DB string
TEMPLATE_FILENAME_GRPC string
TEMPLATE_FILENAME_NRPC string
NEED_CRUD bool
NEED_GRPC bool
NEED_NRPC bool
SERVICE_NAME string
INCLUDE_TABLES string
EXCLUDE_TABLES string
TEMPLATE_FOLDERNAME string
TEMPLATE_FOLDERNAME_MODEL string
TEMPLATE_FOLDERNAME_DB string
TEMPLATE_FOLDERNAME_GRPC string
TEMPLATE_FOLDERNAME_NRPC string
NEED_CRUD bool
NEED_GRPC bool
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_FILENAME_DB = os.Getenv("TEMPLATE_FILENAME_DB")
Settings.TEMPLATE_FILENAME_GRPC = os.Getenv("TEMPLATE_FILENAME_GRPC")
Settings.TEMPLATE_FILENAME_NRPC = os.Getenv("TEMPLATE_FILENAME_NRPC")
Settings.TEMPLATE_FOLDERNAME = os.Getenv("TEMPLATE_FOLDERNAME")
Settings.TEMPLATE_FOLDERNAME_MODEL = os.Getenv("TEMPLATE_FOLDERNAME_MODEL")
Settings.TEMPLATE_FOLDERNAME_DB = os.Getenv("TEMPLATE_FOLDERNAME_DB")
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")
}
//
}