update

models/book/book.go

@@ -36,6 +36,18 @@ func (m *BookDAO) GetBookList(bookType, caseid string, page, pageSize int) ([]mo
 	err := m.db.Sql(sql).Find(&book)
 	return book, err
 
+}
+func (m *BookDAO) GetBookListCount(bookType, caseid string) (int, error) {
+	var book []model.TaBook
+	sql := `select * from ta_book where status = '` + strconv.Itoa(models.STATUS_NORMAL) + `'
+	and a.case_id in('` + strings.Replace(caseid, ",", "','", -1) + `')`
+	if bookType != "" {
+		sql += ` and book_type_id = '` + bookType + `'`
+	}
+	sql += ` order by create_date desc`
+	err := m.db.Sql(sql).Find(&book)
+	return len(book), err
+
 }
 
 func (m *BookDAO) GetBookById(bookId string) (*model.TaBook, error) {
@@ -137,5 +149,4 @@ func (m *BookDAO) GetBookRecommendList(caseid string, page, pageSize int) ([]mod
 	sql += ` order by create_date desc limit ` + strconv.Itoa((page-1)*pageSize) + `, ` + strconv.Itoa(pageSize)
 	err := m.db.Sql(sql).Find(&book)
 	return book, err
-
 }

models/booktype/booktype.go

@@ -57,12 +57,12 @@ func (m *BooktypeDAO) AddBookType(booktype model.TaBookType) (*model.TaBookType,
 	_, err := m.db.Insert(booktype)
 	return &booktype, err
 }
-func (m *BooktypeDAO) EditBookType(booktype model.TaBookType) error {
+func (m *BooktypeDAO) EditBookType(booktype model.TaBookType) (*model.TaBookType, error) {
 	var cols = []string{
 		"book_type_name",
 		"book_type_img",
 	}
 	_, err := m.db.Cols(cols...).Where("book_type_id =?", booktype.BookTypeId).Update(booktype)
-	return err
+	return &booktype, err
 
 }

service/book/book.go

@@ -1,7 +1,10 @@
 package book
 
 import (
+	"errors"
 	"spaceofcheng/services/models/book"
+	"spaceofcheng/services/models/model"
+	"spaceofcheng/services/service"
 	"spaceofcheng/services/utils"
 )
 
@@ -18,3 +21,49 @@ func NewBookServ(ctx *utils.Context) *BookServ {
 		dao: book.NewBookDAO(ctx),
 	}
 }
+func (s *BookServ) GetBooKList(bookType, caseid string, page, pageSize int) (map[string]interface{}, error) {
+	if pageSize == 0 {
+		pageSize = service.PAGENUM
+	}
+	list, err := s.dao.GetBookList(bookType, caseid, page, pageSize)
+	if err != nil {
+		utils.LogError("获取图书列表失败: " + err.Error())
+		return nil, errors.New("获取图书列表失败")
+	}
+	total, err := s.dao.GetBookListCount(bookType, caseid)
+	if err != nil {
+		utils.LogError("获取图书列表失败: " + err.Error())
+		return nil, errors.New("获取图书列表失败")
+	}
+	return map[string]interface{}{
+		"list":     list,
+		"pageSize": pageSize,
+		"pagenum":  total,
+		"page":     page,
+	}, nil
+}
+func (s *BookServ) GetBookById(bookId string) (*model.TaBook, error) {
+	book, err := s.dao.GetBookById(bookId)
+	if err != nil {
+		utils.LogError("获取图书详情失败: " + err.Error())
+		return nil, errors.New("获取图书详情失败")
+	}
+	return book, nil
+}
+func (s *BookServ) AddBook(book model.TaBook) (*model.TaBook, error) {
+	newBook, err := s.dao.AddBook(book)
+	if err != nil {
+		utils.LogError("新增图书失败: " + err.Error())
+		return nil, errors.New("新增图书失败")
+	}
+	return newBook, err
+
+}
+func (s *BookServ) EditBook(book model.TaBook) error {
+	err := s.dao.EditBook(book)
+	if err != nil {
+		utils.LogError("修改图书失败: " + err.Error())
+		return errors.New("修改图书失败")
+	}
+	return nil
+}

service/booktype/booktype.go

@@ -53,11 +53,20 @@ func (s *BooktypeServ) GetBookTypeById(bookTypeId string) (*model.TaBookType, er
 	return bookType, nil
 }
 
-func (s *BooktypeServ) AddBookType(booktype model.TaBookType) (*model.TaBookType, error) {
-	bookType, err := s.dao.AddBookType(booktype)
-	if err != nil {
-		utils.LogError("新增图书分类失败: " + err.Error())
-		return nil, errors.New("新增图书分类失败")
+func (s *BooktypeServ) SaveBookType(booktype model.TaBookType) (*model.TaBookType, error) {
+	if booktype.BookTypeId == "" {
+		bookType, err := s.dao.AddBookType(booktype)
+		if err != nil {
+			utils.LogError("新增图书分类失败: " + err.Error())
+			return nil, errors.New("新增图书分类失败")
+		} else {
+			bookType, err := s.dao.EditBookType(booktype)
+			if err != nil {
+				utils.LogError("修改图书分类失败: " + err.Error())
+				return nil, errors.New("修改图书分类失败")
+			}
+		}
 	}
+
 	return bookType, nil
 }