Golang プラグインを詳しく見る: 5 つのプラグインの機能と特徴を調べる

package httpplugin

import (
    "net/http"
)

type Plugin struct {
    // 自定义属性
}

func NewPlugin() *Plugin {
    return &Plugin{}
}

func (p *Plugin) ServeHTTP(w http.ResponseWriter, r *http.Request, next http.Handler) {
    // 在这里实现插件的具体逻辑
    // 可以进行路由转发、参数解析、请求验证等操作
    // 例如：判断当前请求是否需要进行权限认证
    if needAuth(r) {
        // 进行权限认证
        if !checkAuth(r) {
            // 权限不足，直接返回错误信息
            w.WriteHeader(http.StatusForbidden)
            w.Write([]byte("Permission denied"))
            return
        }
    }

    // 继续处理下一个中间件或目标handler
    next.ServeHTTP(w, r)
}

package dbplugin

import (
    "database/sql"
)

type Plugin struct {
    db *sql.DB
    // 自定义属性
}

func NewPlugin(db *sql.DB) *Plugin {
    return &Plugin{db: db}
}

func (p *Plugin) Query(sql string, args ...interface{}) (*sql.Rows, error) {
    // 在这里实现插件的具体逻辑
    // 执行数据库查询操作
    return p.db.Query(sql, args...)
}

func (p *Plugin) Exec(sql string, args ...interface{}) (sql.Result, error) {
    // 在这里实现插件的具体逻辑
    // 执行数据库写操作
    return p.db.Exec(sql, args...)
}

package logplugin

import (
    "log"
)

type Plugin struct {
    // 自定义属性
}

func NewPlugin() *Plugin {
    return &Plugin{}
}

func (p *Plugin) Info(msg string) {
    // 在这里实现插件的具体逻辑
    // 记录信息级别的日志
    log.Println("[INFO]", msg)
}

func (p *Plugin) Error(err error) {
    // 在这里实现插件的具体逻辑
    // 记录错误级别的日志
    log.Println("[ERROR]", err.Error())
}

package authplugin

import (
    "net/http"
)

type Plugin struct {
    // 自定义属性
}

func NewPlugin() *Plugin {
    return &Plugin{}
}

func (p *Plugin) ServeHTTP(w http.ResponseWriter, r *http.Request, next http.Handler) {
    // 在这里实现插件的具体逻辑
    // 判断当前请求是否需要进行权限认证
    if needAuth(r) {
        // 进行权限认证
        if !checkAuth(r) {
            // 权限不足，直接返回错误信息
            w.WriteHeader(http.StatusForbidden)
            w.Write([]byte("Permission denied"))
            return
        }
    }

    // 继续处理下一个中间件或目标handler
    next.ServeHTTP(w, r)
}

package cacheplugin

import (
    "time"
)

type Plugin struct {
    cache map[string]interface{}
    // 自定义属性
}

func NewPlugin() *Plugin {
    return &Plugin{cache: make(map[string]interface{})}
}

func (p *Plugin) Get(key string) (interface{}, bool) {
    // 在这里实现插件的具体逻辑
    // 查询缓存数据
    val, ok := p.cache[key]
    return val, ok
}

func (p *Plugin) Set(key string, value interface{}, expire time.Duration) {
    // 在这里实现插件的具体逻辑
    // 设置缓存数据
    p.cache[key] = value
    // 设置过期时间
    time.AfterFunc(expire, func() {
        delete(p.cache, key)
    })
}