下载相机图片功能
This commit is contained in:
dongchangxi committed
1 parent
ff9315ba63
commit
083857d130
13 files changed
+929
-25
No files matched your search
@@ -0,0 +1,220 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"changeme/model"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
// DB 结构体
|
||||
type DB struct {
|
||||
*sql.DB
|
||||
}
|
||||
|
||||
var (
|
||||
instance *DB
|
||||
once sync.Once
|
||||
)
|
||||
var TableName = "photo_studio_shoot_order"
|
||||
// GetInstance 返回数据库单例实例
|
||||
func GetInstance() *DB {
|
||||
once.Do(func() {
|
||||
db, err := sql.Open("sqlite3", "test.db")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
instance = &DB{db}
|
||||
//检测表是否存在
|
||||
checkTableSQL := `SELECT count(*) FROM sqlite_master WHERE type='table' AND name='`+TableName+`';`
|
||||
var count int
|
||||
if err := instance.QueryRow(checkTableSQL).Scan(&count); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if count > 0 {
|
||||
return
|
||||
}
|
||||
// 创建表
|
||||
createTableSQL := `CREATE TABLE "`+TableName+`" (
|
||||
"id" INTEGER NOT NULL,
|
||||
"pid" INTEGER NOT NULL,
|
||||
"customer_id" INTEGER NOT NULL,
|
||||
"customer_name" TEXT NOT NULL,
|
||||
"status" INTEGER NOT NULL,
|
||||
"order_id" INTEGER NOT NULL,
|
||||
"digital_order_id" INTEGER NOT NULL,
|
||||
"digital_type" INTEGER NOT NULL,
|
||||
"cover_path" TEXT NOT NULL,
|
||||
"cover_rotate_deg" INTEGER NOT NULL,
|
||||
"shoot_at_count" INTEGER NOT NULL,
|
||||
"shoot_done_count" INTEGER NOT NULL,
|
||||
"upload_cover_count" INTEGER NOT NULL,
|
||||
"upload_highmode_count" INTEGER NOT NULL,
|
||||
"upload_yuv_count" INTEGER NOT NULL,
|
||||
"first_shoot_at" DATETIME NULL,
|
||||
"first_shoot_done_at" DATETIME NULL,
|
||||
"last_shoot_at" DATETIME NULL,
|
||||
"last_shoot_done_at" DATETIME NULL,
|
||||
"check_status" INTEGER NOT NULL,
|
||||
"check_fail_reason" TEXT NOT NULL,
|
||||
"check_at" DATETIME NULL,
|
||||
"shop_id" INTEGER NOT NULL,
|
||||
"eq_id" INTEGER NOT NULL,
|
||||
"master_ver" TEXT NOT NULL,
|
||||
"eq_type" INTEGER NOT NULL,
|
||||
"created_at" DATETIME NULL,
|
||||
"upload_done_at" DATETIME NULL,
|
||||
"minipreview_upload_done_at" DATETIME NULL,
|
||||
"updated_at" DATETIME NULL,
|
||||
"is_privary" INTEGER NOT NULL,
|
||||
"cancel_at" DATETIME NULL,
|
||||
PRIMARY KEY ("id")
|
||||
);`
|
||||
if _, err = instance.Exec(createTableSQL); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
})
|
||||
return instance
|
||||
}
|
||||
|
||||
|
||||
//Add 新增数据
|
||||
func (db *DB) Add(addData *model.AddData) error {
|
||||
v := reflect.ValueOf(addData)
|
||||
// 确保传入的参数是指向结构体的指针
|
||||
if v.Kind() != reflect.Ptr || v.Elem().Kind() != reflect.Struct {
|
||||
return errors.New("invalid input")
|
||||
}
|
||||
// 获取结构体的类型和字段
|
||||
t := v.Elem().Type()
|
||||
fields := []string{}
|
||||
placeholders := []string{}
|
||||
values := []interface{}{}
|
||||
|
||||
for i := 0; i < v.Elem().NumField(); i++ {
|
||||
field := t.Field(i)
|
||||
value := v.Elem().Field(i)
|
||||
// 使用 db 标签作为字段名
|
||||
dbTag := field.Tag.Get("json")
|
||||
fields = append(fields, dbTag)
|
||||
placeholders = append(placeholders, "?")
|
||||
values = append(values, value.Interface())
|
||||
}
|
||||
// 生成 SQL 语句
|
||||
sqlStatement := fmt.Sprintf("INSERT INTO %s (%s) VALUES (%s)",
|
||||
TableName,
|
||||
strings.Join(fields, ", "),
|
||||
strings.Join(placeholders, ", "))
|
||||
// 执行 SQL 语句
|
||||
_, err := db.Exec(sqlStatement, values...)
|
||||
return err
|
||||
}
|
||||
|
||||
//根据查询列表数据
|
||||
// List 方法
|
||||
func (db *DB) List(req *model.PageReqData) (list []*model.Data, err error) {
|
||||
// 基本的 SQL 查询语句
|
||||
query := "SELECT cover_path,pid,customer_name FROM "+TableName+" WHERE 1=1"
|
||||
var args []interface{}
|
||||
|
||||
// 根据请求参数动态构建查询条件
|
||||
if req.Pid > 0 {
|
||||
query += " AND pid = ?"
|
||||
args = append(args, req.Pid)
|
||||
}
|
||||
if req.CustomerId > 0 {
|
||||
query += " AND customer_id = ?"
|
||||
args = append(args, req.CustomerId)
|
||||
}
|
||||
if req.CustomerName != "" {
|
||||
query += " AND customer_name LIKE ?"
|
||||
args = append(args, "%"+req.CustomerName+"%")
|
||||
}
|
||||
if req.Status > 0 {
|
||||
query += " AND status = ?"
|
||||
args = append(args, req.Status)
|
||||
}
|
||||
|
||||
// 添加分页条件
|
||||
if req.Page <= 0 {
|
||||
req.Page = 1
|
||||
}
|
||||
if req.Size <= 0 {
|
||||
req.Size = 20
|
||||
}
|
||||
offset := (req.Page - 1) * req.Size
|
||||
query += fmt.Sprintf(" LIMIT %d OFFSET %d", req.Size, offset)
|
||||
// 执行查询
|
||||
rows, err := db.Query(query, args...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
// 读取查询结果
|
||||
for rows.Next() {
|
||||
var addData model.Data
|
||||
if err := rows.Scan(&addData.CoverPath, &addData.Pid, &addData.CustomerName); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
list = append(list, &addData)
|
||||
}
|
||||
return list, nil
|
||||
}
|
||||
|
||||
func (db *DB) Delete(req *model.DeleteReqData) (err error) {
|
||||
if req.Pid == 0 && req.Id == 0 {
|
||||
return errors.New("参数不能为空")
|
||||
}
|
||||
query := "DELETE FROM "+TableName+" WHERE 1=1"
|
||||
var args []interface{}
|
||||
if req.Pid > 0 {
|
||||
query += " AND pid = ?"
|
||||
args = append(args, req.Pid)
|
||||
}
|
||||
if req.Id > 0 {
|
||||
query += " AND id = ?"
|
||||
args = append(args, req.Id)
|
||||
}
|
||||
_, err = db.Exec(query, args...)
|
||||
return err
|
||||
}
|
||||
|
||||
func (db *DB) Update(addData *model.Data) error {
|
||||
v := reflect.ValueOf(addData)
|
||||
// 确保传入的参数是指向结构体的指针
|
||||
if v.Kind() != reflect.Ptr || v.Elem().Kind() != reflect.Struct {
|
||||
return errors.New("invalid input")
|
||||
}
|
||||
|
||||
// 获取结构体的类型和字段
|
||||
t := v.Elem().Type()
|
||||
fields := []string{}
|
||||
values := []interface{}{}
|
||||
|
||||
for i := 0; i < v.Elem().NumField(); i++ {
|
||||
field := t.Field(i)
|
||||
value := v.Elem().Field(i)
|
||||
// 使用 db 标签作为字段名
|
||||
dbTag := field.Tag.Get("json")
|
||||
if gconv.String(value.Interface()) != "" && gconv.String(value.Interface()) != "0" && dbTag != "id" {
|
||||
fields = append(fields, fmt.Sprintf("%s = ?", dbTag))
|
||||
values = append(values, value.Interface())
|
||||
}
|
||||
}
|
||||
// 生成 Update SQL 语句
|
||||
sqlStatement := fmt.Sprintf("UPDATE %s SET %s WHERE id = %s",
|
||||
TableName,
|
||||
strings.Join(fields, ", "),
|
||||
gconv.String(addData.Id),
|
||||
)
|
||||
// 执行 SQL 语句
|
||||
_, err := db.Exec(sqlStatement, values...)
|
||||
return err
|
||||
}
|
||||
Reference in new issue
Block a user