Minor changes and fixes

This commit is contained in:
David J. Allen 2024-05-07 16:51:07 -06:00
parent 6bf9035d96
commit b873613939
No known key found for this signature in database
GPG key ID: 717C593FF60A2ACC
4 changed files with 17 additions and 6 deletions

View file

@ -1,6 +1,8 @@
package cmd package cmd
import ( import (
"fmt"
magellan "github.com/OpenCHAMI/magellan/internal" magellan "github.com/OpenCHAMI/magellan/internal"
"github.com/OpenCHAMI/magellan/internal/api/smd" "github.com/OpenCHAMI/magellan/internal/api/smd"
"github.com/OpenCHAMI/magellan/internal/db/sqlite" "github.com/OpenCHAMI/magellan/internal/db/sqlite"
@ -37,6 +39,10 @@ var collectCmd = &cobra.Command{
} }
} }
if verbose {
fmt.Printf("access token: %v\n", accessToken)
}
// //
if threads <= 0 { if threads <= 0 {
threads = mathutil.Clamp(len(probeStates), 1, 255) threads = mathutil.Clamp(len(probeStates), 1, 255)

View file

@ -68,7 +68,11 @@ func LoadAccessToken() (string, error) {
} }
// TODO: try to load token from config // TODO: try to load token from config
return "", fmt.Errorf("could not load from environment variable or file") testToken = viper.GetString("access_token")
if testToken != "" {
return testToken, nil
}
return "", fmt.Errorf("could not load token from environment variable, file, or config")
} }
func init() { func init() {

View file

@ -18,6 +18,7 @@ func CreateProbeResultsIfNotExists(path string) (*sqlx.DB, error) {
PRIMARY KEY (host, port) PRIMARY KEY (host, port)
); );
` `
// TODO: it may help with debugging to check for file permissions here first
db, err := sqlx.Open("sqlite3", path) db, err := sqlx.Open("sqlite3", path)
if err != nil { if err != nil {
return nil, fmt.Errorf("could not open database: %v", err) return nil, fmt.Errorf("could not open database: %v", err)