This repository has been archived on 2024-07-22. You can view files and clone it, but cannot push or open issues or pull requests.
e621-sdk-go/main.go

51 lines
983 B
Go
Raw Normal View History

2023-05-22 11:08:08 +00:00
package main
import (
"context"
2023-05-24 14:05:27 +00:00
"e621_to_neo4j/api"
"e621_to_neo4j/e621"
"e621_to_neo4j/neo4jAPI"
2023-05-22 11:08:08 +00:00
"e621_to_neo4j/utils"
"github.com/neo4j/neo4j-go-driver/v5/neo4j"
2023-05-22 11:08:08 +00:00
"log"
2023-05-24 14:05:27 +00:00
"net/http"
2023-05-22 11:08:08 +00:00
)
2023-05-24 14:05:27 +00:00
const USER = "selloo"
2023-05-22 11:08:08 +00:00
func main() {
2023-05-24 14:05:27 +00:00
// Loads Config
2023-05-22 11:08:08 +00:00
config, err := utils.LoadConfig()
if err != nil {
log.Println(err)
}
2023-05-24 14:05:27 +00:00
// Connect to Neo4j DB
driver, err := neo4jAPI.NewConnection(config.Neo4jURL, config.Neo4jUsername, config.Neo4jPassword)
2023-05-22 11:08:08 +00:00
if err != nil {
log.Fatal(err)
}
ctx := context.Background()
defer func(driver neo4j.DriverWithContext, ctx context.Context) {
err := driver.Close(ctx)
if err != nil {
log.Fatal(err)
}
}(driver, ctx)
2023-05-22 11:08:08 +00:00
2023-05-24 14:05:27 +00:00
// Initialize the e621API
e621Client := e621.NewClient(config.E621APIKey, config.E621Username)
2023-05-22 11:08:08 +00:00
2023-05-24 14:05:27 +00:00
// Register the UserHandler with the "/user" route
http.HandleFunc("/user", api.UserHandler(ctx, driver, e621Client))
2023-05-22 11:08:08 +00:00
2023-05-24 14:05:27 +00:00
// Start the HTTP server
err = http.ListenAndServe(":8080", nil)
if err != nil {
2023-05-24 14:05:27 +00:00
return
}
}