kevin
c16a8dfbc4
- Go + Gin + html/template 服务端渲染 - 主页：Google 风格搜索框 + 导航卡片 - 后台：卡片 CRUD、搜索引擎配置、主页背景/标题配置 - 图片上传：支持 jpg/jpeg/png/gif，自动压缩，缩略图参数 ?thumb=1 - 安全：登录日志、修改密码、IP 自动封禁、IP 白名单 - 访问统计：主页访问/卡片点击/搜索追踪、实时流量、IP 统计 - SQLite 存储（modernc.org/sqlite，纯 Go） - 内存 Session + bcrypt 密码哈希
74 lines
2.0 KiB
Go
74 lines
2.0 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"simple_portal/database"
|
|
)
|
|
|
|
// LoginLog 表示一条登录日志记录。
|
|
type LoginLog struct {
|
|
ID int `json:"id"`
|
|
AdminID *int `json:"admin_id"`
|
|
Username string `json:"username"`
|
|
IP string `json:"ip"`
|
|
UserAgent string `json:"user_agent"`
|
|
Success bool `json:"success"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// CreateLoginLog 插入一条登录日志。
|
|
func CreateLoginLog(adminID *int, username, ip, userAgent string, success bool) error {
|
|
_, err := database.DB.Exec(
|
|
"INSERT INTO login_logs (admin_id, username, ip, user_agent, success, created_at) VALUES (?, ?, ?, ?, ?, ?)",
|
|
adminID, username, ip, userAgent, success, time.Now().Format("2006-01-02 15:04:05"),
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create login log: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// LoginLogCount 返回登录日志总数。
|
|
func LoginLogCount() (int, error) {
|
|
var count int
|
|
err := database.DB.QueryRow("SELECT COUNT(*) FROM login_logs").Scan(&count)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("failed to count login logs: %w", err)
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
// GetLoginLogs 分页获取登录日志，倒序排列。
|
|
func GetLoginLogs(page, pageSize int) ([]LoginLog, int, error) {
|
|
total, err := LoginLogCount()
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
offset := (page - 1) * pageSize
|
|
if offset < 0 {
|
|
offset = 0
|
|
}
|
|
|
|
rows, err := database.DB.Query(
|
|
"SELECT id, admin_id, username, ip, user_agent, success, created_at FROM login_logs ORDER BY created_at DESC LIMIT ? OFFSET ?",
|
|
pageSize, offset,
|
|
)
|
|
if err != nil {
|
|
return nil, 0, fmt.Errorf("failed to query login logs: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var logs []LoginLog
|
|
for rows.Next() {
|
|
var l LoginLog
|
|
if err := rows.Scan(&l.ID, &l.AdminID, &l.Username, &l.IP, &l.UserAgent, &l.Success, &l.CreatedAt); err != nil {
|
|
return nil, 0, fmt.Errorf("failed to scan login log: %w", err)
|
|
}
|
|
logs = append(logs, l)
|
|
}
|
|
return logs, total, nil
|
|
}
|