move kubectl config and yaml related to model module (#806)
* move yaml and config to models module * better naming * address comments
This commit is contained in:
Boshi Lian
255
src/KubernetesClient.Models/KubernetesYaml.cs
Normal file
255
src/KubernetesClient.Models/KubernetesYaml.cs
Normal file
@@ -0,0 +1,255 @@
|
||||
using System.IO;
|
||||
using System.Reflection;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
using YamlDotNet.Core;
|
||||
using YamlDotNet.Core.Events;
|
||||
using YamlDotNet.Serialization;
|
||||
using YamlDotNet.Serialization.NamingConventions;
|
||||
using k8s.Models;
|
||||
|
||||
namespace k8s
|
||||
{
|
||||
/// <summary>
|
||||
/// This is a utility class that helps you load objects from YAML files.
|
||||
/// </summary>
|
||||
public static class KubernetesYaml
|
||||
{
|
||||
private static readonly IDeserializer Deserializer =
|
||||
new DeserializerBuilder()
|
||||
.WithNamingConvention(CamelCaseNamingConvention.Instance)
|
||||
.WithTypeConverter(new IntOrStringYamlConverter())
|
||||
.WithTypeConverter(new ByteArrayStringYamlConverter())
|
||||
.WithTypeConverter(new ResourceQuantityYamlConverter())
|
||||
.WithOverridesFromJsonPropertyAttributes()
|
||||
.IgnoreUnmatchedProperties()
|
||||
.Build();
|
||||
|
||||
private static readonly IValueSerializer Serializer =
|
||||
new SerializerBuilder()
|
||||
.DisableAliases()
|
||||
.WithNamingConvention(CamelCaseNamingConvention.Instance)
|
||||
.WithTypeConverter(new IntOrStringYamlConverter())
|
||||
.WithTypeConverter(new ByteArrayStringYamlConverter())
|
||||
.WithTypeConverter(new ResourceQuantityYamlConverter())
|
||||
.WithEventEmitter(e => new StringQuotingEmitter(e))
|
||||
.ConfigureDefaultValuesHandling(DefaultValuesHandling.OmitNull)
|
||||
.WithOverridesFromJsonPropertyAttributes()
|
||||
.BuildValueSerializer();
|
||||
|
||||
private static readonly IDictionary<string, Type> ModelTypeMap = typeof(KubernetesEntityAttribute).Assembly
|
||||
.GetTypes()
|
||||
.Where(t => t.GetCustomAttributes(typeof(KubernetesEntityAttribute), true).Any())
|
||||
.ToDictionary(
|
||||
t =>
|
||||
{
|
||||
var attr = (KubernetesEntityAttribute)t.GetCustomAttribute(
|
||||
typeof(KubernetesEntityAttribute), true);
|
||||
var groupPrefix = string.IsNullOrEmpty(attr.Group) ? "" : $"{attr.Group}/";
|
||||
return $"{groupPrefix}{attr.ApiVersion}/{attr.Kind}";
|
||||
},
|
||||
t => t);
|
||||
|
||||
private class ByteArrayStringYamlConverter : IYamlTypeConverter
|
||||
{
|
||||
public bool Accepts(Type type)
|
||||
{
|
||||
return type == typeof(byte[]);
|
||||
}
|
||||
|
||||
public object ReadYaml(IParser parser, Type type)
|
||||
{
|
||||
if (parser?.Current is Scalar scalar)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (string.IsNullOrEmpty(scalar.Value))
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
return Encoding.UTF8.GetBytes(scalar.Value);
|
||||
}
|
||||
finally
|
||||
{
|
||||
parser.MoveNext();
|
||||
}
|
||||
}
|
||||
|
||||
throw new InvalidOperationException(parser.Current?.ToString());
|
||||
}
|
||||
|
||||
public void WriteYaml(IEmitter emitter, object value, Type type)
|
||||
{
|
||||
var obj = (byte[])value;
|
||||
emitter?.Emit(new Scalar(Encoding.UTF8.GetString(obj)));
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Load a collection of objects from a stream asynchronously
|
||||
///
|
||||
/// caller is responsible for closing the stream
|
||||
/// </summary>
|
||||
/// <param name="stream">
|
||||
/// The stream to load the objects from.
|
||||
/// </param>
|
||||
/// <param name="typeMap">
|
||||
/// A map from apiVersion/kind to Type. For example "v1/Pod" -> typeof(V1Pod). If null, a default mapping will
|
||||
/// be used.
|
||||
/// </param>
|
||||
/// <returns>collection of objects</returns>
|
||||
public static async Task<List<object>> LoadAllFromStreamAsync(Stream stream, IDictionary<string, Type> typeMap = null)
|
||||
{
|
||||
var reader = new StreamReader(stream);
|
||||
var content = await reader.ReadToEndAsync().ConfigureAwait(false);
|
||||
return LoadAllFromString(content, typeMap);
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Load a collection of objects from a file asynchronously
|
||||
/// </summary>
|
||||
/// <param name="fileName">The name of the file to load from.</param>
|
||||
/// <param name="typeMap">
|
||||
/// A map from apiVersion/kind to Type. For example "v1/Pod" -> typeof(V1Pod). If null, a default mapping will
|
||||
/// be used.
|
||||
/// </param>
|
||||
/// <returns>collection of objects</returns>
|
||||
public static async Task<List<object>> LoadAllFromFileAsync(string fileName, IDictionary<string, Type> typeMap = null)
|
||||
{
|
||||
using (var fileStream = File.OpenRead(fileName))
|
||||
{
|
||||
return await LoadAllFromStreamAsync(fileStream, typeMap).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Load a collection of objects from a string
|
||||
/// </summary>
|
||||
/// <param name="content">
|
||||
/// The string to load the objects from.
|
||||
/// </param>
|
||||
/// <param name="typeMap">
|
||||
/// A map from apiVersion/kind to Type. For example "v1/Pod" -> typeof(V1Pod). If null, a default mapping will
|
||||
/// be used.
|
||||
/// </param>
|
||||
/// <returns>collection of objects</returns>
|
||||
public static List<object> LoadAllFromString(string content, IDictionary<string, Type> typeMap = null)
|
||||
{
|
||||
var mergedTypeMap = new Dictionary<string, Type>(ModelTypeMap);
|
||||
// merge in KVPs from typeMap, overriding any in ModelTypeMap
|
||||
typeMap?.ToList().ForEach(x => mergedTypeMap[x.Key] = x.Value);
|
||||
|
||||
var types = new List<Type>();
|
||||
var parser = new Parser(new StringReader(content));
|
||||
parser.Consume<StreamStart>();
|
||||
while (parser.Accept<DocumentStart>(out _))
|
||||
{
|
||||
var obj = Deserializer.Deserialize<KubernetesObject>(parser);
|
||||
types.Add(mergedTypeMap[obj.ApiVersion + "/" + obj.Kind]);
|
||||
}
|
||||
|
||||
parser = new Parser(new StringReader(content));
|
||||
parser.Consume<StreamStart>();
|
||||
var ix = 0;
|
||||
var results = new List<object>();
|
||||
while (parser.Accept<DocumentStart>(out _))
|
||||
{
|
||||
var objType = types[ix++];
|
||||
var obj = Deserializer.Deserialize(parser, objType);
|
||||
results.Add(obj);
|
||||
}
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
public static async Task<T> LoadFromStreamAsync<T>(Stream stream)
|
||||
{
|
||||
var reader = new StreamReader(stream);
|
||||
var content = await reader.ReadToEndAsync().ConfigureAwait(false);
|
||||
return LoadFromString<T>(content);
|
||||
}
|
||||
|
||||
public static async Task<T> LoadFromFileAsync<T>(string file)
|
||||
{
|
||||
using (var fs = File.OpenRead(file))
|
||||
{
|
||||
return await LoadFromStreamAsync<T>(fs).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
|
||||
[Obsolete("use Deserialize")]
|
||||
public static T LoadFromString<T>(string content)
|
||||
{
|
||||
return Deserialize<T>(content);
|
||||
}
|
||||
|
||||
[Obsolete("use Serialize")]
|
||||
public static string SaveToString<T>(T value)
|
||||
{
|
||||
return Serialize(value);
|
||||
}
|
||||
|
||||
public static TValue Deserialize<TValue>(string yaml)
|
||||
{
|
||||
return Deserializer.Deserialize<TValue>(yaml);
|
||||
}
|
||||
|
||||
public static TValue Deserialize<TValue>(Stream yaml)
|
||||
{
|
||||
return Deserializer.Deserialize<TValue>(new StreamReader(yaml));
|
||||
}
|
||||
|
||||
public static string Serialize(object value)
|
||||
{
|
||||
if (value == null)
|
||||
{
|
||||
return "";
|
||||
}
|
||||
|
||||
var stringBuilder = new StringBuilder();
|
||||
var writer = new StringWriter(stringBuilder);
|
||||
var emitter = new Emitter(writer);
|
||||
|
||||
emitter.Emit(new StreamStart());
|
||||
emitter.Emit(new DocumentStart());
|
||||
Serializer.SerializeValue(emitter, value, value.GetType());
|
||||
|
||||
return stringBuilder.ToString();
|
||||
}
|
||||
|
||||
private static TBuilder WithOverridesFromJsonPropertyAttributes<TBuilder>(this TBuilder builder)
|
||||
where TBuilder : BuilderSkeleton<TBuilder>
|
||||
{
|
||||
// Use VersionInfo from the model namespace as that should be stable.
|
||||
// If this is not generated in the future we will get an obvious compiler error.
|
||||
var targetNamespace = typeof(VersionInfo).Namespace;
|
||||
|
||||
// Get all the concrete model types from the code generated namespace.
|
||||
var types = typeof(KubernetesEntityAttribute).Assembly
|
||||
.ExportedTypes
|
||||
.Where(type => type.Namespace == targetNamespace &&
|
||||
!type.IsInterface &&
|
||||
!type.IsAbstract);
|
||||
|
||||
// Map any JsonPropertyAttribute instances to YamlMemberAttribute instances.
|
||||
foreach (var type in types)
|
||||
{
|
||||
foreach (var property in type.GetProperties())
|
||||
{
|
||||
var jsonAttribute = property.GetCustomAttribute<JsonPropertyNameAttribute>();
|
||||
if (jsonAttribute == null)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
var yamlAttribute = new YamlMemberAttribute { Alias = jsonAttribute.Name, ApplyNamingConventions = false };
|
||||
builder.WithAttributeOverride(type, property.Name, yamlAttribute);
|
||||
}
|
||||
}
|
||||
|
||||
return builder;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user