Skip to content

Commit

Permalink
Merge branch 'development' into development_nodeselector
Browse files Browse the repository at this point in the history
  • Loading branch information
samaradel committed Feb 23, 2025
2 parents f151995 + bdf1c49 commit 96ffbe8
Show file tree
Hide file tree
Showing 40 changed files with 737 additions and 259 deletions.
2 changes: 1 addition & 1 deletion packages/grid_client/docs/network/mycelium_network.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ This configuration is set on [workload data](../../src/zos/zmachine.ts) and [net

In case user needs to support mycelium:

- flag `mycelium` field in workload with true, also the `planetary` field should be flagged with true.
- flag `mycelium` field in workload with true.
- User can support hex seed which ip will be generated from or they'll be generated automatically.
- providing seeds, that's what makes ip fixed from anywhere.
> Note user can't use same network seed in multiple nodes.
Expand Down
139 changes: 139 additions & 0 deletions packages/grid_client/scripts/applications/nostr.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,139 @@
import { Features, FilterOptions, GatewayNameModel, GridClient, MachinesModel, NodeInfo } from "../../src";
import { config, getClient } from "../client_loader";
import { log, pingNodes } from "../utils";

async function deploy(client: GridClient, vms: MachinesModel, subdomain: string, gatewayNode: NodeInfo) {
// Deploy VM
const resultVM = await client.machines.deploy(vms);
log("================= Deploying VM =================");
log(resultVM);
log("================= Deploying VM =================");

// Get WG interface details
const wgnet = (await client.machines.getObj(vms.name))[0].interfaces[0];

// Deploy Gateway
const gateway: GatewayNameModel = {
name: subdomain,
network: wgnet.network,
node_id: gatewayNode.nodeId,
tls_passthrough: false,
backends: [`http://${wgnet.ip}:8080`],
};

const resultGateway = await client.gateway.deploy_name(gateway);
log("================= Deploying Gateway =================");
log(resultGateway);
log("================= Deploying Gateway =================");
}

async function getDeployment(client: GridClient, name: string, subdomain: string) {
// Get VM deployment
const resultVM = await client.machines.getObj(name);
log("================= Getting VM Deployment =================");
log(resultVM);
log("================= Getting VM Deployment =================");

// Get Gateway deployment
const resultGateway = await client.gateway.getObj(subdomain);
log("================= Getting Gateway Deployment =================");
log(resultGateway);
log(`https://${resultGateway[0].domain}`);
log("================= Getting Gateway Deployment =================");
}

async function cancel(client: GridClient, name: string, subdomain: string) {
// Cancel VM deployment
const resultVM = await client.machines.delete({ name });
log("================= Canceling VM Deployment =================");
log(resultVM);
log("================= Canceling VM Deployment =================");

// Cancel Gateway deployment
const resultGateway = await client.gateway.delete_name({ name: subdomain });
log("================= Canceling Gateway Deployment =================");
log(resultGateway);
log("================= Canceling Gateway Deployment =================");
}

async function main() {
const name = "newnostr1";
const grid3 = await getClient(`nostr/${name}`);
const subdomain = `ntt${grid3.twinId}${name}`;
const instanceCapacity = { cru: 2, mru: 4, sru: 50 };

// VM Query Options
const vmQueryOptions: FilterOptions = {
features: [Features.wireguard, Features.mycelium],
cru: instanceCapacity.cru,
mru: instanceCapacity.mru,
sru: instanceCapacity.sru,
availableFor: grid3.twinId,
farmId: 1,
};

// Gateway Query Options
const gatewayQueryOptions: FilterOptions = {
features: [Features.wireguard, Features.mycelium],
gateway: true,
availableFor: grid3.twinId,
};

const gatewayNodes = await grid3.capacity.filterNodes(gatewayQueryOptions);
const gatewayNodeId = await pingNodes(grid3, gatewayNodes);
const gatewayNode = gatewayNodes.find(node => node.nodeId == gatewayNodeId);
const nodes = await grid3.capacity.filterNodes(vmQueryOptions);
const vmNode = await pingNodes(grid3, nodes);
const domain = `${subdomain}.${gatewayNode!.publicConfig.domain}`;

const vms: MachinesModel = {
name,
network: {
name: "nostrnet",
ip_range: "10.252.0.0/16",
addAccess: true,
accessNodeId: gatewayNode!.nodeId,
},
machines: [
{
name: "nostr",
node_id: vmNode,
disks: [
{
name: "nsDisk",
size: instanceCapacity.sru,
mountpoint: "/mnt/data",
},
],
planetary: true,
public_ip: false,
public_ip6: false,
mycelium: true,
cpu: instanceCapacity.cru,
memory: 1024 * instanceCapacity.mru,
rootfs_size: 0,
flist: "https://hub.grid.tf/tf-official-apps/nostr_relay-mycelium.flist",
entrypoint: "/sbin/zinit init",
env: {
SSH_KEY: config.ssh_key,
NOSTR_HOSTNAME: domain,
},
},
],
metadata: "",
description: "Deploying Nostr instance via TS Grid3 client",
};

// Deploy VM and Gateway
await deploy(grid3, vms, subdomain, gatewayNode!);

// Get the deployment details
await getDeployment(grid3, name, subdomain);

// Uncomment the line below to cancel the deployment
// await cancel(grid3, name, subdomain);

await grid3.disconnect();
}

main();
9 changes: 6 additions & 3 deletions packages/grid_client/scripts/dynamic_single_vm.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import { FilterOptions, MachinesModel } from "../src";
import { Features, FilterOptions, MachinesModel } from "../src";
import { config, getClient } from "./client_loader";
import { log } from "./utils";
import { log, pingNodes } from "./utils";

async function deploy(client, vms) {
const res = await client.machines.deploy(vms);
Expand Down Expand Up @@ -33,8 +33,11 @@ async function main() {
sru: 7,
availableFor: grid3.twinId,
country: "Belgium",
features: [Features.mycelium, Features.yggdrasil],
};

const nodes = await grid3.capacity.filterNodes(vmQueryOptions);
const vmNode = await pingNodes(grid3, nodes);
const vms: MachinesModel = {
name,
network: {
Expand All @@ -44,7 +47,7 @@ async function main() {
machines: [
{
name: "testvm",
node_id: +(await grid3.capacity.filterNodes(vmQueryOptions))[0].nodeId,
node_id: vmNode,
disks: [
{
name: "dynamicDisk",
Expand Down
32 changes: 22 additions & 10 deletions packages/grid_client/src/clients/tf-grid/contracts.ts
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,10 @@ export interface GetConsumptionOptions {
graphqlURL: string;
id: number;
}
export interface Consumption {
amountBilled: number;
discountReceived: DiscountLevel;
}

export interface GetDiscountPackageOptions {
graphqlURL: string;
Expand Down Expand Up @@ -268,7 +272,7 @@ class TFContracts extends Contracts {
const gqlClient = new Graphql(options.graphqlURL);

const body = `query getConsumption($contractId: BigInt!){
contractBillReports(where: {contractID_eq: $contractId} , orderBy: timestamp_DESC) {
contractBillReports(where: {contractID_eq: $contractId} , orderBy: timestamp_DESC, limit:1) {
discountReceived
}
Expand All @@ -282,7 +286,7 @@ class TFContracts extends Contracts {
if (billReports.length === 0) {
return "None";
} else {
const discountPackage = billReports[billReports.length - 1].discountReceived;
const discountPackage = billReports[0].discountReceived;
return discountPackage;
}
} catch (err) {
Expand All @@ -291,17 +295,19 @@ class TFContracts extends Contracts {
}
}
/**
* Get contract consumption per hour in TFT.
* Get the contract consumption details per hour in TFT.
*
* @param {GetConsumptionOptions} options
* @returns {Promise<number>}
* @returns {Promise<Consumption>} A promise resolving to the consumption details,
* including the amount billed and the discount received.
*/
async getConsumption(options: GetConsumptionOptions): Promise<number> {
async getConsumption(options: GetConsumptionOptions): Promise<Consumption> {
const gqlClient = new Graphql(options.graphqlURL);
const body = `query getConsumption($contractId: BigInt!){
contractBillReports(where: {contractID_eq: $contractId}, limit: 2 , orderBy: timestamp_DESC) {
amountBilled
timestamp
discountReceived
}
nodeContracts(where: {contractID_eq: $contractId}) {
createdAt
Expand All @@ -318,7 +324,10 @@ class TFContracts extends Contracts {
const gqlConsumption: GqlConsumption = response["data"] as GqlConsumption;
const billReports = gqlConsumption.contractBillReports;
if (billReports.length === 0) {
return 0;
return {
amountBilled: 0,
discountReceived: "None",
};
} else {
let duration = 1;
const amountBilled = new Decimal(billReports[0].amountBilled);
Expand All @@ -338,10 +347,13 @@ class TFContracts extends Contracts {
}
}
}
return amountBilled
.div(duration || 1)
.div(10 ** 7)
.toNumber();
return {
amountBilled: amountBilled
.div(duration || 1)
.div(10 ** 7)
.toNumber(),
discountReceived: billReports[0].discountReceived,
};
}
} catch (err) {
(err as Error).message = formatErrorMessage(`Error getting consumption for contract ${options.id}.`, err);
Expand Down
10 changes: 6 additions & 4 deletions packages/grid_client/src/modules/contracts.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ import { GridClientError } from "@threefold/types";
import * as PATH from "path";

import {
Consumption,
ContractsOverdue,
type DiscountLevel,
GqlContracts,
Expand Down Expand Up @@ -539,17 +540,18 @@ class Contracts {
return this.client.contracts.getDiscountPackage({ id: options.id, graphqlURL: this.config.graphqlURL });
}
/**
* Get contract consumption per hour in TFT.
* Get the contract consumption details per hour in TFT.
*
* @param {ContractConsumption} options
* @returns {Promise<number>}
* @param {ContractConsumption} options - The contract consumption parameters.
* @returns {Promise<Consumption>} A promise resolving to the consumption details,
* including the amount billed and the discount received.
* @decorators
* - `@expose`: Exposes the method for external use.
* - `@validateInput`: Validates the input options.
*/
@expose
@validateInput
async getConsumption(options: ContractConsumption): Promise<number> {
async getConsumption(options: ContractConsumption): Promise<Consumption> {
return this.client.contracts.getConsumption({ id: options.id, graphqlURL: this.config.graphqlURL });
}

Expand Down
26 changes: 16 additions & 10 deletions packages/grid_client/src/modules/k8s.ts
Original file line number Diff line number Diff line change
Expand Up @@ -143,22 +143,25 @@ class K8sModule extends BaseModule {
throw new ValidationError(`Another master with the same name ${master.name} already exists.`);
masters_names.push(master.name);

if (network) break;
// Sets the network and contractMetadata based on the node's zos version
const nodeTwinId = await this.capacity.getNodeTwinId(master.node_id);
const features = await this.rmb.request([nodeTwinId], "zos.system.node_features_get", "", 20, 3);
if (features.some(item => item.includes("zmachine-light") || item.includes("network-light"))) {
network = new ZNetworkLight(options.network.name, options.network.ip_range, this.config);
await network.load();
if (!network) {
network = new ZNetworkLight(options.network.name, options.network.ip_range, this.config);
await network.load();
}
contractMetadata = JSON.stringify({
version: 4,
type: "kubernetes",
name: options.name,
projectName: this.config.projectName || `kubernetes/${options.name}`,
});
} else {
network = new Network(options.network.name, options.network.ip_range, this.config);
await network.load();
if (!network) {
network = new Network(options.network.name, options.network.ip_range, this.config);
await network.load();
}
contractMetadata = JSON.stringify({
version: 3,
type: "kubernetes",
Expand Down Expand Up @@ -219,21 +222,24 @@ class K8sModule extends BaseModule {
throw new ValidationError(`Another worker with the same name ${worker.name} already exists.`);
workers_names.push(worker.name);

if (network) break;
const nodeTwinId = await this.capacity.getNodeTwinId(worker.node_id);
const features = await this.rmb.request([nodeTwinId], "zos.system.node_features_get", "", 20, 3);
if (features.some(item => item.includes("zmachine-light") || item.includes("network-light"))) {
network = new ZNetworkLight(options.network.name, options.network.ip_range, this.config);
await network.load();
if (!network) {
network = new ZNetworkLight(options.network.name, options.network.ip_range, this.config);
await network.load();
}
contractMetadata = JSON.stringify({
version: 4,
type: "kubernetes",
name: options.name,
projectName: this.config.projectName || `kubernetes/${options.name}`,
});
} else {
network = new Network(options.network.name, options.network.ip_range, this.config);
await network.load();
if (!network) {
network = new Network(options.network.name, options.network.ip_range, this.config);
await network.load();
}
contractMetadata = JSON.stringify({
version: 3,
type: "kubernetes",
Expand Down
1 change: 1 addition & 0 deletions packages/grid_client/src/modules/models.ts
Original file line number Diff line number Diff line change
Expand Up @@ -624,6 +624,7 @@ class FilterOptions {
@Expose() @IsOptional() @IsInt({ each: true }) @Min(1, { each: true }) nodeExclude?: number[];
@Expose() @IsOptional() @IsInt({ each: true }) @Min(1, { each: true }) farmIds?: number[];
@Expose() @IsOptional() @IsInt() @Min(1) farmId?: number;
@Expose() @IsOptional() @IsInt() @Min(1) nodeId?: number;
@Expose() @IsOptional() @IsString() farmName?: string;
@Expose() @IsOptional() @IsString() country?: string;
@Expose() @IsOptional() @IsString() city?: string;
Expand Down
4 changes: 3 additions & 1 deletion packages/grid_client/src/modules/tfchain.ts
Original file line number Diff line number Diff line change
Expand Up @@ -477,7 +477,9 @@ class TFChain implements blockchainInterface {
});
}
await (
await client.termsAndConditions.accept({ documentLink: "https://library.threefold.me/info/legal/#/" })
await client.termsAndConditions.accept({
documentLink: "https://manual.grid.tf/knowledge_base/legal/terms_conditions_all3.html",
})
).apply();
const ret = await (await client.twins.create({ relay })).apply();
if (disconnect) await client.disconnect();
Expand Down
7 changes: 6 additions & 1 deletion packages/grid_client/src/primitives/networklight.ts
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ interface NetworkMetadata {
class ZNetworkLight {
node: Node;
capacity: Nodes;
NodeIds: number[];
NodeIds: number[] = [];
deployments: Deployment[] = [];
network: NetworkLight;
contracts: Required<GqlNodeContract>[];
Expand Down Expand Up @@ -135,6 +135,7 @@ class ZNetworkLight {

this.node = new Node();
this.node.node_id = nodeId;
this.NodeIds.push(nodeId);

return znet_light_workload;
}
Expand Down Expand Up @@ -162,6 +163,10 @@ class ZNetworkLight {
ip = Addr(subnet).mask(32).increment().increment();
} else if (this.network["node_id"] === node_id) {
ip = Addr(this.getNodeSubnet(node_id)).mask(32).increment().increment();
const reserved_ips = this.getNodeReservedIps(node_id);
while (reserved_ips.includes(ip.toString().split("/")[0])) {
ip = ip.increment();
}
} else {
throw new ValidationError("node_id or subnet must be specified.");
}
Expand Down
Loading

0 comments on commit 96ffbe8

Please sign in to comment.