pm/models/course.go

package models

import (
	"errors"
	"zhiyuan/pkg/db"
)

type Course struct {
	ID        int    `json:"id"`
	Name      string `json:"name"`
	Intro     string `json:"intro"`
	Type      string `json:"type"`
	RoleIds   string `json:"role_ids"`
	Checks    int    `json:"checks"`
	Show      int    `json:"show"`
	CreatedAt int    `json:"created_at"`
	UpdatedAt int    `json:"updated_at"`
}

func (Course) TableName() string {
	return "zy_course"
}

func (a Course) GetOne(where map[string]interface{}, fields []string, retVal interface{}) (*Course, error) {
	if retVal == nil {
		var course *Course
		err := db.GetOne(a.TableName(), where, fields, &course)
		return course, err
	} else {
		err := db.GetOne(a.TableName(), where, fields, retVal)
		return nil, err
	}
}

func (a Course) GetMulti(where map[string]interface{}, fields []string, retVal interface{}) ([]*Course, error) {
	if retVal == nil {
		var course []*Course
		err := db.GetMulti(a.TableName(), where, fields, &course)
		return course, err
	} else {
		err := db.GetMulti(a.TableName(), where, fields, retVal)
		return nil, err
	}
}

func (a Course) GetOneRaw(where map[string]string, param map[string]interface{}, retVal interface{}) (*Course, error) {
	course, err := a.GetMultiRaw(where, param, retVal)
	if err != nil {
		return nil, err
	}
	if len(course) == 0 {
		return nil, errors.New("empty record2")
	}
	return course[0], nil
}

func (a Course) GetMultiRaw(where map[string]string, param map[string]interface{}, retVal interface{}) ([]*Course, error) {
	field := "SELECT * FROM zy_course "
	if retVal == nil {
		var course []*Course
		err := db.GetMultiRaw(field, where, param, &course)
		return course, err
	} else {
		err := db.GetMultiRaw(field, where, param, retVal)
		return nil, err
	}
}