Запрос к базе данных для базовой аутентификации с использованием go-http-auth с помощью martini-go

StackOverflow https://stackoverflow.com//questions/23006639

Вопрос

Я пытаюсь использовать go-http-auth с martini-go.В приведенном здесь примере - https://github.com/abbot/go-http-auth

package main

import (
        auth "github.com/abbot/go-http-auth"
        "fmt"
        "net/http"
)

func Secret(user, realm string) string {
        if user == "john" {
                // password is "hello"
                return "$1$dlPL2MqE$oQmn16q49SqdmhenQuNgs1"
        }
        return ""
}

func handle(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
        fmt.Fprintf(w, "<html><body><h1>Hello, %s!</h1></body></html>", r.Username)
}


func main() {
    db, err := sql.Open("postgres", "postgres://blabla:blabla@localhost/my_db")
    authenticator := auth.NewBasicAuthenticator("example.com", Secret)
    m := martini.Classic()
    m.Map(db)
    m.Get("/users", authenticator.Wrap(MyUserHandler))
    m.Run()

}

Секретная функция использует жестко запрограммированного пользователя «john».

Аутентификация прошла успешно, когда я выполняю

curl --user john:hello localhost:3000/users

Очевидно, что это тривиальный пример с жестко закодированными именем пользователя и паролем.

сейчас меняю Secret функция для этого

func Secret(user, realm string) string {

    fmt.Println("Executing Secret")

    var db *sql.DB

    var (
        username string
        password string
    )

    err := db.QueryRow("select username, password from users where username = ?", user).Scan(&username, &password)

    if err == sql.ErrNoRows {
        return ""
    }

    if err != nil {
        log.Fatal(err)
    }
    //if user == "john" {
        //// password is "hello"
        //return "$1$dlPL2MqE$oQmn16q49SqdmhenQuNgs1"
    //}
    //return ""
    return ""

}

Но это терпит неудачу с PANIC: runtime error: invalid memory address or nil pointer dereference. Очевидно, это потому, что я пытаюсь создать экземпляр var db *sql.DB в Secret функция.я не могу пройти db *sql.DB в Secret функционировать либо потому, что auth.BasicNewAuthentication ожидает Secret аргумент, соответствующий type func (string, string) string.

Как я могу правильно реализовать запрос к базе данных и вернуть пароль для сравнения?

Это было полезно?

Решение

Вы можете использовать простое замыкание, чтобы передать ссылку на вашу БД в функцию аутентификации:

authenticator := auth.NewBasicAuthenticator("example.com", func(user, realm string) string {
    return Secret(db, user, realm)
})

…а затем измените свой Secret принять базу данных в качестве первого аргумента:

func Secret(db *sql.DB, user, realm string) string {
    // do your db lookup here…
}

Другие советы

Альтернативный подход к ответу Аттилы.Вы можете определить структуру, определить Secret() обработчик и передать только указанную функцию (go сохраняет ссылку на «владельца») в authhandler.

type SecretDb struct {
  db *DB
}
func (db *SecretDb) Secret(user, realm string) string {
 // .. use db.db here
}


func main() {
   secretdb = SecretDb{db}
   ...
   auth.NewBasicAuthenticator("example.com", secretdb.Secret)
   ...
}
Лицензировано под: CC-BY-SA с атрибуция
Не связан с StackOverflow
scroll top