✨ init project

service/example/enter.go

@@ -0,0 +1,7 @@
+package example
+
+type ServiceGroup struct {
+	CustomerService
+	FileUploadAndDownloadService
+	AttachmentCategoryService
+}

service/example/exa_attachment_category.go

@@ -0,0 +1,66 @@
+package example
+
+import (
+	"errors"
+	"git.echol.cn/loser/lckt/global"
+	"git.echol.cn/loser/lckt/model/example"
+	"gorm.io/gorm"
+)
+
+type AttachmentCategoryService struct{}
+
+// AddCategory 创建/更新的分类
+func (a *AttachmentCategoryService) AddCategory(req *example.ExaAttachmentCategory) (err error) {
+	// 检查是否已存在相同名称的分类
+	if (!errors.Is(global.GVA_DB.Take(&example.ExaAttachmentCategory{}, "name = ? and pid = ?", req.Name, req.Pid).Error, gorm.ErrRecordNotFound)) {
+		return errors.New("分类名称已存在")
+	}
+	if req.ID > 0 {
+		if err = global.GVA_DB.Model(&example.ExaAttachmentCategory{}).Where("id = ?", req.ID).Updates(&example.ExaAttachmentCategory{
+			Name: req.Name,
+			Pid:  req.Pid,
+		}).Error; err != nil {
+			return err
+		}
+	} else {
+		if err = global.GVA_DB.Create(&example.ExaAttachmentCategory{
+			Name: req.Name,
+			Pid:  req.Pid,
+		}).Error; err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// DeleteCategory 删除分类
+func (a *AttachmentCategoryService) DeleteCategory(id *int) error {
+	var childCount int64
+	global.GVA_DB.Model(&example.ExaAttachmentCategory{}).Where("pid = ?", id).Count(&childCount)
+	if childCount > 0 {
+		return errors.New("请先删除子级")
+	}
+	return global.GVA_DB.Where("id = ?", id).Unscoped().Delete(&example.ExaAttachmentCategory{}).Error
+}
+
+// GetCategoryList 分类列表
+func (a *AttachmentCategoryService) GetCategoryList() (res []*example.ExaAttachmentCategory, err error) {
+	var fileLists []example.ExaAttachmentCategory
+	err = global.GVA_DB.Model(&example.ExaAttachmentCategory{}).Find(&fileLists).Error
+	if err != nil {
+		return res, err
+	}
+	return a.getChildrenList(fileLists, 0), nil
+}
+
+// getChildrenList 子类
+func (a *AttachmentCategoryService) getChildrenList(categories []example.ExaAttachmentCategory, parentID uint) []*example.ExaAttachmentCategory {
+	var tree []*example.ExaAttachmentCategory
+	for _, category := range categories {
+		if category.Pid == parentID {
+			category.Children = a.getChildrenList(categories, category.ID)
+			tree = append(tree, &category)
+		}
+	}
+	return tree
+}

service/example/exa_breakpoint_continue.go

@@ -0,0 +1,71 @@
+package example
+
+import (
+	"errors"
+
+	"git.echol.cn/loser/lckt/global"
+	"git.echol.cn/loser/lckt/model/example"
+	"gorm.io/gorm"
+)
+
+type FileUploadAndDownloadService struct{}
+
+var FileUploadAndDownloadServiceApp = new(FileUploadAndDownloadService)
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: FindOrCreateFile
+//@description: 上传文件时检测当前文件属性，如果没有文件则创建，有则返回文件的当前切片
+//@param: fileMd5 string, fileName string, chunkTotal int
+//@return: file model.ExaFile, err error
+
+func (e *FileUploadAndDownloadService) FindOrCreateFile(fileMd5 string, fileName string, chunkTotal int) (file example.ExaFile, err error) {
+	var cfile example.ExaFile
+	cfile.FileMd5 = fileMd5
+	cfile.FileName = fileName
+	cfile.ChunkTotal = chunkTotal
+
+	if errors.Is(global.GVA_DB.Where("file_md5 = ? AND is_finish = ?", fileMd5, true).First(&file).Error, gorm.ErrRecordNotFound) {
+		err = global.GVA_DB.Where("file_md5 = ? AND file_name = ?", fileMd5, fileName).Preload("ExaFileChunk").FirstOrCreate(&file, cfile).Error
+		return file, err
+	}
+	cfile.IsFinish = true
+	cfile.FilePath = file.FilePath
+	err = global.GVA_DB.Create(&cfile).Error
+	return cfile, err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: CreateFileChunk
+//@description: 创建文件切片记录
+//@param: id uint, fileChunkPath string, fileChunkNumber int
+//@return: error
+
+func (e *FileUploadAndDownloadService) CreateFileChunk(id uint, fileChunkPath string, fileChunkNumber int) error {
+	var chunk example.ExaFileChunk
+	chunk.FileChunkPath = fileChunkPath
+	chunk.ExaFileID = id
+	chunk.FileChunkNumber = fileChunkNumber
+	err := global.GVA_DB.Create(&chunk).Error
+	return err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: DeleteFileChunk
+//@description: 删除文件切片记录
+//@param: fileMd5 string, fileName string, filePath string
+//@return: error
+
+func (e *FileUploadAndDownloadService) DeleteFileChunk(fileMd5 string, filePath string) error {
+	var chunks []example.ExaFileChunk
+	var file example.ExaFile
+	err := global.GVA_DB.Where("file_md5 = ?", fileMd5).First(&file).
+		Updates(map[string]interface{}{
+			"IsFinish":  true,
+			"file_path": filePath,
+		}).Error
+	if err != nil {
+		return err
+	}
+	err = global.GVA_DB.Where("exa_file_id = ?", file.ID).Delete(&chunks).Unscoped().Error
+	return err
+}

service/example/exa_customer.go

@@ -0,0 +1,87 @@
+package example
+
+import (
+	"git.echol.cn/loser/lckt/global"
+	"git.echol.cn/loser/lckt/model/common/request"
+	"git.echol.cn/loser/lckt/model/example"
+	"git.echol.cn/loser/lckt/model/system"
+	systemService "git.echol.cn/loser/lckt/service/system"
+)
+
+type CustomerService struct{}
+
+var CustomerServiceApp = new(CustomerService)
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: CreateExaCustomer
+//@description: 创建客户
+//@param: e model.ExaCustomer
+//@return: err error
+
+func (exa *CustomerService) CreateExaCustomer(e example.ExaCustomer) (err error) {
+	err = global.GVA_DB.Create(&e).Error
+	return err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: DeleteFileChunk
+//@description: 删除客户
+//@param: e model.ExaCustomer
+//@return: err error
+
+func (exa *CustomerService) DeleteExaCustomer(e example.ExaCustomer) (err error) {
+	err = global.GVA_DB.Delete(&e).Error
+	return err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: UpdateExaCustomer
+//@description: 更新客户
+//@param: e *model.ExaCustomer
+//@return: err error
+
+func (exa *CustomerService) UpdateExaCustomer(e *example.ExaCustomer) (err error) {
+	err = global.GVA_DB.Save(e).Error
+	return err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: GetExaCustomer
+//@description: 获取客户信息
+//@param: id uint
+//@return: customer model.ExaCustomer, err error
+
+func (exa *CustomerService) GetExaCustomer(id uint) (customer example.ExaCustomer, err error) {
+	err = global.GVA_DB.Where("id = ?", id).First(&customer).Error
+	return
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: GetCustomerInfoList
+//@description: 分页获取客户列表
+//@param: sysUserAuthorityID string, info request.PageInfo
+//@return: list interface{}, total int64, err error
+
+func (exa *CustomerService) GetCustomerInfoList(sysUserAuthorityID uint, info request.PageInfo) (list interface{}, total int64, err error) {
+	limit := info.PageSize
+	offset := info.PageSize * (info.Page - 1)
+	db := global.GVA_DB.Model(&example.ExaCustomer{})
+	var a system.SysAuthority
+	a.AuthorityId = sysUserAuthorityID
+	auth, err := systemService.AuthorityServiceApp.GetAuthorityInfo(a)
+	if err != nil {
+		return
+	}
+	var dataId []uint
+	for _, v := range auth.DataAuthorityId {
+		dataId = append(dataId, v.AuthorityId)
+	}
+	var CustomerList []example.ExaCustomer
+	err = db.Where("sys_user_authority_id in ?", dataId).Count(&total).Error
+	if err != nil {
+		return CustomerList, total, err
+	} else {
+		err = db.Limit(limit).Offset(offset).Preload("SysUser").Where("sys_user_authority_id in ?", dataId).Find(&CustomerList).Error
+	}
+	return CustomerList, total, err
+}

service/example/exa_file_upload_download.go

@@ -0,0 +1,123 @@
+package example
+
+import (
+	"errors"
+	"mime/multipart"
+	"strings"
+
+	"git.echol.cn/loser/lckt/global"
+	"git.echol.cn/loser/lckt/model/example"
+	"git.echol.cn/loser/lckt/model/example/request"
+	"git.echol.cn/loser/lckt/utils/upload"
+)
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: Upload
+//@description: 创建文件上传记录
+//@param: file model.ExaFileUploadAndDownload
+//@return: error
+
+func (e *FileUploadAndDownloadService) Upload(file example.ExaFileUploadAndDownload) error {
+	return global.GVA_DB.Create(&file).Error
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: FindFile
+//@description: 查询文件记录
+//@param: id uint
+//@return: model.ExaFileUploadAndDownload, error
+
+func (e *FileUploadAndDownloadService) FindFile(id uint) (example.ExaFileUploadAndDownload, error) {
+	var file example.ExaFileUploadAndDownload
+	err := global.GVA_DB.Where("id = ?", id).First(&file).Error
+	return file, err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: DeleteFile
+//@description: 删除文件记录
+//@param: file model.ExaFileUploadAndDownload
+//@return: err error
+
+func (e *FileUploadAndDownloadService) DeleteFile(file example.ExaFileUploadAndDownload) (err error) {
+	var fileFromDb example.ExaFileUploadAndDownload
+	fileFromDb, err = e.FindFile(file.ID)
+	if err != nil {
+		return
+	}
+	oss := upload.NewOss()
+	if err = oss.DeleteFile(fileFromDb.Key); err != nil {
+		return errors.New("文件删除失败")
+	}
+	err = global.GVA_DB.Where("id = ?", file.ID).Unscoped().Delete(&file).Error
+	return err
+}
+
+// EditFileName 编辑文件名或者备注
+func (e *FileUploadAndDownloadService) EditFileName(file example.ExaFileUploadAndDownload) (err error) {
+	var fileFromDb example.ExaFileUploadAndDownload
+	return global.GVA_DB.Where("id = ?", file.ID).First(&fileFromDb).Update("name", file.Name).Error
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: GetFileRecordInfoList
+//@description: 分页获取数据
+//@param: info request.ExaAttachmentCategorySearch
+//@return: list interface{}, total int64, err error
+
+func (e *FileUploadAndDownloadService) GetFileRecordInfoList(info request.ExaAttachmentCategorySearch) (list []example.ExaFileUploadAndDownload, total int64, err error) {
+	limit := info.PageSize
+	offset := info.PageSize * (info.Page - 1)
+	db := global.GVA_DB.Model(&example.ExaFileUploadAndDownload{})
+
+	if len(info.Keyword) > 0 {
+		db = db.Where("name LIKE ?", "%"+info.Keyword+"%")
+	}
+
+	if info.ClassId > 0 {
+		db = db.Where("class_id = ?", info.ClassId)
+	}
+
+	err = db.Count(&total).Error
+	if err != nil {
+		return
+	}
+	err = db.Limit(limit).Offset(offset).Order("id desc").Find(&list).Error
+	return list, total, err
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: UploadFile
+//@description: 根据配置文件判断是文件上传到本地或者七牛云
+//@param: header *multipart.FileHeader, noSave string
+//@return: file model.ExaFileUploadAndDownload, err error
+
+func (e *FileUploadAndDownloadService) UploadFile(header *multipart.FileHeader, noSave string, classId int) (file example.ExaFileUploadAndDownload, err error) {
+	oss := upload.NewOss()
+	filePath, key, uploadErr := oss.UploadFile(header)
+	if uploadErr != nil {
+		return file, uploadErr
+	}
+	s := strings.Split(header.Filename, ".")
+	f := example.ExaFileUploadAndDownload{
+		Url:     filePath,
+		Name:    header.Filename,
+		ClassId: classId,
+		Tag:     s[len(s)-1],
+		Key:     key,
+	}
+	if noSave == "0" {
+		return f, e.Upload(f)
+	}
+	return f, nil
+}
+
+//@author: [piexlmax](https://github.com/piexlmax)
+//@function: ImportURL
+//@description: 导入URL
+//@param: file model.ExaFileUploadAndDownload
+//@return: error
+
+func (e *FileUploadAndDownloadService) ImportURL(file *[]example.ExaFileUploadAndDownload) error {
+	return global.GVA_DB.Create(&file).Error
+}