Add import command (temporary)

This command only available during migration process and will be removed
in future versions.
This commit is contained in:
ditatompel 2024-05-06 23:53:23 +07:00
parent 3b0420d950
commit 2c7e1a4fdb
No known key found for this signature in database
GPG key ID: 31D3D06D77950979

97
cmd/import.go Normal file
View file

@ -0,0 +1,97 @@
package cmd
import (
"encoding/json"
"fmt"
"net/http"
"github.com/ditatompel/xmr-nodes/internal/database"
"github.com/spf13/cobra"
)
type importClient struct {
db *database.DB
}
func newImport(db *database.DB) *importClient {
return &importClient{db: db}
}
type importData struct {
Hostname string `json:"hostname"`
Port int `json:"port"`
Protocol string `json:"protocol"`
IsTor bool `json:"is_tor"`
}
var importCmd = &cobra.Command{
Use: "import",
Short: "Import Monero nodes from old API",
Long: `Import Monero nodes from old API.
This command only available during migration process and will be removed in future versions.`,
Run: func(_ *cobra.Command, _ []string) {
if err := database.ConnectDB(); err != nil {
panic(err)
}
req, err := http.NewRequest(http.MethodGet, "https://api.ditatompel.com/monero/remote-node?limit=500", nil)
if err != nil {
panic(err)
}
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
panic(err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
panic(fmt.Errorf("status code: %d", resp.StatusCode))
}
response := struct {
Data []importData `json:"data"`
}{}
err = json.NewDecoder(resp.Body).Decode(&response)
if err != nil {
panic(err)
}
action := newImport(database.GetDB())
for _, node := range response.Data {
action.processData(node)
}
fmt.Println("Total Source Data: ", len(response.Data))
fmt.Println("Done!")
},
}
func init() {
rootCmd.AddCommand(importCmd)
}
func (i *importClient) processData(node importData) error {
query := `SELECT id FROM tbl_node WHERE hostname = ? AND port = ? AND protocol = ?`
row := i.db.QueryRow(query, node.Hostname, node.Port, node.Protocol)
var id int
err := row.Scan(&id)
if err == nil {
// fmt.Printf("Skipping %s://%s:%d\n", node.Protocol, node.Hostname, node.Port)
return fmt.Errorf("node already exists")
}
// insert
query = `INSERT INTO tbl_node (hostname, port, protocol, is_tor, nettype, ip_addr, last_check_status) VALUES (?, ?, ?, ?, ?, ?, ?)`
_, err = i.db.Exec(query, node.Hostname, node.Port, node.Protocol, node.IsTor, "", "", "[2,2,2,2,2]")
if err != nil {
return err
}
fmt.Printf("Imported %s://%s:%d\n", node.Protocol, node.Hostname, node.Port)
return nil
}