✏️ 分离 data

data/dl.go

@@ -1,21 +0,0 @@
-package data
-
-import (
-	"io"
-	"net/http"
-	"os"
-)
-
-func DownloadTo(url, file string) error {
-	resp, err := http.Get(url)
-	if err == nil {
-		var f *os.File
-		f, err = os.Create(file)
-		if err == nil {
-			_, err = io.Copy(f, resp.Body)
-			resp.Body.Close()
-			f.Close()
-		}
-	}
-	return err
-}

data/math.go

@@ -1,17 +0,0 @@
-package data
-
-// min 返回两数最大值，该函数将被内联
-func Max(a, b int) int {
-	if a > b {
-		return a
-	}
-	return b
-}
-
-// min 返回两数最小值，该函数将被内联
-func Min(a, b int) int {
-	if a > b {
-		return b
-	}
-	return a
-}

data/sqlite.go

@@ -1,278 +0,0 @@
-// Package data 数据库/数据处理相关工具
-package data
-
-import (
-	"database/sql"
-	"reflect"
-	"strings"
-
-	_ "modernc.org/sqlite" // 引入sqlite
-)
-
-// Sqlite 数据库对象
-type Sqlite struct {
-	DB     *sql.DB
-	DBPath string
-}
-
-// Create 生成数据库
-// 默认结构体的第一个元素为主键
-// 返回错误
-func (db *Sqlite) Create(table string, objptr interface{}) (err error) {
-	if db.DB == nil {
-		database, err := sql.Open("sqlite", db.DBPath)
-		if err != nil {
-			return err
-		}
-		db.DB = database
-	}
-	var (
-		tags  = tags(objptr)
-		kinds = kinds(objptr)
-		top   = len(tags) - 1
-		cmd   = []string{}
-	)
-	cmd = append(cmd, "CREATE TABLE IF NOT EXISTS")
-	cmd = append(cmd, table)
-	cmd = append(cmd, "(")
-	for i := range tags {
-		cmd = append(cmd, tags[i])
-		cmd = append(cmd, kinds[i])
-		switch i {
-		default:
-			cmd = append(cmd, "NULL,")
-		case 0:
-			cmd = append(cmd, "PRIMARY KEY")
-			cmd = append(cmd, "NOT NULL,")
-		case top:
-			cmd = append(cmd, "NULL);")
-		}
-	}
-	if _, err := db.DB.Exec(strings.Join(cmd, " ")); err != nil {
-		return err
-	}
-	return nil
-}
-
-// Insert 插入数据集
-// 默认结构体的第一个元素为主键
-// 返回错误
-func (db *Sqlite) Insert(table string, objptr interface{}) (err error) {
-	rows, err := db.DB.Query("SELECT * FROM " + table)
-	if err != nil {
-		return err
-	}
-	if rows.Err() != nil {
-		return rows.Err()
-	}
-	tags, _ := rows.Columns()
-	rows.Close()
-	var (
-		values = values(objptr)
-		top    = len(tags) - 1
-		cmd    = []string{}
-	)
-	cmd = append(cmd, "REPLACE INTO")
-	cmd = append(cmd, table)
-	for i := range tags {
-		switch i {
-		default:
-			cmd = append(cmd, tags[i])
-			cmd = append(cmd, ",")
-		case 0:
-			cmd = append(cmd, "(")
-			cmd = append(cmd, tags[i])
-			cmd = append(cmd, ",")
-		case top:
-			cmd = append(cmd, tags[i])
-			cmd = append(cmd, ")")
-		}
-	}
-	for i := range tags {
-		switch i {
-		default:
-			cmd = append(cmd, "?")
-			cmd = append(cmd, ",")
-		case 0:
-			cmd = append(cmd, "VALUES (")
-			cmd = append(cmd, "?")
-			cmd = append(cmd, ",")
-		case top:
-			cmd = append(cmd, "?")
-			cmd = append(cmd, ")")
-		}
-	}
-	stmt, err := db.DB.Prepare(strings.Join(cmd, " "))
-	if err != nil {
-		return err
-	}
-	_, err = stmt.Exec(values...)
-	if err != nil {
-		return err
-	}
-	return nil
-}
-
-// Find 查询数据库
-// condition 可为"WHERE id = 0"
-// 默认字段与结构体元素顺序一致
-// 返回错误
-func (db *Sqlite) Find(table string, objptr interface{}, condition string) (err error) {
-	var cmd = []string{}
-	cmd = append(cmd, "SELECT * FROM ")
-	cmd = append(cmd, table)
-	cmd = append(cmd, condition)
-	rows, err := db.DB.Query(strings.Join(cmd, " "))
-	if err != nil {
-		return err
-	}
-	if rows.Err() != nil {
-		return rows.Err()
-	}
-	defer rows.Close()
-
-	for rows.Next() {
-		if err != nil {
-			return err
-		}
-		err = rows.Scan(addrs(objptr)...)
-		if err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-// ListTables 列出所有表名
-// 返回所有表名+错误
-func (db *Sqlite) ListTables() (s []string, err error) {
-	rows, err := db.DB.Query("SELECT name FROM sqlite_master where type='table' order by name;")
-	if err != nil {
-		return
-	}
-	if rows.Err() != nil {
-		return nil, rows.Err()
-	}
-	defer rows.Close()
-
-	for rows.Next() {
-		if err != nil {
-			return
-		}
-		objptr := new(string)
-		err = rows.Scan(objptr)
-		if err == nil {
-			s = append(s, *objptr)
-		}
-	}
-	return
-}
-
-// Del 删除数据库
-// condition 可为"WHERE id = 0"
-// 返回错误
-func (db *Sqlite) Del(table string, condition string) (err error) {
-	var cmd = []string{}
-	cmd = append(cmd, "DELETE FROM")
-	cmd = append(cmd, table)
-	cmd = append(cmd, condition)
-	stmt, err := db.DB.Prepare(strings.Join(cmd, " "))
-	if err != nil {
-		return err
-	}
-	_, err = stmt.Exec()
-	if err != nil {
-		return err
-	}
-	return nil
-}
-
-// Count 查询数据库行数
-// 返回行数以及错误
-func (db *Sqlite) Count(table string) (num int, err error) {
-	var cmd = []string{}
-	cmd = append(cmd, "SELECT * FROM")
-	cmd = append(cmd, table)
-	rows, err := db.DB.Query(strings.Join(cmd, " "))
-	if err != nil {
-		return num, err
-	}
-	if rows.Err() != nil {
-		return num, rows.Err()
-	}
-	defer rows.Close()
-	for rows.Next() {
-		num++
-	}
-	return num, nil
-}
-
-// tags 反射 返回结构体对象的 tag 数组
-func tags(objptr interface{}) []string {
-	var tags []string
-	elem := reflect.ValueOf(objptr).Elem()
-	// 判断第一个元素是否为匿名字段
-	if elem.Type().Field(0).Anonymous {
-		elem = elem.Field(0)
-	}
-	for i, flen := 0, elem.Type().NumField(); i < flen; i++ {
-		tags = append(tags, elem.Type().Field(i).Tag.Get("db"))
-	}
-	return tags
-}
-
-// kinds 反射 返回结构体对象的 kinds 数组
-func kinds(objptr interface{}) []string {
-	var kinds []string
-	elem := reflect.ValueOf(objptr).Elem()
-	// 判断第一个元素是否为匿名字段
-	if elem.Type().Field(0).Anonymous {
-		elem = elem.Field(0)
-	}
-	for i, flen := 0, elem.Type().NumField(); i < flen; i++ {
-		switch elem.Field(i).Type().String() {
-		case "int64":
-			kinds = append(kinds, "INT")
-		case "string":
-			kinds = append(kinds, "TEXT")
-		default:
-			kinds = append(kinds, "TEXT")
-		}
-	}
-	return kinds
-}
-
-// values 反射 返回结构体对象的 values 数组
-func values(objptr interface{}) []interface{} {
-	var values []interface{}
-	elem := reflect.ValueOf(objptr).Elem()
-	// 判断第一个元素是否为匿名字段
-	if elem.Type().Field(0).Anonymous {
-		elem = elem.Field(0)
-	}
-	for i, flen := 0, elem.Type().NumField(); i < flen; i++ {
-		switch elem.Field(i).Type().String() {
-		case "int64":
-			values = append(values, elem.Field(i).Int())
-		case "string":
-			values = append(values, elem.Field(i).String())
-		default:
-			values = append(values, elem.Field(i).String())
-		}
-	}
-	return values
-}
-
-// addrs 反射 返回结构体对象的 addrs 数组
-func addrs(objptr interface{}) []interface{} {
-	var addrs []interface{}
-	elem := reflect.ValueOf(objptr).Elem()
-	// 判断第一个元素是否为匿名字段
-	if elem.Type().Field(0).Anonymous {
-		elem = elem.Field(0)
-	}
-	for i, flen := 0, elem.Type().NumField(); i < flen; i++ {
-		addrs = append(addrs, elem.Field(i).Addr().Interface())
-	}
-	return addrs
-}