added sql storage
This commit is contained in:
parent
c7f600f8e7
commit
5732f06932
13
api.go
13
api.go
|
@ -22,12 +22,6 @@ type JSONMessage struct {
|
||||||
Message string `json:"message"`
|
Message string `json:"message"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type LogMessage struct {
|
|
||||||
Level int `json:"level"`
|
|
||||||
Namespace string `json:"namespace"`
|
|
||||||
Payload interface{} `json:"payload"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func apiInfo(w http.ResponseWriter, req *http.Request) {
|
func apiInfo(w http.ResponseWriter, req *http.Request) {
|
||||||
version := struct {
|
version := struct {
|
||||||
Version string `json:"version"`
|
Version string `json:"version"`
|
||||||
|
@ -64,13 +58,12 @@ func put(w http.ResponseWriter, req *http.Request) {
|
||||||
}
|
}
|
||||||
parse_err := json.Unmarshal(body, &message)
|
parse_err := json.Unmarshal(body, &message)
|
||||||
if parse_err != nil {
|
if parse_err != nil {
|
||||||
if err := json.NewEncoder(w).Encode(JSONError{parse_err.Error()}); err != nil {
|
b, _ := json.Marshal(JSONError{err.Error()})
|
||||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
http.Error(w, string(b), http.StatusMethodNotAllowed)
|
||||||
}
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = json.Marshal(message.Payload)
|
_, err = message.save()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
b, _ := json.Marshal(JSONError{err.Error()})
|
b, _ := json.Marshal(JSONError{err.Error()})
|
||||||
http.Error(w, string(b), http.StatusMethodNotAllowed)
|
http.Error(w, string(b), http.StatusMethodNotAllowed)
|
||||||
|
|
|
@ -0,0 +1,25 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
func dbSetup(config *Configs) {
|
||||||
|
var err error
|
||||||
|
psqlConfig := fmt.Sprintf(
|
||||||
|
"dbname=%s host=%s sslmode=disable",
|
||||||
|
config.Dbname,
|
||||||
|
config.Dbhost,
|
||||||
|
)
|
||||||
|
db, err = sqlx.Open(
|
||||||
|
"postgres",
|
||||||
|
psqlConfig,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Issues connecting to db:", err)
|
||||||
|
} else {
|
||||||
|
log.Printf("connected to db %s", psqlConfig)
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,10 @@
|
||||||
|
DROP TABLE log;
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS
|
||||||
|
log (
|
||||||
|
log_id SERIAL,
|
||||||
|
log_time timestamp DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
namespace varchar(256) NULL,
|
||||||
|
level int NOT NULL DEFAULT 0,
|
||||||
|
payload json NOT NULL
|
||||||
|
);
|
|
@ -0,0 +1,37 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"encoding/json"
|
||||||
|
)
|
||||||
|
|
||||||
|
type LogMessage struct {
|
||||||
|
Namespace string `json:"namespace"`
|
||||||
|
Level int `json:"level"`
|
||||||
|
Payload interface{} `json:"payload"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *LogMessage) json() ([]byte, error) {
|
||||||
|
return json.Marshal(l.Payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *LogMessage) save() (sql.Result, error) {
|
||||||
|
j, err := l.json()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
params := map[string]interface{}{
|
||||||
|
"namespace": l.Namespace,
|
||||||
|
"level": l.Level,
|
||||||
|
"payload": j,
|
||||||
|
}
|
||||||
|
// TODO: actually make this return something useful as a first param, might
|
||||||
|
// need RETURNING log_id?
|
||||||
|
return db.NamedExecMap(`
|
||||||
|
INSERT INTO log
|
||||||
|
(namespace, level, payload)
|
||||||
|
VALUES
|
||||||
|
(:namespace, :level, :payload)`,
|
||||||
|
params,
|
||||||
|
)
|
||||||
|
}
|
7
main.go
7
main.go
|
@ -3,13 +3,17 @@ package main
|
||||||
import (
|
import (
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
"github.com/kelseyhightower/envconfig"
|
"github.com/kelseyhightower/envconfig"
|
||||||
|
_ "github.com/lib/pq"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
var verbose = flag.Bool("verbose", false, "increase logging verbosity")
|
var verbose = flag.Bool("verbose", false, "increase logging verbosity")
|
||||||
|
|
||||||
|
var db *sqlx.DB
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
||||||
|
@ -21,7 +25,8 @@ func main() {
|
||||||
if err := config.validate(); err != nil {
|
if err := config.validate(); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
log.Printf("%+v", config)
|
log.Printf("config: %#v", config)
|
||||||
|
dbSetup(config)
|
||||||
|
|
||||||
http.Handle("/api/v0/info/", JsonHandler(apiInfo))
|
http.Handle("/api/v0/info/", JsonHandler(apiInfo))
|
||||||
http.Handle("/api/v0/put/", JsonHandler(put))
|
http.Handle("/api/v0/put/", JsonHandler(put))
|
||||||
|
|
Loading…
Reference in New Issue