working filesystem storage
This commit is contained in:
parent
808790a7c3
commit
b750d88d48
@ -3,7 +3,7 @@ surreal_url = "localhost:8000"
|
|||||||
surreal_username = "root"
|
surreal_username = "root"
|
||||||
surreal_password = "root"
|
surreal_password = "root"
|
||||||
surreal_ns = "test"
|
surreal_ns = "test"
|
||||||
surreal_db = "v1.19.1"
|
surreal_db = "v1.19.2"
|
||||||
|
|
||||||
# Crawler config
|
# Crawler config
|
||||||
crawl_filter = "en.wikipedia.com"
|
crawl_filter = "en.wikipedia.com"
|
||||||
|
@ -14,22 +14,6 @@ services:
|
|||||||
- --pass
|
- --pass
|
||||||
- root
|
- root
|
||||||
- rocksdb:/mydata/database.db
|
- rocksdb:/mydata/database.db
|
||||||
minio:
|
|
||||||
image: quay.io/minio/minio
|
|
||||||
ports:
|
|
||||||
- 9000:9000
|
|
||||||
- 9001:9001
|
|
||||||
environment:
|
|
||||||
- MINIO_ROOT_USER=root
|
|
||||||
- MINIO_ROOT_PASSWORD=an8charpassword
|
|
||||||
- MINIO_PROMETHEUS_AUTH_TYPE=public
|
|
||||||
volumes:
|
|
||||||
- minio_storage:/data
|
|
||||||
command:
|
|
||||||
- server
|
|
||||||
- /data
|
|
||||||
- --console-address
|
|
||||||
- ":9001"
|
|
||||||
|
|
||||||
alloy:
|
alloy:
|
||||||
image: grafana/alloy:latest
|
image: grafana/alloy:latest
|
||||||
|
@ -7,15 +7,11 @@ scrape_configs:
|
|||||||
static_configs:
|
static_configs:
|
||||||
# change this your machine's ip, localhost won't work
|
# change this your machine's ip, localhost won't work
|
||||||
# because localhost refers to the docker container.
|
# because localhost refers to the docker container.
|
||||||
# - targets: ['172.20.239.48:2500']
|
- targets: ['172.20.239.48:2500']
|
||||||
- targets: ['192.168.8.209:2500']
|
#- targets: ['192.168.8.209:2500']
|
||||||
- job_name: loki
|
- job_name: loki
|
||||||
static_configs:
|
static_configs:
|
||||||
- targets: ['loki:3100']
|
- targets: ['loki:3100']
|
||||||
- job_name: prometheus
|
- job_name: prometheus
|
||||||
static_configs:
|
static_configs:
|
||||||
- targets: ['localhost:9090']
|
- targets: ['localhost:9090']
|
||||||
- job_name: minio
|
|
||||||
metrics_path: /minio/v2/metrics/cluster
|
|
||||||
static_configs:
|
|
||||||
- targets: ['minio:9000']
|
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
use std::{ffi::OsStr, path::PathBuf};
|
use std::{ffi::OsStr, path::PathBuf};
|
||||||
|
|
||||||
use tokio::fs;
|
use tokio::fs;
|
||||||
use tracing::{error, instrument, trace, warn};
|
use tracing::{debug, error, info, instrument, trace, warn};
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
#[instrument(skip(data))]
|
#[instrument(skip(data))]
|
||||||
@ -21,6 +21,8 @@ pub async fn store(data: &str, url: &Url) {
|
|||||||
(url_path.clone(), "index.html".into())
|
(url_path.clone(), "index.html".into())
|
||||||
};
|
};
|
||||||
|
|
||||||
|
info!("Writing at: {:?} {:?}", basepath, filename);
|
||||||
|
|
||||||
// create the folders
|
// create the folders
|
||||||
if let Err(err) = fs::create_dir_all(&basepath).await {
|
if let Err(err) = fs::create_dir_all(&basepath).await {
|
||||||
error!("Dir creation: {err} {:?}", basepath);
|
error!("Dir creation: {err} {:?}", basepath);
|
||||||
|
@ -19,8 +19,7 @@ impl TokenSink for Website {
|
|||||||
TagToken(tag) => {
|
TagToken(tag) => {
|
||||||
if tag.kind == StartTag {
|
if tag.kind == StartTag {
|
||||||
match tag.name {
|
match tag.name {
|
||||||
// this should be all the html
|
// this should be all the html elements that have links
|
||||||
// elements that have links
|
|
||||||
local_name!("a")
|
local_name!("a")
|
||||||
| local_name!("audio")
|
| local_name!("audio")
|
||||||
| local_name!("area")
|
| local_name!("area")
|
||||||
@ -35,37 +34,9 @@ impl TokenSink for Website {
|
|||||||
let attr_name = attr.name.local.to_string();
|
let attr_name = attr.name.local.to_string();
|
||||||
if attr_name == "src" || attr_name == "href" || attr_name == "data"
|
if attr_name == "src" || attr_name == "href" || attr_name == "data"
|
||||||
{
|
{
|
||||||
let url: Option<Url> = match Url::parse(&attr.value) {
|
trace!("Found `{}` in html `{}` tag", &attr.value, tag.name);
|
||||||
Ok(ok) => {
|
let url = try_get_url(&self.site, &attr.value);
|
||||||
trace!("Found `{}` in the html on `{}` tag", ok.to_string(), tag.name);
|
|
||||||
Some(ok)
|
|
||||||
},
|
|
||||||
Err(e) => {
|
|
||||||
if attr.value.starts_with('#') {
|
|
||||||
trace!("Rejecting # url");
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
match e {
|
|
||||||
url::ParseError::RelativeUrlWithoutBase => {
|
|
||||||
let origin = self.site.origin().ascii_serialization();
|
|
||||||
let url = origin.clone() + &attr.value;
|
|
||||||
trace!("Built `{url}` from `{origin} + {}`", &attr.value.to_string());
|
|
||||||
if let Ok(url) = Url::parse(&url) {
|
|
||||||
trace!("Saved relative url `{}` AS: `{}`", &attr.value, url);
|
|
||||||
Some(url)
|
|
||||||
} else {
|
|
||||||
error!("Failed to reconstruct a url from relative url: `{}` on site: `{}`", &attr.value, self.site.to_string());
|
|
||||||
None
|
|
||||||
}
|
|
||||||
},
|
|
||||||
_ => {
|
|
||||||
error!("MISC error: {:?} {:?}", e, &attr.value);
|
|
||||||
None
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
if let Some(mut parsed) = url {
|
if let Some(mut parsed) = url {
|
||||||
parsed.set_query(None);
|
parsed.set_query(None);
|
||||||
parsed.set_fragment(None);
|
parsed.set_fragment(None);
|
||||||
@ -119,3 +90,56 @@ pub async fn parse(site: &Website, data: &str) -> Vec<Website> {
|
|||||||
other_sites
|
other_sites
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[instrument]
|
||||||
|
fn try_get_url(parent: &Url, link: &str) -> Option<Url> {
|
||||||
|
match Url::parse(link) {
|
||||||
|
Ok(ok) => Some(ok),
|
||||||
|
Err(e) => {
|
||||||
|
if link.starts_with('#') {
|
||||||
|
trace!("Rejecting # url");
|
||||||
|
None
|
||||||
|
} else if link.starts_with("//") {
|
||||||
|
// if a url starts with "//" is assumed that it will adopt
|
||||||
|
// the same scheme as it's parent
|
||||||
|
// https://stackoverflow.com/questions/9646407/two-forward-slashes-in-a-url-src-href-attribute
|
||||||
|
let scheme = parent.scheme();
|
||||||
|
|
||||||
|
match Url::parse(&format!("{scheme}://{}", link)) {
|
||||||
|
Ok(url) => Some(url),
|
||||||
|
Err(err) => {
|
||||||
|
error!("Failed parsing realative scheme url: {}", err);
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// # This is some sort of realative url, gonna try patching it up into an absolute
|
||||||
|
// url
|
||||||
|
match e {
|
||||||
|
url::ParseError::RelativeUrlWithoutBase => {
|
||||||
|
// Is: scheme://host:port
|
||||||
|
let origin = parent.origin().ascii_serialization();
|
||||||
|
let url = origin.clone() + link;
|
||||||
|
|
||||||
|
trace!("Built `{url}` from `{origin} + {}`", link.to_string());
|
||||||
|
|
||||||
|
if let Ok(url) = Url::parse(&url) {
|
||||||
|
trace!("Saved relative url `{}` AS: `{}`", link, url);
|
||||||
|
Some(url)
|
||||||
|
} else {
|
||||||
|
error!(
|
||||||
|
"Failed to reconstruct a url from relative url: `{}` on site: `{}`",
|
||||||
|
link,
|
||||||
|
parent.to_string()
|
||||||
|
);
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
error!("MISC error: {:?} {:?}", e, link);
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user